Windows socket source
2009. 8. 6. 18:23
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 | #include <stdio.h> #include <stdlib.h> #include <string.h> #include <winsock2.h> #define BUFSIZE 1024 #define PORT 80 void ErrorHandling( char *message); int main( int argc, char *argv[]) { WSADATA wsaData; SOCKET hSocket; char message[BUFSIZE]; int len, i, option, result = 0; SOCKADDR_IN servAddr; struct hostent *host; char *lists[] = {0,}; #if 0 if (argc!=2) { Usuage(argv[0]); } #endif #if 1 if (WSAStartup(MAKEWORD(2, 2), &wsaData) !=0) ErrorHandling( "WSAStartup() error!" ); #endif for (i=0; i<4; i++) { host = gethostbyname(lists[i]); printf ( "trying to connect %s...\n" , lists[i]); #if 1 hSocket=socket(PF_INET, SOCK_STREAM, 0); if (hSocket == INVALID_SOCKET) ErrorHandling( "socket() error" ); #endif option = TRUE; setsockopt(hSocket, SOL_SOCKET, SO_REUSEADDR, ( char *)&option, sizeof (option)); memset (&servAddr, 0, sizeof (servAddr)); servAddr.sin_family = AF_INET; servAddr.sin_addr.s_addr = inet_addr(inet_ntoa( *( struct in_addr*)host->h_addr_list[0])); servAddr.sin_port = htons(PORT); if (connect(hSocket, (SOCKADDR*)&servAddr, sizeof (servAddr)) == SOCKET_ERROR) ErrorHandling( "connect() error!" ); //sprintf(message, "GET /index.html HTTP/1.1"); sprintf (message, "hello" ); send(hSocket, message, strlen (message), 0); #if 1 len=recv(hSocket, message, BUFSIZE-1, 0); message[len] = 0; if (len == -1) { printf ( "Can`t connect %s.\n" , lists[i]); result--; } if (len > 0) { printf ( "Success.connected to %s.\n" , lists[i]); result++; } closesocket(hSocket); #endif } if (result > 2) { printf ( "This machine seems to be clean\n" ); } else if (result <2 && result >0) printf ( "Warning. This machine needs to be other inspection.\n" ); else { printf ( "Infected!. Need some help.\n" ); } WSACleanup(); return 0; } void ErrorHandling( char *message) { fputs (message, stderr); fputc ( '\n' , stderr); exit (1); } </winsock2.h></string.h></stdlib.h></stdio.h> |