chatcli.cpp

《unix网络编程技术与分析》一书的C源程序

/* File : chatcli.cpp  */

#include <stdio.h>
#include <stdlib.h>
#include <vector.h>
#include "pthread.h"
#include "Mysocket.h"
#include "MyThread.h"
#include "TcpCliThr.h"

const int MAX_MESSAGE_LEN = 1000;
const int DEFAULT_PORT = 1234;

/****************************************************
class name : ChatServer
Function: A client used to chat 
******************************************************/

class ChatClient : public TcpCliThr
{
public:
   ChatClient(int port,char* server) : TcpCliThr(port,server) {}
   ~ChatClient(){WaitAllThr();} // Must wait, or Client'object will be destoried before execute ~TcpCliThr()
   
   void DealRecv(MyThread* thread);
   void DealSend(MyThread* thread);
protected:
   char* getMessage(char* buf,int len, FILE* fp);    
};

void ChatClient::DealRecv(MyThread* thread) 
{
      char buf[MAX_MESSAGE_LEN + 1];
      int socket = ((Receiver *)thread)->socket;
      
      while(1) {
         int len = recv(socket,buf,MAX_MESSAGE_LEN,0);
         if (len < 1) break;
         buf[len] = '\0';
         printf("\n%s",buf);
         }
      close(socket);
}
   
void ChatClient::DealSend(MyThread* thread) 
{
      char buf[MAX_MESSAGE_LEN + 1];
      
      int socket = ((Sender *)thread)->socket;
      printf("Connected to server. \n");
      /* send name to server */
      printf("Input name : ");
      if ( fgets(buf, MAX_MESSAGE_LEN, stdin) == NULL) {
         printf("\nExit.\n");
         close(socket);
         return;
         }
      Send(buf,strlen(buf),0);
      
      /* send message to server */  
      while (getMessage(buf, MAX_MESSAGE_LEN, stdin) != NULL) {
         Send(buf, strlen(buf),0);
         }
      printf("\nExit.\n");
      close(socket);
}

char* ChatClient::getMessage(char* buf,int len, FILE* fp) 
{
   printf("Talk:");
   fflush(stdout);
   return(fgets(buf, MAX_MESSAGE_LEN, fp));
}

/*******************************************************************/
int main(int argc, char *argv[]) 
{
  if (argc < 2) {       
    printf("Usage: %s <IP Address> <port>\n",argv[0]); 
    exit(1); 
    } 
  if (argc == 2) {
     ChatClient chat(DEFAULT_PORT,argv[1]);
     chat.ConnectServ();
     }
  if (argc == 3) {
     ChatClient chat(atoi(argv[2]),argv[1]);
     chat.ConnectServ();
     }
}