Linux 控制台聊天程序

//客户端代码 

#include<stdio.h>

#include<string.h>

#include<errno.h>

#include<sys/socket.h>

#include<resolv.h>

#include<stdlib.h>

#include<netinet/in.h>

#include<arpa/inet.h>

#include<unistd.h>

#define MAXBUF 1024

int main(int argc, char** argv)

{

 int sockfd,len;

 struct sockaddr_in dest;

 char buffer[MAXBUF+1];

 if(argc!=3)

   {

    printf("(Fail)");

    exit(0);

   }

if((sockfd=socket(AF_INET,SOCK_STREAM,0))<0) //create socket

  {

   perror("(Socket)");

   exit(errno);

  }

 printf("(socket created)\n");

 bzero(&dest,sizeof(dest)); //clear dest

 dest.sin_family=AF_INET;

 dest.sin_port=htons(atoi(argv[2]));

 if(inet_aton(argv[1],(struct in_addr*)&dest.sin_addr.s_addr)==0)

   {

    perror(argv[1]);

    exit(errno);

   }

 printf("(address created)\n");

 if(connect(sockfd,(struct sockaddr_in*)&dest,sizeof(dest))!=0)

   {

    perror("(Connect)");

    exit(errno);

   }

 printf("(server connected)\n");

 printf("(server sents first, both send in turn)\n");

 while(1)

 {

  bzero(buffer,MAXBUF+1);

  len=recv(sockfd,buffer,MAXBUF,0);

  if(len>0)

    printf("server:%s\n",buffer,len);

  else

    {

     if(len<0)

       printf("(Fail)\n");

     else

       printf("(Quit)\n");

       break;

    }

 bzero(buffer,MAXBUF+1);

 fgets(buffer,MAXBUF,stdin);

 if(!strncasecmp(buffer,"quit",4))

   {

    printf("(Quit)\n");

    break;

   }

 len=send(sockfd,buffer,strlen(buffer)-1,0);

 if(len<0)

   {

    printf("(Fail)\n");

    break;

   }

 }

close(sockfd);

return 0;

} 

/***********************************************************************************************************************************************************************/
//服务器代码

 #include<stdio.h>

#include<stdlib.h>

#include<errno.h>

#include<string.h>

#include<sys/types.h>

#include<netinet/in.h>

#include<sys/socket.h>

#include<sys/wait.h>

#include<unistd.h>

#include<arpa/inet.h>

#define MAXBUF 1024

int main(int argc, char** argv)

{

 int sockfd, new_fd;

 socklen_t len;

 struct sockaddr_in my_addr, their_addr;

 unsigned int myport, lisnum;

 char buf[MAXBUF+1];

 if(argv[1])

   myport=atoi(argv[1]); //atoi converts string to int

 else 

   myport=7838;

 if(argv[2])

   lisnum=atoi(argv[2]);

 else

   lisnum=2;

 if((sockfd=socket(PF_INET,SOCK_STREAM,0))==-1) //create socket

   {

    perror("socket");

    exit(1);

   }

 bzero(&my_addr,sizeof(struct sockaddr_in)); //clear my_addr to zero

 my_addr.sin_family=PF_INET;

 my_addr.sin_port=htons(myport); //converts the type order

 if(argv[3])

   my_addr.sin_addr.s_addr=inet_addr(argv[3]); //inet_addr converts IP address to int

 else

   my_addr.sin_addr.s_addr=INADDR_ANY;

 if(bind(sockfd,(const struct sockaddr_in*)&my_addr,sizeof(struct sockaddr_in))==-1) //bind 

   {

    perror("bind");

    exit(1);

   }

if (listen(sockfd,lisnum)==-1) //listen

   {

    perror("listen");

    exit(1);

   }

while(1)

{

 printf("\n(----Server start....)\n");

 printf("(server sents first, both send in turn)\n");

 len=sizeof(struct sockaddr_in);

 if((new_fd=accept(sockfd,(struct sockaddr_in*)&their_addr,&len))==-1) //accept

   {

    perror("accept");

    exit(errno);

   }

 else

   printf("(server: got connection from %s, port %d, socket %d)\n",inet_ntoa(their_addr.sin_addr),ntohs(their_addr.sin_port),new_fd);

while(1)

{

 bzero(buf,MAXBUF+1); //clear buffer

 fgets(buf,MAXBUF,stdin);

 if(!strncasecmp(buf,"quit",4)) //if received "quit", then quit

   {

    printf("\n(Quit!)\n");

    break;

   }

 len=send(new_fd,buf,strlen(buf)-1,0); //send messages to the other socket

 if(len<0)

   {

    printf("(Fail)\n");

    break;

   }

 bzero(buf,MAXBUF+1); //clear buffer

 len=recv(new_fd,buf,MAXBUF,0);

 if(len>0)

   printf("client:%s\n",buf);

 else

   {

    if(len<0)

      printf("(Fail)");

    else

      printf("(Quit)\n");

      break;

   }

}

close(new_fd);

}

close(sockfd);

return 0;

}