save registers: fix small problems

- Don't put C code at the beginning of save save_registers macro, as
  it will change the registers in unpredictable ways. Therefore, test
  whether ust_reg_stack_ptr has been initialized in assembly code.
- Save rflags at the begginning of the stub because using the test
  instruction may change its content.