IO DAY2

?

#include<my_head.h>
//定义注册函数***************************************************
int do_register()
{
? ? //以追加的形式打开文件
? ? FILE *wfp = 0;
? ? char name[20];
? ? char pwd[20];
? ? printf("请输入注册账号：");
? ? fgets(name,sizeof(name),stdin);
? ? name[strlen(name)-1] = 0;
? ? printf("请输入注册密码：");
? ? fgets(pwd,sizeof(pwd),stdin);
? ? pwd[strlen(pwd)-1] = 0;
? ? //打开文件
? ? if((wfp = fopen("./usr.txt","a+")) == 0)
? ? {
? ? ? ? perror("fopen");
? ? ? ? return -1;
? ? }

? ? //将注册账号和密码写入到文件中
? ? fprintf(wfp,"%s %s\n",name,pwd);
? ? //关闭文件
? ? fclose(wfp);
? ? printf("注册成功\n");
}
//定义登录函数*****************************************************
int do_login()
{
? ? //定义容器，存放账号密码
? ? char input_name[20];
? ? char input_pwd[20];
? ? char file_name[20];
? ? char file_pwd[20];
? ? FILE *rfp = 0;
? ? printf("请输入登录账号：");
? ? fgets(input_name,sizeof(input_name),stdin);
? ? input_name[strlen(input_name)-1] = 0;
? ? printf("请输入登录密码：");
? ? fgets(input_pwd,sizeof(input_pwd),stdin);
? ? input_pwd[strlen(input_pwd)-1] = 0;
? ? //打开文件
? ? if((rfp = fopen("./usr.txt","a+")) == 0)
? ? {
? ? ? ? perror("fopen");
? ? ? ? return -1;
? ? }
? ??
? ? while(1)
? ? {
? ? ? ? int res = fscanf(rfp,"%s %s",file_name,file_pwd);
? ? ? ? if(res <0)
? ? ? ? {
? ? ? ? ? ? printf("登录失败\n");
? ? ? ? ? ? return 1;
? ? ? ? }
? ? ? ? if(strcmp(input_name,file_name) == 0 &&strcmp(input_pwd,file_pwd) == 0)
? ? ? ? {
? ? ? ? ? ? printf("登录成功\n");
? ? ? ? ? ? return 0;
? ? ? ? }
? ? }

}

/******************************* 主程序 **********************************/
int main(int argc, const char *argv[])
{
? ? char menu;
? ? while(1)
? ? {
? ? ? ? printf("\t\t==== 1 注册 ====\n");
? ? ? ? printf("\t\t==== 2 登录 ====\n");
? ? ? ? printf("\t\t==== 0 退出 ====\n");

? ? ? ? printf("请输入选项:");
? ? ? ? scanf("%c",&menu);
? ? ? ? while(getchar() != '\n');
? ? ? ? switch(menu)
? ? ? ? {
? ? ? ? ? ? case'1':
? ? ? ? ? ? ? ? {
? ? ? ? ? ? ? ? ? ? do_register();
? ? ? ? ? ? ? ? }
? ? ? ? ? ? ? ? break;
? ? ? ? ? ? case'2':
? ? ? ? ? ? ? ? {
? ? ? ? ? ? ? ? ? ? do_login();
? ? ? ? ? ? ? ? }
? ? ? ? ? ? ? ? break;
? ? ? ? ? ? case'0':
? ? ? ? ? ? ? ? exit(EXIT_SUCCESS);

? ? ? ? ? ? default:printf("输入有误，请重新输入！！！\n");
? ? ? ? }
? ??
? ? ? ? printf("输入任意键按回车清屏！！！\n");
? ? ? ? while(getchar() != '\n');
? ? ? ? system("clear");
? ? }
? ??
? ??
? ? return 0;
}