Similar to the QEMU and LXC drivers, add a helper function to
lookup a domain, and use it instead of much copy and paste.