Linux——传输层基于TCP协议的套接字编程实例

TCP

1.含义：同udp相同也是传输层的一种协议。（Transmission Control Protocol 传输控制协议）。

2.特点：面向连接、传输可靠、提供字节流服务

    面向连接：服务端必须与客户端建立连接。

    传输可靠：一系列的机制（应答等）保证数据传输的可靠性。

    提供字节流服务：发送数据不会像udp那样一次发送（要么成功，要么失败），而是灵活的发送（能发多少发多少）。

3.流程：

TCP的3次握手，4次挥手

4.实现过程：

（1）创建一个TcpSocket类封装相应的功能

（2）服务端程序

（3）客户端程序

5.存在问题：

服务端一次与一个客户端只能通信一次

6.实现：

多进程实现：

#include <signal.h>
#include <sys/wait.h>
#include "tcp_socket.hpp"

void sigcb(int signo){
  //在一次回调中处理完所有的僵尸进程
  //  -1：等待任意退出的子进程
  //  WNOHANG：设置为非阻塞等待
  while(waitpid(-1, NULL, WNOHANG) > 0);
}

int main(int argc, char* argv[]){
  //先判断参数是否够三个
  if(argc != 3){
    cout << "./tcp_srv 192.168.136.146 9000" << endl;
    return -1;
  }
  signal(SIGCHLD, sigcb);
  string ip = argv[1];
  uint16_t port = atoi(argv[2]);
  TcpSocket sock;
  //1.创建套接字
  CHECK_RET(sock.Socket());
  //2.绑定地址信息
  CHECK_RET(sock.Bind(ip,port));
  //3.开始监听
  CHECK_RET(sock.Listen());
  TcpSocket newsock;
  while(1){
    //5.获取已完成连接
    bool ret = sock.Accept(newsock);
    if(ret == false){
      continue;
    }
    //if(fork() == 0){
    //  if(fork() > 0){
    //    exit(0);
    //  }
    if(fork() == 0){
      while(1){
        //直到获取到新连接，否则就阻塞等待
        string buf;
        //接收客户端发送的数据
        ret = newsock.Recv(buf);
        if(ret == false){
          newsock.Close();
          continue;
        }
        cout << "client say:" << buf << endl;
        //7.发送数据
        buf.clear();
        cin >> buf;
        newsock.Send(buf);
      }
      newsock.Close();
      exit(0);
    }
    newsock.Close();
  }
  //8.关闭套接字
  sock.Close();
  return -1;
  }

多线程实现：

//多线程：服务端程序
#include <pthread.h>
#include "tcp_socket.hpp"


void* thr_start(void* arg){
  //直到获取到新连接，否则就阻塞等待
  TcpSocket* newsock = (TcpSocket*)arg;
  while(1){
    string buf;
    //接收客户端发送的数据
    int ret = newsock->Recv(buf);
    if(ret == false){
      newsock->Close();
      return NULL;
    }
    cout << "client say:" << buf <<endl;
    //7.发送数据
    buf.clear();
    cin >> buf;
    newsock->Send(buf);
  }
  newsock->Close();
  delete newsock;
  return NULL;
}



int main(int argc, char* argv[]){
  //先判断参数是否够三个
  if(argc != 3){
    cout << "./tcp_srv 192.168.136.146 9000" << endl;
    return -1;
  }
  string ip = argv[1];
  uint16_t port = atoi(argv[2]);
  TcpSocket sock;
  //1.创建套接字
  CHECK_RET(sock.Socket());
  //2.绑定地址信息
  CHECK_RET(sock.Bind(ip,port));
  //3.开始监听
  CHECK_RET(sock.Listen());
  while(1){
    TcpSocket* newsock = new TcpSocket();
    //5.获取新连接
    bool ret = sock.Accept(*newsock);
    if(ret == false){
      delete newsock;
      continue;
    }
    pthread_t tid;
    //创建新线程，让子线程来解决这些问题
    pthread_create(&tid, NULL, thr_start, (void*)newsock);
    pthread_detach(tid);
  }
  //8.关闭套接字
  sock.Close();
  return -1;
}