States that rely on a major river in the U.S. West are pushing for federal legislation to implement a plan to protect the water amid a prolonged drought.