本文共 2133 字，大约阅读时间需要 7 分钟。

#include "Acceptor.h"#include "InetAddress.h"#include "TcpStream.h"#include 
   
    #include 
    
     #include 
     
      int write_n(int fd, const void* buf, int length){
     int written = 0;  while (written < length)  {
       int nw = ::write(fd, static_cast
      
       (buf) + written, length - written);    if (nw > 0)    {
         written += nw;    }    else if (nw == 0)    {
         break;  // EOF    }    else if (errno != EINTR)    {
         perror("write");      break;    }  }  return written;}void run(TcpStreamPtr stream){
     // Caution: a bad example for closing connection  std::thread thr([&stream] () {
       char buf[8192];    int nr = 0;    // 接收stdin（输入缓冲区）数据，发送给服务器    while ( (nr = stream->receiveSome(buf, sizeof(buf))) > 0)    {
         int nw = write_n(STDOUT_FILENO, buf, nr);      if (nw < nr)      {
           break;      }    }    ::exit(0);  // should somehow notify main thread instead  });  //接收服务器信息，接收后输出到stdout（输出缓冲区）  char buf[8192];  int nr = 0;  while ( (nr = ::read(STDIN_FILENO, buf, sizeof(buf))) > 0)  {
       int nw = stream->sendAll(buf, nr);    if (nw < nr)    {
         break;    }  }  stream->shutdownWrite();  thr.join();}int main(int argc, const char* argv[]){
     if (argc < 3)  {
       printf("Usage:\n  %s hostname port\n  %s -l port\n", argv[0], argv[0]);    return 0;  }  int port = atoi(argv[2]);  // 服务器-l  if (strcmp(argv[1], "-l") == 0)  {
       std::unique_ptr
       
         acceptor(new Acceptor(InetAddress(port)));    TcpStreamPtr stream(acceptor->accept());    if (stream)    {
         acceptor.reset();  // stop listening      run(std::move(stream));    }    else    {
         perror("accept");    }  }  else //客户端  {
       InetAddress addr(port);    const char* hostname = argv[1];    if (InetAddress::resolve(hostname, &addr))    {
         TcpStreamPtr stream(TcpStream::connect(addr));      if (stream)      {
           run(std::move(stream));      }      else      {
           printf("Unable to connect %s\n", addr.toIpPort().c_str());        perror("");      }    }    else    {
         printf("Unable to resolve %s\n", hostname);    }  }}
       
      
     
    
   

转载地址：http://cfhpn.baihongyu.com/