Water companies in parts of England have been told to step up drought planning as the government takes measures to protect customer supplies, Defra said.