简单实现linux聊天室程序

欢迎您访问我爱IT技术网，今天小编为你分享的电脑教程是服务器系列之：【简单实现linux聊天室程序】，下面是详细的分享！

简单实现linux聊天室程序

#include "i.h"
 
#define START_PORT 8089
 
struct sockaddr_in my_addr;
int my_id;
 
int my_log();
 

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); 
 
    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);
}


int init()
{  
  struct ifreq req;
  struct sockaddr_in *host;
  int port;
 
  i_init();
  
  bzero(&my_addr, sizeof(struct sockaddr));
  my_addr.sin_family=AF_INET;
  strcpy(req.ifr_name, "lo");
  if ( ioctl(sockfd, SIOCGIFADDR, &req) < 0 ) 
  {
    perror("get local ip error");
    exit(1);
    } 
 
  host=(struct sockaddr_in*)&(req.ifr_addr);
  printf("ip: %s\n", 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:%d\n", port); 
  printf("init successful!!!\n"); 
 
  return(0);
 
}


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);
}


void exit_sys()
{
  close(sockfd);
  close(mainfd);
  kill(0, SIGABRT);
 
  exit(0);
}

 

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()
{
  
  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);
}


int ordnary_msg_recv(struct msg *pmsg)
{
  char time_info[25];
  char end_symble;
  end_symble='&';
 
  
  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); 
     
     ordnary_msg_recv(&msg_recv);
  }
}
 

 

int login()
{
  
  printf("*****login>>\n");
  printf("id:");
  scanf("%d", &my_id);
  
  char password[15];
  printf("\npassword(*less 15 char):");
  scanf("%s", password);
  getchar();
   
  
  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); 
   
  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);
  
  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");
  
  char name[10];
 
  bzero(name, 10);
  printf("input your chat name(less 8 char):");
  scanf("%s", name);
 
  //name[9]=';';     
  
  char password[15];
 
  bzero(password, 15);
  printf("\ninput your password(less 14 char):");
  scanf("%s", password);
 
  
  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); 
 
  
  i_sendto(sockfd, (struct msg*)®_msg, sizeof(struct msg), 0, 
    (struct sockaddr*)&server, ADDR_LEN);
  
  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);
 
  
  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()
{
  
  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);
}

 
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聊天室程序的服务器维护教程分享，更多电脑教程请移步到>>电脑教程频道。