步骤6：修改注册功能头文件和源文件