736d735
< drainManager(NULL),
2319,2320c2318,2319
< unsigned int
< UFSHostDevice::drain(DrainManager *dm)
---
> DrainState
> UFSHostDevice::drain()
2323d2321
< drainManager = dm;
2325,2326c2323
< setDrainState(DrainState::Draining);
< return 1;
---
> return DrainState::Draining;
2329,2330c2326
< setDrainState(DrainState::Drained);
< return 0;
---
> return DrainState::Drained;
2341c2337
< if (drainManager == NULL) {
---
> if (drainState() != DrainState::Draining)
2343d2338
< }
2350,2351c2345
< drainManager->signalDrainDone();
< drainManager = NULL;
---
> signalDrainDone();