avx: fix register saving/restoring in ZUC functions

Some XMM registers were not being saved for Windows.
FUNC_SAVE and FUNC_RESTORE macros have been modified
to add these missing registers, and some functions
have been refactored to use them, simplifying the code.