1 //tcpclient.h
2 #ifndef TCPCLIENT_H
3 #define TCPCLIENT_H
4 #include <winsock2.h>
5 #include <ws2tcpip.h>
6 #include <iostream>
7 #include <cstdio>
8
9
10 extern SOCKET ConnectSocket;
11 int ConnectSer();
12
13 #endif // TCPCLIENT_H
1 //tcpclient.c
2 #include "tcpclient.h"
3
4 using namespace std;
5
6 // Need to link with Ws2_32.lib
7 //#pragma comment(lib, "ws2_32.lib")
8 SOCKET ConnectSocket; //定义连接Socket，在发送接收时使用
9 /*******************************************************
10 ** 函数名称 ： ConnectSer
11 ** 函数功能 ： 连接服务器
12 ** 输 入 ： 无
13 ** 输 出： 无
14 ** 备 注： 服务器IP:192.168.5.1 PORT:8010 超时连接：1s
15 ********************************************************/
16 int ConnectSer()
17 {
18 //----------------------
19 // Initialize Winsock
20 bool ret = false;
21 int error = -1;
22 int len = sizeof(int);
23 WSADATA wsaData;
24 int iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
25 if (iResult != NO_ERROR) {
26 ret = false;
27 return ret;
28 }
29 //----------------------
30 // Create a SOCKET for connecting to server
31 ConnectSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
32 if (ConnectSocket == INVALID_SOCKET) {
33 WSACleanup();
34 ret = false;
35 return ret;
36 }
37 //----------------------
38 // The sockaddr_in structure specifies the address family,
39 // IP address, and port of the server to be connected to.
40 sockaddr_in clientService;
41 clientService.sin_family = AF_INET;
42 clientService.sin_addr.s_addr = inet_addr("192.168.5.1");
43 clientService.sin_port = htons(8010);
44
45 //连接超时1秒
46 struct timeval timeout ;
47 fd_set r;
48 unsigned long ul = 1;
49 ioctlsocket(ConnectSocket, FIONBIO,&ul); //设置为非阻塞模式
50 //如果连接失败，则设置超时
51 if(-1 == connect(ConnectSocket, (SOCKADDR *) &clientService, sizeof(clientService) )){
52 FD_ZERO(&r);
53 FD_SET(ConnectSocket, &r);
54 timeout.tv_sec = 1; //连接超时1秒
55 timeout.tv_usec =0;
56 if( select(0, NULL, &r, NULL, &timeout) > 0 ){
57 getsockopt(ConnectSocket, SOL_SOCKET, SO_ERROR, (char *)&error, /*(socklen_t *)*/&len);
58 if(error == 0)
59 ret = true;
60 else
61 ret = false;
62 }
63 else
64 ret = false;
65 }
66 //如果连接成功，返回ret置1
67 else
68 ret = true;
69 ul = 0;
70 ioctlsocket(ConnectSocket, FIONBIO, &ul); //设置为阻塞模式 阻塞时间为timeout时间
71 return ret;
72 }
