simple-TCP-client-.zip_visual c

# pragma comment(lib, "ws2_32.lib") #include <STDIO.H> #include <WINSOCK2.H> SOCKET g_sockClient = INVALID_SOCKET; //提示程序的使用方法 void usage(); //捕获关闭信息 BOOL WINAPI CtrlHandler(DWORD dwEvent); int main(int argc, char* argv[]) { unsigned long destAddr; //目标地址 int nPort; //目标端口 //只有两个参数，即没有给出端口号 if (argc == 2){ destAddr = inet_addr(argv[1]);//获取要连接的目标主机的IP地址 if (destAddr == INADDR_NONE){ usage(); //不是合法的IP地址，则显示程序的使用方法 return -1; } nPort = 23;//取得IP地址之后，默认目标端口号为23，即telnet端口 } else //给出了目标IP地址和目标TCP端口号 if (argc ==3){ destAddr = inet_addr(argv[1]); //获取要连接的目标主机的IP地址 if (destAddr == INADDR_NONE){ usage(); //不是合法的IP地址，则提示程序的使用方法 return -1; } nPort = atoi(argv[2]);//获取目标端口号 if (nPort<=0 || nPort > 65535){ usage();//端口非法，提示用户本程序正确的使用方法 return -1; } } else{ //没有给出参数，则提示用户本程序正确的使用方法 usage(); return -1; } //以上步骤之后，获得了目标IP，和目标TCP端口 //设置钩子，捕获控制台程序的关闭消息 if (!SetConsoleCtrlHandler(CtrlHandler, TRUE)){ printf("SetConsoleCtrlHandler: %d\n", GetLastError()); return -1; } //初始化Winsock WSADATA wsaData; WSAStartup(WINSOCK_VERSION, &wsaData); //创建SOCK_STREAM类型的套接口 g_sockClient = socket(AF_INET, SOCK_STREAM, 0); //如果创建失败，则处理 if (g_sockClient == INVALID_SOCKET){ WSACleanup(); return -1; } //定义sockaddr_in类型的结构变量to struct sockaddr_in to; //初始化to memset(&to, 0, sizeof(to)); //给to的成员变量赋值 to.sin_addr.s_addr = destAddr;//IP地址 to.sin_family = AF_INET; //地址族 to.sin_port = htons(nPort);//端口号 //显示用户正要连接的IP和端口号 printf("connecting %s: %d ...... ", inet_ntoa(to.sin_addr), nPort); //如果connect连接不成功，处理 if (connect(g_sockClient, (struct sockaddr *) &to, sizeof(to)) == SOCKET_ERROR){ if (g_sockClient != INVALID_SOCKET) closesocket(g_sockClient); //关闭套接字g_sockClient printf("Failed. (connect %d) \n", WSAGetLastError()); WSACleanup(); return -1; } else printf("Successfully. \n");//提示连接成功 //定义83字节的数组，用于发送数据 char bufa[83]; //定义1000字节的数组，用于接收服务器返回的信息 char bufb[1000]; //为了使用select检查是否有数据到来，定义端口描述字集合 fd_set readSet; //定义select函数的超时值， timeval类型结构体tv struct timeval tv; int ret, len; //不断的发送数据、接收数据 while (1){ //初始化发送数据缓存 memset(bufa, 0, 83); //读取用户输入的一行信息到bufa gets(bufa); //最多取输入的前80个字符 len = strlen(bufa); if (len > 80) len = 80; bufa[len] = '\r'; bufa[len+1] = '\n'; bufa[len+2] = 0; //发送数据 ret = send(g_sockClient, bufa, strlen(bufa), 0); //如果发送失败，处理 if(ret == SOCKET_ERROR){ printf("send: %d\n", WSAGetLastError()); break; } //初始化readSet FD_ZERO(&readSet); //将套接字g_sockClient加入readSet FD_SET(g_sockClient, &readSet); tv.tv_sec = 3;//3秒 tv.tv_usec = 0;//毫秒 最长等待时间为3秒+0毫秒 //调用select，有数据可读或等待超时返回 ret = select(0, &readSet, NULL, NULL, &tv); //如果调用select结果返回出错 if (ret == SOCKET_ERROR){ printf("selcet: %d \n", WSAGetLastError()); break; } //如果select等待超时： if (ret = 0){ printf("Timeout, No Response From Server. \n"); break; } //如果g_sockClient在readSet里面，即g_sockClient这时候可读 if (FD_ISSET(g_sockClient, &readSet)){ //准备接收数据 memset(bufb, 0, 1000); //接收数据 ret = recv(g_sockClient, bufb, 1000, 0); //接收数据失败，则提示 if (ret == SOCKET_ERROR){ printf("recv: %d \n", WSAGetLastError()); break; } else{ printf("%s", bufb);//接收成功，则显示接收到的数据 } } } //关闭g_sockClient if (g_sockClient != INVALID_SOCKET) closesocket(g_sockClient); //卸载 WSACleanup(); printf("Stopped. \n"); return 0; } void usage() { printf("usage: \tmyelnet x.x.x.x port \t\t\t(0< port < 65535)\n"); } BOOL WINAPI CtrlHandler(DWORD dwEvent) { switch (dwEvent){ case CTRL_C_EVENT: //用户按下CTRL+C case CTRL_LOGOFF_EVENT://当前用户注销 case CTRL_SHUTDOWN_EVENT://系统关闭 case CTRL_CLOSE_EVENT://试图关闭控制台程序 //出现以上情况，均提示停止并做清理工作 printf("Stopping ...... \n"); closesocket(g_sockClient);//关闭套接字 g_sockClient = INVALID_SOCKET; break; default: return FALSE; } return TRUE; }