8.21 IO（2）

使用分文件编译，实现注册登录界面，使用已经学过的fgets，fscanf，fpritnf函数。

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

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;
	}
	}
	return 0;
}

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;
}

login.c

#include"log.h"
int login()
{
	user log,reg;
	printf("请输入登录账户：");
	fgets(log.name,sizeof(log.name),stdin);
	printf("请输入登录密码：");
	fgets(log.code,sizeof(log.code),stdin);
	
	log.name[strlen(log.name)-1]='\0';
	log.code[strlen(log.code)-1]='\0';
	
	FILE *fp = fopen("./user.txt","r");
	if(fp==NULL)
	{
		perror("fopen");
		return -1;
	}
	fscanf(fp,"%s%s",reg.name,reg.code);
 
	if(strcmp(log.name,reg.name)==0&&strcmp(log.code,reg.code)==0)
	{
		printf("恭喜您登录成功，可以取钱\n");
	}
}