Add support for custom runtime upgrade logic

This adds support for registering custom runtime upgrade logic that can
be implemented in the runtime directly instead of putting it into a module.