代码：
#ifndef _i_h #define _i_h #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #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 代码二：
#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, '\0', 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 user\n"); printf("\t c--chat history\n"); printf("\t h--help menu\n"); printf("\t e--exit the system\n"); 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 error\n"); 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:%d\n", 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 munu\n"); printf("\t s--send message\n"); printf("\t r--reply to\n"); printf("\t c--chat history\n"); printf("\t f--send files\n"); printf("\t e--exit the system\n"); 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("\n\t%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 : %d\n", log_msg.id_from);//printf("password: %s\n", 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 success\n"); return(0); } else { printf("login error:%s\n", 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(®_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(®_msg); /* save to history */ /* send regist informatin to server */ i_sendto(sockfd, (struct msg*)®_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) success\n", 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 input\n"); 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; }}
登录后复制
以上就是linux聊天室程序如何实现的详细内容，更多请关注【创想鸟】其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至253000106@qq.com举报，一经查实，本站将立刻删除。
发布者：PHP中文网，转转请注明出处：https://www.chuangxiangniao.com/p/3165144.html
