代码三：
#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) < 0 ) /* get local ip address */
{
perror("get local ip error");
exit(1);
}
host = (struct sockaddr_in*)&(req.ifr_addr);
printf("ip: %sn", inet_ntoa(host->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(®_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) 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;
}
}
