Add protection to prevent message receiving functions from dropping messages

[RobertZ2011]

If a message receiving function contains multiple await points it's possible for the function to be written in a way that would result in messages getting lost. Add ways to prevent this from happening.