代码：
#ifndef _i_h #define _i_h #include <math.h> #include <stdio.h> #include <sys> #include <arpa> #include <netinet> #include <unistd.h> #include <sys> #include <sys> #include <fcntl.h> #include <stdlib.h> #include <errno.h> #include <string.h> #include <inttypes.h> #include <time.h> #include <sys> #include <net> #include <signal.h> #include <ncurses.h> #include <math.h> #define sevr_ip "127.0.0.1" #define sevr_port 8081 #define cntnt_len 150 #define msg_len sizeof(struct msg) #define addr_len sizeof(struct sockaddr) #define usr_len sizeof(struct user) #define prt_len 8 #define hstr_len sizeof(struct chat_history) /* declare global variables */ int mainfd;/* used as chat histroy file handle*/ int sockfd;/* used as socket local handle */ int count; struct sockaddr_in server; /* msg is used for communicating message */ struct msg { int flag; /* flag meaning:1,ordinary; 2,log msg; 3,reg msg, other,file*/ int id_from; int id_to; char content[cntnt_len]; char append[10]; }; /* user is used information list */ struct user { int id; char name[10]; char password[10]; char *p_chatlog; struct sockaddr user_addr; }; /* chat_history used for reading chat history */ struct chat_history { char content[cntnt_len]; char time[25]; int to; int from; int count; }; /* i_functions below is funtions needed by both client and sever */ extern int i_saveto_chat(struct msg *pmsg); int i_clean_stdin () { while ('n' == getchar()) { continue; } return(0); } int i_print(char *pmsg, int size) { int i = 1; for (i; iid_from; hstr.to = pmsg->id_to; strncpy(hstr.content, pmsg->content, cntnt_len); strncpy(hstr.time, i_get_time(), 25); i_lseek(mainfd, 0, seek_end); i_write(mainfd, &hstr, hstr_len); return(0); } int i_print_history(int len, int i) { struct chat_history chat_reader; int j; int position; bzero(&chat_reader, hstr_len); if (i != 0) { position = len*i*hstr_len; i_lseek(mainfd, position, seek_end); } else { position = len*i*hstr_len; i_lseek(mainfd, hstr_len, seek_set); } for (j = 1; j <p><strong>代码二：</strong></p> <pre class="brush:bash;">#include "i.h" int user_list_fd; /* start:initialization */ int init() { i_init(); user_list_fd = i_open("./user_list", o_rdwr|o_creat); struct user usr; /* init the user list file's fist user to 0*/ memset((struct user*)&usr, ' ', sizeof(struct user)); i_lseek(user_list_fd, 0, seek_set); i_write(user_list_fd, (char*)&usr, usr_len); /* bind the struct sockaddr_in server to the sockfd */ i_bind(sockfd, (struct sockaddr*)&server, addr_len); struct chat_history apple; bzero(&apple, hstr_len); i_lseek(mainfd, 0, seek_set); i_write(mainfd, &apple, hstr_len); i_lseek(mainfd, -hstr_len, seek_end); i_read(mainfd, &apple, hstr_len); count = apple.count; return(0); } /* end:initialization */ /* start:message control */ int send_msg(struct msg *msg_recv, struct sockaddr *addr) { int i; struct user usr; /* a common message come */ printf("a ordinar message come !n"); i = msg_recv->id_to; i_lseek(user_list_fd, i*usr_len, seek_set); i_read(user_list_fd, &usr, usr_len); strncpy(msg_recv->append, usr.name, 10); i_sendto(sockfd, msg_recv, msg_len, 0, &(usr.user_addr), addr_len); printf("id%d send a message to id%d sucess!n", msg_recv->id_from, msg_recv->id_to); return(0); } int check_login(struct msg *msg_recv, struct sockaddr *addr) { int i = msg_recv->id_from;; struct user usr; /* a login requet */ printf("a login request come!n"); /* get the id's information */ i_lseek(user_list_fd, i*usr_len, seek_set); i_read(user_list_fd, &usr, usr_len); int n; n = strcmp(usr.password, msg_recv->content); /* 如果验证成功，则发送成功信息 */ if (n == 0) { /* save user new address */ i_lseek(user_list_fd, -usr_len, seek_cur); usr.user_addr = *addr; i_write(user_list_fd, &usr, usr_len); /* tell user pass */ i_sendto(sockfd, (struct msg*)msg_recv, sizeof(struct msg), 0, &(usr.user_addr), addr_len); } else { /* 出错的话的respond */ if (0 != n) { printf("id %d login error.n", i); bzero(msg_recv->content, cntnt_len); msg_recv->flag = -1; i_sendto(sockfd, (struct msg*)msg_recv, sizeof(struct msg), 0, &(usr.user_addr), addr_len); } return(1); } printf("id %d login sucess!n", i); return(0); } int reg_user(struct msg *msg_recv, struct sockaddr *addr) { struct user usr; printf("a regit requet come:n"); /* find the last user and hava the please to add a new user */ int n; i_lseek(user_list_fd, -usr_len, seek_end); i_read(user_list_fd, &usr, usr_len); /* 把新用户的信息赋值到usr然后填入到user list file中 */ const char *name; const char *password; name = &(msg_recv->content[0]); password = &(msg_recv->content[10]); strcpy((usr.name), name); strcpy(usr.password, password); memcpy(&(usr.user_addr),addr, addr_len); usr.id = (usr.id + 1); i_lseek(user_list_fd, 0, seek_end); i_write(user_list_fd, &usr, usr_len); msg_recv->id_from = usr.id; /* regist to the user list then tell the user reg success */ i_sendto(sockfd, (struct msg*)msg_recv, sizeof(struct msg), 0, addr, addr_len); printf("id %d regist sucess!n", usr.id); return(0); } int msg_cntl() { struct msg msg_recv; struct sockaddr addr_recv; printf("begin listen input...n"); int size = addr_len; for (;;) { bzero(&msg_recv, msg_len); i_recvfrom(sockfd, &msg_recv, sizeof(struct msg), 0, &addr_recv, &size); printf("message received...n"); i_saveto_chat(&msg_recv); switch (msg_recv.flag) { case 1 : send_msg(&msg_recv,(struct sockaddr*)&addr_recv);/* send ordinary chat */ break; case 2 : check_login(&msg_recv, (struct sockaddr*)&addr_recv); break; case 3 : reg_user(&msg_recv, (struct sockaddr*)&addr_recv); break; default : break; } } return(0); } /* end:message control*/ /* start:exit_sys()*/ int exit_sys() { close(sockfd); close(mainfd); close(user_list_fd); printf("exit system"); kill(0, sigabrt); exit(0); } /* end:exit_sys()*/ /* start:chat_history*/ int get_page_size() { struct chat_history page_size_reader; i_lseek(mainfd, -hstr_len, seek_end); i_read(mainfd, &page_size_reader, hstr_len); return(page_size_reader.count); } int read_chat_history() { printf("****char*history***"); printf("(n-nextpage; p-prepage; q-quit)n"); int page_num;/* */ int remains; int berry = get_page_size(); page_num = berry / 8; remains = berry % 8; if (remains != 0) page_num ++; else page_num = page_num; printf("there are %d page total %d items", page_num, berry); int i = -1; while (1) { char flag; if ((berry + i*8) >= 0) { printf("(%d~%d)n", (berry + i*8), (berry + (i+1)*8)); i_print_history(prt_len, i); printf("@@@n"); while ('n' == (flag = getchar())) { } switch (flag) { case 'p' : i--; break; case 'n' : i++; break; case 'q' : return(0); default : break; } if (i >= 0) { printf("have at the end!n"); printf("return to menu!n"); } } else { printf("(1~%d)n", remains); i_print_history(remains, 0); printf("#########over##############n"); return(0); } } return(0); } /* end:chat_history*/ /* start:menu*/ int menu() { sleep(1); printf("----------help----menu---------n"); printf("t r--report to usern"); printf("t c--chat historyn"); printf("t h--help menun"); printf("t e--exit the systemn"); printf("----------help_menu---------n"); int command = 0; printf("input command>"); command = getchar(); switch(command) { case 'c': read_chat_history(); break; case 'e': exit_sys(); break; case 'r': //report(); //break; default : menu(); break; } getchar(); return(0); } /* end:menu*/ int main() { init(); pid_t pid; switch (pid = fork()) { case -1 : perror("fork errorn"); exit(1); break; case 0 : msg_cntl(); break; default : menu(); break; } return(0); }
代码三：
#include "i.h" #define start_port 8089 struct sockaddr_in my_addr; int my_id; int my_log();/* declare funtion*/ /* */ int i_send_msg() { int id; struct msg the_msg; char end = '@'; printf("input recver id:"); scanf("%d", &id); getchar(); printf("ninput content:"); i_input(the_msg.content); char flag = 'y'; if (1) { the_msg.flag = 1; the_msg.id_from = my_id; the_msg.id_to = id; i_sendto(sockfd, &the_msg, sizeof(struct msg), 0, (struct sockaddr*)&server, sizeof(struct sockaddr)); i_saveto_chat(&the_msg); /* save to history */ printf("send to id:%d success.n", my_id); return(0); } else return(1); return(0); } int reply() { return(0); } int send_file() { return(0); } /**/ /* start:initialize */ int init() { struct ifreq req; struct sockaddr_in *host; int port; i_init(); /* init user addr */ bzero(&my_addr, sizeof(struct sockaddr)); my_addr.sin_family = af_inet; strcpy(req.ifr_name, "lo"); if ( ioctl(sockfd, siocgifaddr, &req) sin_addr)); memcpy(&my_addr, (struct sockaddr_in*)&(req.ifr_addr), sizeof(struct sockaddr_in)); port = start_port; do { port++; my_addr.sin_port = htons(port); bind(sockfd, (struct sockaddr*)&my_addr, sizeof(struct sockaddr)); } while (errno == eaddrinuse); struct chat_history apple; memset(&apple, 'b', hstr_len); i_lseek(mainfd, 0, seek_set); apple.count = 0; i_write(mainfd, &apple, hstr_len); i_lseek(mainfd, -hstr_len, seek_end); i_read(mainfd, &apple, hstr_len); count = apple.count; printf("port:%dn", port); printf("init successful!!!n"); return(0); } /* end:initialize */ /* start:chat_history*/ int get_page_size() { struct chat_history page_size_reader; i_lseek(mainfd, -hstr_len, seek_end); i_read(mainfd, &page_size_reader, hstr_len); return(page_size_reader.count); } int read_chat_history() { printf("****char*history***"); printf("(n-nextpage; p-prepage; q-quit)n"); int page_num;/* */ int remains; int berry = get_page_size(); page_num = berry / 8; remains = berry % 8; if (remains != 0) page_num ++; else page_num = page_num; printf("there are %d page total %d items", page_num, berry); int i = -1; while (1) { char flag; if ((berry + i*8) >= 0) { printf("(%d~%d)n", (berry + i*8), (berry + (i+1)*8)); i_print_history(prt_len, i); printf("@@@n"); while ('n' == (flag = getchar())) { } switch (flag) { case 'p' : i--; break; case 'n' : i++; break; case 'q' : return(0); default : break; } if (i >= 0) { printf("have at the end!n"); printf("return to menu!n"); } } else { printf("(1~%d)n", remains); i_print_history(remains, 0); printf("#########over##############n"); return(0); } } return(0); } /* end:chat_history*/ /* start:exit_sys*/ void exit_sys() { close(sockfd); close(mainfd); kill(0, sigabrt); exit(0); } /* end:exit_sys*/ /* start:menu*/ int print_menu() { printf("n--------------help--menu----------------n"); printf("t h--help munun"); printf("t s--send messagen"); printf("t r--reply ton"); printf("t c--chat historyn"); printf("t f--send filesn"); printf("t e--exit the systemn"); printf("----------------help--menu----------------n"); } int get_input(char *command) { printf(">"); scanf("%c", command); return(1); } int menu() { /* to avoid the output at mixed with the sub process */ sleep(1); print_menu(); char command; while (1 == get_input(&command)) { switch(command) { case 'h': print_menu(); break; case 's': i_send_msg(); break; case 'r': reply(); break; case 'f': send_file(); break; case 'c': read_chat_history(); break; case 'e': exit_sys(); break; default : printf(">"); break; } } return(0); } /* end:menu*/ /* start:message contol :send_msg and recv_msg */ int ordnary_msg_recv(struct msg *pmsg) { char time_info[25]; char end_symble; end_symble = '&'; /* handle the msg */ printf("message:from %s(id%d) to u:n", pmsg->append, pmsg->id_from); i_print(pmsg->content, msg_len); printf("nt%s", i_get_time()); return(0); } int file_msg_recv(struct msg *pmsg) { } int handle_msg(struct msg *pmsg) { if (pmsg->flag == 1) { ordnary_msg_recv(pmsg); return(0); } else if (pmsg->flag >= 4) { file_msg_recv(pmsg); return(0); } return(0); } int listen_msg() { struct msg msg_recv; struct sockaddr addr_recv; int len = addr_len; printf("begin listen...n"); for ( ; ; ) { i_recvfrom(sockfd, &msg_recv, msg_len, 0, &addr_recv, &len); i_saveto_chat(&msg_recv); /* save to history */ ordnary_msg_recv(&msg_recv); } } /* end:message contol*/ /* start:log process :login and regist */ int login() { /* input id:*/ printf("*****login>>n"); printf("id:"); scanf("%d", &my_id); /* input password*/ char password[15]; printf("npassword(*less 15 char):"); scanf("%s", password); getchar(); /* send login information */ struct msg log_msg; bzero(&log_msg, msg_len); log_msg.flag = 2; log_msg.id_from = my_id; log_msg.id_to = 0; strncpy(log_msg.content, password, 15); i_saveto_chat(&log_msg); /* save to history */ i_sendto(sockfd, (struct msg*)&log_msg, msg_len, 0, (struct sockaddr*)&server, sizeof(struct sockaddr)); //printf("log_msg : %dn", log_msg.id_from); //printf("password: %sn", log_msg.content); /* after input msg ,wait for server respond*/ struct sockaddr in_addr; int len = addr_len; i_recvfrom(sockfd, (struct msg*)&log_msg, msg_len,0, &in_addr, &len); if (2 == log_msg.flag) { printf("login successn"); return(0); } else { printf("login error:%sn", log_msg.content); printf("please relog..n"); menu(); } return (0); } int regist() { printf("*****regist>>n"); /* input chat name */ char name[10]; bzero(name, 10); printf("input your chat name(less 8 char):"); scanf("%s", name); //name[9] = ';'; /* add a ; symbol in the end of name */ /* input password */ char password[15]; bzero(password, 15); printf("ninput your password(less 14 char):"); scanf("%s", password); /* send regist information*/ struct msg reg_msg; bzero(&reg_msg, msg_len); reg_msg.flag = 3; reg_msg.id_from = 0; reg_msg.id_to = 0; bzero(reg_msg.content, cntnt_len); strncpy(reg_msg.content, name, 10); strncpy(&(reg_msg.content[10]), password, 15); reg_msg.content[25] = 'n'; i_saveto_chat(&reg_msg); /* save to history */ /* send regist informatin to server */ i_sendto(sockfd, (struct msg*)&reg_msg, sizeof(struct msg), 0, (struct sockaddr*)&server, addr_len); /* after input msg ,wait for server respond*/ printf("wating for server reply...n"); struct sockaddr in_addr; struct msg msg_back; int len = addr_len; bzero(&in_addr, addr_len); bzero(&msg_back, msg_len); i_recvfrom(sockfd,(struct msg*)&msg_back, msg_len,0, &in_addr, &len); /* check whether pass */ if (3 != msg_back.flag) { printf("error: %s n", msg_back.content); exit(1); } else my_id = msg_back.id_to; printf("congratulate! you have regist" "id %s(id %d) successn", msg_back.content, msg_back.id_to); login(); return(0); } int my_log() { /* choose login or regist*/ char flag; printf("are you want login or regist(l/r)n"); scanf("%c", &flag); getchar(); switch (flag){ case 'l' : login(); break; case 'r' : regist(); break; default : printf("error inputn"); my_log(); break; } return (0); } /* end:log */ int main() { init(); printf("n************welcome!************n"); my_log(); pid_t pid; switch (pid = fork()) { case -1 : perror("fork error!n"); exit(1); break; case 0 : listen_msg(); break; default : menu(); break; } }
© 版权声明
文章版权归作者所有，未经允许请勿转载。
THE END
