TCP Socket Port Check

写了两个小程序，主要是用于linux和windows下TCP端口的检测，自带的telnet无法满足我批量检测的需要，在我眼里这类端口检测程序最为关键的是超时的限制，若端口不能却要老久才返回结果，有点不爽，在不改系统默认超时时间的前提上，引入socket的非阻塞模式达到超时限制的目的，下面是具体程序

1. windows下的tcp端口检测程序

简介：引入select模式作为超时限制

 //TCP Port Check program in windows

 #include <stdio.h>

 #include <winsock2.h>

 #pragma comment(lib,"ws2_32.lib")

 int main(int argc, char *argv[])

 {

     SOCKET sockfd;

     SOCKADDR_IN sockaddr;

     int port;

     unsigned long ip;

     WSADATA wsa;

     int timeout=,ret;

     struct timeval tv;

     struct fd_set fs;

     unsigned long ul = ;

     if(argc != )

     {

         printf("Usage: %s IP Port TimeOut\n",argv[]);

         return -;

     }

     if((ip=inet_addr(argv[]))==INADDR_NONE)

     {

         printf("IP Address Error\n");

         return -;

     }

     if((port=atoi(argv[]))==)

     {

         printf("Port Error\n");

         return -;

     }

     if((timeout=atoi(argv[]))==)

     {

         printf("Timeout Error\n");

         return -;

     }

     WSAStartup(MAKEWORD(,),&wsa);    //initialize Ws2_32.dll

     if((sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) <= )    //create a tcp socket

     {

         printf("Create socket fail!\n");

         return -;

     }

     //socket non-block mode set

     if((ret = ioctlsocket(sockfd, FIONBIO, (unsigned long*)&ul))==SOCKET_ERROR)

     {

         closesocket(sockfd);

         return -;

     }

     sockaddr.sin_family = AF_INET;

     sockaddr.sin_port = htons(port);

     sockaddr.sin_addr.S_un.S_addr = inet_addr(argv[]);

     //connect to target ip with port

     connect(sockfd, (SOCKADDR *)&sockaddr, sizeof(sockaddr));

     tv.tv_sec = timeout;

     tv.tv_usec = ;

     FD_ZERO(&fs);

     FD_SET(sockfd,&fs);

     ret = select(sockfd+,NULL,&fs,NULL,&tv);

     if(ret<)

     {

         printf("Select Error\n");

         closesocket(sockfd);

         return -;

     }

     else if(ret == )

     {

         printf(" %s %d Connect fail!\n",argv[],port);

         closesocket(sockfd);

         return -;

     }

     else

     {

         printf(" %s %d Connected success!\n",argv[],port);

         closesocket(sockfd);

     }

     WSACleanup(); //clean up Ws2_32.dll

     return ;

 }

2. Linux下TCP端口检测程序

#include <stdlib.h>

#include <sys/socket.h>

#include <sys/types.h>

#include <unistd.h>

#include <stdio.h>

#include <string.h>

#include <netdb.h>

#include <time.h>

#include <errno.h>

int main(int argc, char *argv[])

{

        int sockfd=socket(AF_INET,SOCK_STREAM,);

        struct sockaddr_in client;

        char ip[];

        int port;

        int timeout;

        struct timeval tv;

        if(argc != )

        {

                printf("Usage : %s ip port timeout\n", argv[]);

                exit();

        }

        strcpy(ip, argv[]);

        port = atoi(argv[]);

        timeout = atoi(argv[]);

        bzero(&client, sizeof(struct sockaddr_in));

        client.sin_family = AF_INET;

        client.sin_addr.s_addr = inet_addr(ip);

        client.sin_port = htons(port);

        tv.tv_sec = ;

        tv.tv_usec = ;

        //set timeout with setsockopt

        if(setsockopt(sockfd,SOL_SOCKET,SO_RCVTIMEO,(char*)&tv, sizeof(tv))<)

        {

            perror("setsockopt failed\n");

            exit();

        }

        if(setsockopt(sockfd,SOL_SOCKET,SO_SNDTIMEO,(char*)&tv, sizeof(tv))<)

        {

            perror("setsockopt failed\n");

            exit();

        }

        if(!connect(sockfd,(struct sockaddr*)&client,sizeof(struct sockaddr_in)))

        {

                printf("connect ok\n");

        }else{

                printf("connect fail\n");

        }

    close(sockfd);

        return ;

}

3. 总结

代码不难，关键是能提高效率，有时候简单的东西反而更好，继续努力

巴特西

TCP Socket Port Check

最新文章

热门文章