简单的用户信息管理

userInfo.h

#ifndef USER_INFO_H
#define USER_INFO_H

#include<stdlib.h>
#include<sys/types.h>
#include<stdio.h>
#include<string.h>
#include<fcntl.h>
#include<sys/stat.h>
#include<unistd.h>
#include<time.h>
#include<sys/socket.h>
#include<arpa/inet.h>

#include<stdint.h>
#define NAME_LEN 32
#define PASSWD_LEN NAME_LEN

enum{
SEX_MALE=1,
SEX_FEMALE
};

extern char *psex[3];

typedef struct{
char    login_name[NAME_LEN];
char    login_passwd[PASSWD_LEN];
char    user_name[NAME_LEN];
uint8_t user_sex;
uint8_t user_age;

}userInfo_t;

#endif

userList.h

#ifndef USER_LIST_H
#define USER_LIST_H

#include<userInfo.h>

#define USERDATA_FILE "userdata.log"

typedef struct userList{
userInfo_t data;
struct userList *next,*prev;
}sUserList,*pUserList;

pUserList userlist_init();

void userlist_destroy(pUserList p);

#endif

userList.c

#include"userList.h"

char *psex[3]={
NULL,"man","female"
};

pUserList userlist_init()
{
pUserList p;
p=(pUserList)malloc(sizeof(sUserList));
if(NULL==p)
{
perror("malloc");
exit(1);
}
bzero(&p->data,sizeof(userInfo_t));
p->next=p->prev=p;
return p;
}

void userlist_destroy(pUserList p)
{
if(p!=NULL)
{
free(p);
p=NULL;
}
}

main.c

#include "userList.h"
//功能：对用户数据进行管理
//链表
//链表的头结点：数据域为空的结点，用它来管理整个链表

const char *pmenu = "1.Insert 2.Delete 3.Search 4.Print 5.Save 6.Quit:";

enum {
INSERT=1, DELETE, SEARCH, PRINT, SAVE, QUIT
};

void insert_func(pUserList phead);
void delete_func(pUserList phead);
void search_func(pUserList phead);
void print_func(pUserList phead);
void save_func(pUserList phead);
void load_func(pUserList phead);

void (*pfunc[])(pUserList) = {
NULL, insert_func, delete_func, search_func, print_func, save_func
};

int main()
{
pUserList phead; //头结点指针
unsigned int choose;

phead = userlist_init();
load_func(phead);

while (1) {
  printf(pmenu);
  scanf("%u", &choose);   // %d %u %ld %lld %lu %llu
  if ( choose < INSERT || choose > QUIT ) {
   continue;
  }
  if ( choose == QUIT )
   break;

pfunc[choose](phead);

}

return 0;
}

void insert_func(pUserList phead)
{
printf("[%s]\n", __func__); //__LINE__ __FILE__
pUserList pnew, ptrav;

pnew = userlist_init();
printf("input the login name:");
scanf("%s", pnew->data.login_name);
printf("input the login passwd:");
scanf("%s", pnew->data.login_passwd);
printf("input the user name:");
scanf("%s", pnew->data.user_name);
do {
  printf("input the user sex(1:男性 2:女性):");
  scanf("%u", &pnew->data.user_sex);
} while ( pnew->data.user_sex != 1 && pnew->data.user_sex != 2 );
do {
  printf("input the user age:");
  scanf("%u", &pnew->data.user_age);
} while ( pnew->data.user_age <=0 || pnew->data.user_age >= 250 );

for ( ptrav=phead->next; ptrav!=phead; ptrav=ptrav->next) {
  if ( strcmp(pnew->data.login_name, ptrav->data.login_name) == 0 ) {
   printf("Insert Failed.\n");
   userlist_destroy(pnew);
   return ;
  }
} //查到没有这个人就插入

pnew->prev = phead->prev;
pnew->next = phead;
phead->prev->next = pnew;
phead->prev = pnew;
}
void delete_func(pUserList phead)
{
printf("[%s]\n", __func__);
pUserList ptrav;
char name_buf[NAME_LEN];
printf("input delete name:");
scanf("%s", name_buf);

for ( ptrav=phead->next; ptrav!=phead; ptrav=ptrav->next) {
  if ( strcmp(name_buf, ptrav->data.login_name) == 0 ) {
   ptrav->next->prev = ptrav->prev;
   ptrav->prev->next = ptrav->next;
   userlist_destroy(ptrav);
   return ;
  }
}
printf("Not Find.\n");
}
void search_func(pUserList phead)
{//按登录名查找
printf("[%s]\n", __func__);
pUserList ptrav;
char name_buf[NAME_LEN];
printf("input search name:");
scanf("%s", name_buf);

for ( ptrav=phead->next; ptrav!=phead; ptrav=ptrav->next) {
  if ( strcmp(name_buf, ptrav->data.login_name) == 0 ) {
   printf("[%s,%s,%s,%s,%d]\n", ptrav->data.login_name,
    ptrav->data.login_passwd,
    ptrav->data.user_name,
    psex[ptrav->data.user_sex],
    ptrav->data.user_age);
   return ;
  }
}
printf("Not Find.\n");
}

void print_func(pUserList phead)
{
printf("[%s]\n", __func__);
pUserList ptrav;

printf("HEAD->");
for ( ptrav=phead->next; ptrav!=phead; ptrav=ptrav->next) {
  printf("[%s,%s,%s,%s,%d]->", ptrav->data.login_name,
    ptrav->data.login_passwd,
    ptrav->data.user_name,
    psex[ptrav->data.user_sex],
    ptrav->data.user_age);
}
printf("HEAD\n");

}

void save_func(pUserList phead)
{
printf("[%s]\n", __func__);
int fd;

fd = open(USERDATA_FILE, O_WRONLY|O_CREAT|O_TRUNC, 0644);
if ( fd < 0 ) {
perror("open()");
return ;
}

pUserList ptrav;

for ( ptrav=phead->next; ptrav!=phead; ptrav=ptrav->next) {
write(fd, &ptrav->data, sizeof(userInfo_t));
}

close(fd);
}

void load_func(pUserList phead)
{
printf("[%s]\n", __func__);
int fd;
int ret;
pUserList pnew;

fd = open(USERDATA_FILE, O_RDONLY);
if ( fd < 0 ) {
perror("open()");
return ;
}

while (1) {
  pnew = userlist_init();
  ret = read(fd, &pnew->data, sizeof(userInfo_t));
  if ( ret <= 0 ) {
   userlist_destroy(pnew);
   break;
  }
  pnew->prev = phead->prev;
  pnew->next = phead;
  phead->prev->next = pnew;
  phead->prev = pnew;
}

close(fd);
}

简单的用户信息管理

猜你喜欢