1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
|
void server(string ip) {
#ifdef WIN32
WSADATA wsaData;
if ((WSAStartup(MAKEWORD(2,2), &wsaData)!=0)) {
cout << SRV_WSA_ERR;
return;
}
#endif
SOCKET sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if(sock<0) {
cout << SRV_SOCK_ERR;
return;
} else cout << SRV_SOCK_OK;
sockaddr_in server_addr, client_addr;
server_addr.sin_family=AF_INET;
server_addr.sin_addr.s_addr = inet_addr(ip.c_str());
server_addr.sin_port=htons(40000);
socklen_t client_addr_size = sizeof(client_addr);
struct timeval tv;
tv.tv_sec = 5;
setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, (char *)&tv, sizeof(struct timeval));
if(bind (sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
cout << SRV_BIND_ERR;
return;
} else cout << SRV_BIND_OK;
int sb;
if(listen(sock,0)<0) cout << SRV_LISTEN_ERR;
else cout << SRV_LISTEN_OK;
SOCKET a = accept(sock, (struct sockaddr *)&client_addr, &client_addr_size);
if(a<0) cout << SRV_ACCEPT_ERR;
else cout << SRV_ACCEPT_OK << " " << inet_ntoa(client_addr.sin_addr) << ":" << ntohs(client_addr.sin_port) << "\n";
while(b_server) {
sb = recv(a, buffmsg, MTUSIZE, 0);
if(sb>0) {
boost::mutex::scoped_lock lock(io_mutex);
cout << "Received:\n" << buffmsg << "\n";
}
Wait(1);
}
boost::mutex::scoped_lock lock(io_mutex);
#ifdef __linux__
close(sock); close(a);
#else
closesocket(a); closesocket(sock);
WSACleanup();
#endif
}
int main(int argc, char* argv[]) {
int s;
ip=string(argv[1]);
boost::thread wt_server = boost::thread(boost::bind(server,ip));
#ifdef WIN32
WSADATA wsaData;
if ((WSAStartup(MAKEWORD(2,2), &wsaData)!=0)) {
cout << CL_WSA_ERR;
return 0;
}
#endif
SOCKET sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if(sock<0) {
cout << CL_SOCK_ERR;
return 0;
} else cout << CL_SOCK_OK;
sockaddr_in client_addr;
Wait(100);
ip=string(argv[2]);
client_addr.sin_family=AF_INET;
client_addr.sin_addr.s_addr = inet_addr(ip.c_str());
client_addr.sin_port=htons(40000);
struct timeval tv;
tv.tv_sec = 5;
setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, (char *)&tv, sizeof(struct timeval));
if(connect(sock, (struct sockaddr *)&client_addr, sizeof(client_addr))<0)
cout << CL_CONNECT_ERR;
else cout << CL_CONNECT_OK;
while(b_client) {
cin.getline(getCmd,1000);
if(strcmp(getCmd,"exit")==0) {
b_server = b_client = false;
wt_server.join();
#ifdef __linux__
close(sock);
#else
closesocket(sock); WSACleanup();
#endif
}
else if(strlen(getCmd)>0) {
s = send(sock, getCmd, strlen(getCmd)+1, 0);
if(s<0) cout << CL_SEND_ERR;
}
Wait(1);
}
return 0;
}
|