main.c
#include"log.h"
int main(int argc, const char *argv[])
{
int ch;
while(1)
{
printf("\t\t\t1.注册\n");
printf("\t\t\t2.登录\n");
printf("\t\t\t3.退出\n");
printf("请输入你的选择:");
scanf("%d",&ch);
getchar();
switch(ch)
{
case 1:
regist();
break;
case 2:
login();
break;
case 3:
exit(EXIT_SUCCESS);
break;
default:
printf("账户或密码错误,请重新输入");
break;
}
}
return 0;
}
login.c
#include"log.h"
int login()
{
User log;
printf("请输入登录账户:");
fgets(log.name, sizeof(log.name), stdin);
if (log.name[strlen(log.name) - 1] == '\n')
{
log.name[strlen(log.name) - 1] = '\0';
}
printf("请输入登录密码:");
fgets(log.code, sizeof(log.code), stdin);
if (log.code[strlen(log.code) - 1] == '\n')
{
log.code[strlen(log.code) - 1] = '\0';
}
FILE *fp = fopen("./user.txt", "r");
if (fp == NULL) {
perror("fopen");
return -1;
}
User reg;
while (fscanf(fp, "%s %s", reg.name, reg.code) != EOF)
{
if (strcmp(log.name, reg.name) == 0 &&
strcmp(log.code, reg.code) == 0)
{
printf("恭喜您登陆成功,可以取钱\n");
fclose(fp);
return 0;
}
}
printf("登录失败\n");
fclose(fp);
return -1;
}
regist.c
#include"log.h"
int regist()
{
user reg;
printf("请输入注册的账户:");
fgets(reg.name,sizeof(reg.name),stdin);
printf("请输入注册的密码:");
fgets(reg.code,sizeof(reg.code),stdin);
FILE *fp = fopen("./user.txt","w");
if(fp==NULL)
{
perror("fopen");
return -1;
}
fprintf(fp,"%s%s",reg.name,reg.code);
printf("注册成功\n");
fclose(fp);
return 0;
}
log.h
#ifndef _LOG_H_
#define _LOG_H_
#include<myhead.h>
typedef struct
{
char name[20];
char code[20];
}user;
int login();
int regist();
#endif