TCP双向通信（windows系统）

#include <stdlib.h> #include <iostream.h> #include <winsock2.h> #pragma comment(lib, "wsock32.lib") #define NUM_THREAD 2 //定义线程个数 #define BUFFER_SIZE 256 //定义缓冲区大小 HANDLE session_thread_handle[NUM_THREAD]; //定义线程句柄 SOCKET clientSocket; //客户套接字 char recvBuffer[BUFFER_SIZE]; //接收缓冲区 char sendBuffer[BUFFER_SIZE]; //发送缓冲区 void AdminThread(); //管理线程函数 void RecvThread(); //接收线程 void SendThread(); //发送线程 void KillAllThread(); //线程回收 void main() { SOCKET sock; //socket //char szMsg[] = "this is a TCP test package";//被发送的字段 //1.启动SOCKET库，版本为2.0 WORD wVersionRequested; WSADATA wsaData; int err; wVersionRequested = MAKEWORD( 2, 0 ); err = WSAStartup( wVersionRequested, &wsaData ); if ( err != 0 ) { cout<<"Socket2.0初始化失败，Exit!"; return; } if ( LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 0 ) { WSACleanup( ); return; } //2.创建socket， sock = socket( AF_INET, //internetwork: UDP, TCP, etc SOCK_STREAM, //SOCK_STREAM说明是TCP socket 0 //protocol ); if (sock == INVALID_SOCKET ) { cout<<"Socket 创建失败，Exit!"; return; } //3.bind套接字，相当于给本地套接字赋值 sockaddr_in myaddr; memset(&myaddr,0,sizeof(myaddr)); myaddr.sin_family=AF_INET; myaddr.sin_addr.s_addr = ADDR_ANY; myaddr.sin_port=htons(7861); bind(sock,(sockaddr*)&myaddr,sizeof(myaddr)); //4.设置发往的地址 sockaddr_in addrto; //发往的地址 memset(&addrto,0,sizeof(addrto)); //5.listen socket if (listen(sock,5) == SOCKET_ERROR ) { closesocket(sock); WSACleanup(); abort(); } else { cout<<"listen succeed!"<<endl; } //6. accept 对方连接 int nlen = sizeof(addrto); clientSocket = accept(sock,(sockaddr*)&addrto,&nlen); if (clientSocket < 0) { cout<<"client socket error, exit!"<<endl; abort(); } else { cout<<"accepted client socket!"<<endl; } //调用管理线程函数创建线程、线程同步、回收线程 AdminThread(); if (!closesocket(sock)) { WSAGetLastError(); return; } if (!closesocket(clientSocket)) { WSAGetLastError(); return; } if (!WSACleanup()) { WSAGetLastError(); return; } } void AdminThread()//管理线程函数 { for(int i=0; i<NUM_THREAD; i++) { DWORD session_thread_id = i; switch(i) { case 0://利用CreateThread()函数创建接收线程 session_thread_handle[i] = CreateThread(0, 0, (LPTHREAD_START_ROUTINE)RecvThread, 0, 0, &session_thread_id); break; case 1://利用CreateThread()函数创建发送线程 session_thread_handle[i] = CreateThread(0, 0, (LPTHREAD_START_ROUTINE)SendThread, 0, 0, &session_thread_id); break; default: break; } } //利用WaitForMultipleObjects()函数实现线程同步 WaitForMultipleObjects(NUM_THREAD, session_thread_handle, TRUE, 0xffffffff); //线程回收 KillAllThread(); } void RecvThread()//接收线程 { int iResult = 0; while(true) { iResult = recv(clientSocket,recvBuffer,sizeof(recvBuffer),0); if (iResult < 0) { cout<<"客户方Socket关闭，退出！"<<endl; break; } cout<<"客户端："<<recvBuffer<<endl; memset(recvBuffer,0,sizeof(recvBuffer)); } } void SendThread()//发送线程 { while(true) { cin>>sendBuffer; // strncat(sendBuffer, "\x0D\x0A", sizeof(sendBuffer)); if( send(clientSocket, sendBuffer, strlen(sendBuffer), 0) == SOCKET_ERROR ) cout<<"error, error id = "<<WSAGetLastError()<<endl; } } void KillAllThread()//线程回收 { for(int i=0; i<NUM_THREAD; i++) { if(session_thread_handle[i]) { TerminateThread(session_thread_handle[i], 0); CloseHandle(session_thread_handle[i]); } } }