mini telnet 服务器客户端源码

#include "mynet.h" #include "fun.h" int flag = 1; //是否收到心跳包判断变量 int flag1 = 0; //记录心跳包数量 int flag2 = 0; //记录上次心跳数量 /*********************************创建用户信息**************************************/ /* *功能：创建用户信息 *参数：链表头指针 *返回值：返回建立链表头指针 */ ND *Creat(ND *head) { ND* newnode = NULL; //新结点 char flag; //是否继续输入控制变量 ND* p = NULL; //用于遍历链表的指针变量 do { newnode = (ND* )malloc(sizeof(ND)); //申请新结点空间 if ( NULL == newnode ) //判断空间是否分配成功 { printf("malloc error--%s\n", __func__); exit(0); } printf("Input user name(1～128bit)：____"); //输入姓名 scanf("%s", newnode->name); while ( strlen(newnode->name) > 128 ) //判断名字是否合格 { printf("length of name is wrong____"); scanf("%s", newnode->name); } printf("Input password(6～32bit)：____"); //输入密码 scanf("%s", newnode->passwd); while ( strlen(newnode->passwd) < 6 || strlen(newnode->passwd) > 32 )//判断密码是否合格 { printf("length of password is wrong____"); scanf("%s", newnode->passwd); } getchar(); newnode->next = head; // 头插法地址区赋值 head = newnode; printf("\nInput result is:\n"); printf("\tname\tpassword\n"); printf("\t%s\t%s\n", newnode->name, newnode->passwd); //输出信息 printf("\ncontinue input?? y/n:\n"); //是否继续输入的判断 scanf("%c", &flag); if ( flag == 'n') { break; } }while(1); return head; } /*********************************end**************************************/ /*********************************显示所有用户信息*************************************/ /* *功能：输出链表数据 *参数：链表头指针 *返回值：无返回值 */ void Output(ND* head) { if ( NULL == head ) //判断链表是否为空 { printf("link is empty%s\n", __func__); exit(0); } printf("\tname\tpassword\n"); //输出项目语句 while ( NULL != head) //遍历链表输出数据 { printf("\t%s\t%s\n", head->name, head->passwd);//输出信息 head = head->next; //链表头指针指向下一个结点 } } /*********************************end**************************************/ /************************************释放链表*****************************************/ /* *功能：释放链表 *参数：链表头指针 *返回值：无返回值 */ void Free(ND *head) { if ( NULL == head ) //判断链表是否为空 { printf("link is empty%s\n", __func__); exit(0); } ND *p = NULL; //中间结点指针 while(NULL != head) { p = head; //new指向链表头结点 head = p->next; //head指向下一个结点 free(p); //将头结点释放掉 } } /*********************************end**************************************/ /*******************************用户信息保存至文件*************************************/ /* *功能：将数据信息写入指定文件 *参数：链表头指针 *返回值：返回函数值 */ void Wr_file(ND *head) { ND *p = head; //p指向链表头 if ( NULL == head ) //判断链表是否为空 { printf("link is empty%s\n", __func__); exit(0); } FILE *fp = fopen("/home/iotek/mini telnet/userdata", "w");//打开指定文件"userdata"，将fp指向文件首地址，并准备写入 if ( NULL == fp ) //判断文件是否打开成功 { printf("open error\n"); exit(0); //文件打开失败，退出系统 } while( NULL != p ) //遍历链表，并将结点内容写入文件 { fprintf(fp, "\t%s\t%s\n", p->name, p->passwd);//将信息写入文件student p = p->next; //指针移向下一个结点 } fclose (fp); //写入完毕，关闭文件 } /*********************************end**************************************/ /******************************从文件中读取用户信息************************************/ /* *功能：从指定文件中读取信息 *参数：无参 *返回值：返回读出链表头指针 */ ND* R_file() { ND *head1 = NULL; //读出链表头指针 ND *tail = NULL; //读出链表尾指针 ND *new = NULL; //新建链表结点指针 FILE *fp = fopen("/home/iotek/mini telnet/userdata","r");//打开文件userdata，并将fp指向该文件，准备读取 if ( NULL == fp ) //判断文件是否打开成功 { printf("open error\n"); exit(0); //文件打开失败，退出系统 } while( !feof(fp) ) //读取文件，并将数据赋给新建链表,feof函数控制读取次数 { new = (ND *)malloc(sizeof(ND)); //申请新空间，并将new指向新空间地址 if ( NULL == new) //判断空间是否分配成功 { printf("malloc error%s\n", __func__); exit(0); //文件打开失败，退出系统 } fscanf(fp, "\t%s\t%s\n", new->name, new->passwd);//从文件中读取信息 if ( NULL == head1 ) //判断如果为链表第一个结点 { head1 = new; //新结点作为链表第一个节点 } else //如果链表已经存在，作为链表尾结点插入到链表中 { tail->next = new; //尾指针指向新结点 } new->next = NULL; //新结点地址区赋空值，作为链表尾 tail = new; //尾结点指向新建的结点 } fclose(fp); //关闭文件 return head1; //返回读出链表头指针 } /*********************************end**************************************/ /*********************************验证用户名和密码*************************************/ /* *功能：用户名与密码核对 *参数：用户输入消息内容指针 *返回值：成功返回1， 失败返回0 */ int Check_user(char *buf) { int flag = 0; //判断变量 ND* p = NULL; //移动指针变量 char buf_cat[BUFFER_SIZE]; //链接后字符串 ND* head = R_file(); //读取文本文件信息 p = head; //p首先指向链表头 while ( NULL != p ) { memset(buf_cat, '\0', BUFFER_SIZE); strcpy(buf_cat, p->name); strcat(buf_cat, p->passwd);//姓名和密码拼接 strcat(buf_cat, "\n"); if ( strcmp( buf+1, buf_cat ) == 0 )//核对用户名与密码 { flag = 1; } p = p->next; //p遍历链表 } return flag; } /*********************************end**************************************/ /*******************************执行指令，保存至缓存区*********************************/ /* *功能：从缓冲区执行命令输出显示结果 *参数：命令字符串, 套接字 *返回值：0失败，1成功 */ int Popen_fun(int sockfd, char *cmd) { FILE *fp = NULL; //定义文件指针 int ret; //读取结果返回值 char buf1[BUFFER_SIZE]; //存放执行结果 if ((fp = popen(cmd+1, "r")) == NULL) //执行指令cmd，并写入fp指向的文件 { perror("popen"); return; } memset(cmd, '\0', BUFFER_SIZE); memset(buf1, '\0', BUFFER_SIZE); ret = fread(buf1, sizeof(char), sizeof(buf1), fp);//从文件中读取结果到 buf1 // printf("ret = %d \n", ret); if ( ret == 0 ) //读取失败处理 { strcat(cmd, "\n....Command error!\n\n"); //提示语句写到客户端 write(sockfd, cmd, strlen(cmd)); memset(cmd, '\0', BUFFER_SIZE); memset(buf1, '\0', BUFFER_SIZE); } else { strcat(cmd, buf1); //执行结果写回客户端 write(sockfd, cmd, strlen(cmd)); memset(cmd, '\0', BUFFER_SIZE); memset(buf1, '\0', BUFFER_SIZE); } } /*********************************end**************************************/ /**********************************解析客户端消息**************************************/ /* *功能：操作选择函数 *参数：操作字符串，已链接套接字 *返沪值：判断信息 */ int Do_function(int sockfd, char buf[BUFFER_SIZE], struct sockaddr_in cliaddr) { ND *head = NULL; int flag; //判定变量 char warn[BUFFER_SIZE]; //用于记录需返回客户端显示的信息 char ch[LEN2]; //记录转换的IP地址 signal(SIGALRM, Handle_heart);