Laguna of Cakel

 
// 저작권
// http://www.sockaddr.com/ExampleSourceCode.html
// http://cakel.tistory.com
// 교육용을 전제로 자유롭게 쓰실수 있습니다.
// 
// ping.h - RFC 규격에 맞게 ICMP 와 IP 헤더를 지정하는 파일입니다.
// 


// packing 용량 단위 설정하는 부분입니다. alignment 를 1 로 두어서 최소 자료 전송을 노립니다.
// 대신에 접근하는 속도가 떨어지므로 좋은 방법만은 아닙니다.
#pragma pack(1)

#define ICMP_ECHOREPLAY 0
#define ICMP_ECHOREQ	8

// IP 헤더 부분 -- RFC 791
// http://www.faqs.org/rfcs/rfc791.html
// http://blog.naver.com/after1007/50014330586 참고

typedef struct tagIPHDR
{
	u_char	VIHL;			// 버전과 IHL(Internet Header Length) 값
	u_char	TOS;			// Type of Service (서비스의 형태)
	short	TotLen;			// Total Length (전체 길이)
	short	ID;				// Identification (식별자)
	short	FlagOff;		// Fragment Offset (단편 좌표)
	u_char	TTL;			// Time to Live (생존 시간)
	u_char	Protocol;		// Protocol (프토토콜, 전송 규약)
	u_short	Checksum;		// 체크섬(값 비교)
	struct	in_addr iaSrc;	// Internet Address - Source (인터넷 주소 - 원본)
	struct	in_addr	iaDsc;	// Internet Address - Destination (인터넷 주소 - 대상)

}	IPHDR, *PIPHDR;

// ICMP 헤더 -- RFC 792
// http://www.faqs.org/rfcs/rfc792.html
// http://www.microsoft.com/korea/technet/deploy/tcpintro5.asp
// http://technet2.microsoft.com/WindowsServer/ko/Library/732438fe-70c5-4e68-9663-ecbd955d29ea1042.mspx?mfr=true
// http://blog.naver.com/isnta/20024953151 참고

typedef struct tagICMPHDR
{
	u_char	Type;			// 형식
	u_char	Code;			// 코드
	u_short	Checksum;		// 체크섬
	u_short	ID;				// 식별자
	u_short	Seq;			// 순서
	char	Data;			// 자료(데이터)

}	ICMPHDR, *PICMPHDR;

#define REQ_DATASIZE 32		// 에코(Echo) 요구 자료 길이

// ICMP Echo Request - 에코 자료 요구
typedef struct tagECHOREQUEST
{
	ICMPHDR	icmpHdr;
	DWORD	dwTime;
	char	cData[REQ_DATASIZE];
}	ECHOREQUEST, *PECHOREQUEST;

// ICMP Echo Reply			// 에코 응답
typedef struct tagECHOREPLY
{
	IPHDR		ipHdr;
	ECHOREQUEST	echoRequest;
	char		cFiller[256];
}	ECHOREPLY, *PECHOREPLY;

#pragma pack()		// packing 용량 단위 기본값으로 변경합니다.
 

 
//
//	저작권
//	http://www.sockaddr.com/ExampleSourceCode.html
//	http://cakel.tistory.com
//	교육용을 전제로 자유롭게 쓰실수 있습니다.
//
//	ping.cpp  -- ICMP 와 RAW 소켓을 사용하는 Ping 프로그램입니다.
//

#include <stdio.h>
#include <stdlib.h>
#include <winsock.h>
#include "ping.h"

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

// 내부 함수
void Ping(LPCSTR pstrHost);
void ReportError(LPCSTR pstrFrom);
int WaitForEchoReply(SOCKET s);
u_short in_cksum(u_short *addr, int Len);

// ICMP 에코 요구/응답 함수
int SendEchoRequest(SOCKET, LPSOCKADDR_IN);
DWORD RecvEchoReply(SOCKET, LPSOCKADDR_IN, u_char*);

// 시작
int main(int argc, char* *argv)
{
	WSADATA wsaData;
	WORD wVersionRequested = MAKEWORD(1,1);
	int nRet;

	// 인수 비교
	if(argc != 2)
	{
		fprintf(stderr,"\n사용법 : ping [호스트명]\n");
		return 1;

	}

	// 윈속 시작
	nRet = WSAStartup(wVersionRequested, &wsaData);
	if(nRet)
	{
		fprintf(stderr,"\n윈속을 시작하는데 오류 발생했습니다.\n");
		return 1;

	}
	
	// ping 작업을 합니다.
	Ping(argv[1]);

	// 윈속을 비웁니다.
	WSACleanup();

	return 0;
}

// Ping() 함수
// SendEchoRequest() 함수와 RecvEchoReply() 함수를 수행하고
// 결과를 출력합니다.

void Ping(LPCSTR pstrHost)
{
	SOCKET		rawSocket;
	LPHOSTENT	lpHost;
	struct		sockaddr_in saDest;
	struct		sockaddr_in saSrc;
	DWORD		dwTimeSent;
	DWORD		dwElapsed;
	u_char		cTTL;
	int			nLoop;
	int			nRet;

	// Raw 소켓을 생성합니다.
	rawSocket = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP);
	if(rawSocket == SOCKET_ERROR)
	{
		ReportError("socket() 함수 문제 발생");
		return;

	}

	// 호스트를 찾아 봅니다(Lookup host)
	lpHost = gethostbyname(pstrHost);
	if(lpHost == NULL)
	{
		fprintf(stderr, "\n호스트를 찾을수가 없습니다 : %s\n", pstrHost);
		return;

	}
	
	// 목적지 소켓 주소로 연결을 시도 합니다.
	saDest.sin_addr.s_addr = *((u_long FAR *)(lpHost->h_addr));
	saDest.sin_family = AF_INET;
	saDest.sin_port = 0;

	// 사용자에게 프로그램이 할려고 하는 작업을 표시 합니다.
	printf("\n%s [%s] 호스트에게 %d 바이트의 자료를 핑합니다.\n",
		pstrHost,
		inet_ntoa(saDest.sin_addr),
		REQ_DATASIZE);

	// 몇번(4) 핑 작업을 합니다.
	for(nLoop = 0; nLoop < 4; nLoop++)
	{
		// ICMP 에코 요구(Echo request)를 보냅니다.
		SendEchoRequest(rawSocket, &saDest);

		// select() 함수로 자료가 받을때 까지 기다립니다.
		nRet = WaitForEchoReply(rawSocket);
		if(nRet == SOCKET_ERROR)
		{
			ReportError("select() 함수에서 문제 발생");
			break;

		}

		if(!nRet)
		{
			printf("\n시간 초과");
			break;
		}
		
		// 응답을 받습니다.
		dwTimeSent = RecvEchoReply(rawSocket, &saSrc, &cTTL);

		// 소요된 시간을 계산합니다.
		dwElapsed = GetTickCount() - dwTimeSent;
		printf("\n%s 에서 응답을 받았습니다. 바이트=%d 시간 %ldms TTL=%d",
			inet_ntoa(saSrc.sin_addr), REQ_DATASIZE,
			dwElapsed, cTTL);
		}
	printf("\n");
	nRet = closesocket(rawSocket);
	if(nRet == SOCKET_ERROR)
		ReportError("closesocket() 함수 수행중 오류 발생.");

}

// SendEchoRequest()
// 에고 요구 헤더 정보를 채우고 목적지로 보냅니다.
int SendEchoRequest(SOCKET s, LPSOCKADDR_IN lpstToAddr)
{
	static ECHOREQUEST echoReq;
	static nId = 1;
	static nSeq = 1;
	int nRet;

	// 에코 요구 정보를 채웁니다.
	echoReq.icmpHdr.Type		= ICMP_ECHOREQ;
	echoReq.icmpHdr.Code		= 0;
	echoReq.icmpHdr.Checksum	= 0;
	echoReq.icmpHdr.ID			= nId++;
	echoReq.icmpHdr.Seq			= nSeq++;

	// 보낼 정보를 채웁니다.
	for(nRet = 0; nRet < REQ_DATASIZE; nRet++)
		echoReq.cData[nRet] = ' '+nRet;

	// 보낼 때 틱 횟수(tick count, 시간) 저장합니다.
	echoReq.dwTime = GetTickCount();

	// 보낼 자료를 패킷에다 넣고 체크섬(비교값)을 계산합니다.
	echoReq.icmpHdr.Checksum = in_cksum( (u_short*)&echoReq, sizeof(ECHOREQUEST) );

	// 에코 요구 신호를 보냅니다.
	// sendto(소켓, 전송할 데이터, 데이터의 길이, 역할 나타내는 플래그, 원격 호스트 주소, 원격 호스트 주소의 크기)
	// 소켓이 접속의 유무와 관계 없이 일방적으로 전송하는 함수 입니다.
	// http://white.chungbuk.ac.kr/~jchern/sendto.html 참고

	nRet = sendto(s,					// 소켓
				(LPSTR)&echoReq,		// 버퍼
				sizeof(ECHOREQUEST),	// 버퍼 크기
				0,						// 플래그
				(LPSOCKADDR)lpstToAddr,	// 목적지
				sizeof(SOCKADDR_IN));	// 주소 길이

	if(nRet == SOCKET_ERROR)
		ReportError("sendto() 함수 수행중 문제가 발생했습니다.");

	return (nRet);

}

// RecvEchoReply() 함수
// 들어오는 자료를 받고 받은 항목들을 분석합니다.
DWORD RecvEchoReply(SOCKET s, LPSOCKADDR_IN lpsaFrom, u_char* pTTL)
{
	ECHOREPLY echoReply;
	int nRet;
	int nAddrLen = sizeof(struct sockaddr_in);

	// 에코 응답을 받습니다.
	// recvform(소켓 식별자, 수신된 데이터를 받을 버퍼, 버퍼의 길이, 역할을 나타내는 플래그, 준 곳의 주소를
	// 리턴 받기 위한 버퍼 포인터, 버퍼 사이즈를 지정한 포인터)
	// 소켓으로 부터 데이터그램을 받고, 송신자의 주소와 포트를 반환하는 함수 입니다.
	// http://white.chungbuk.ac.kr/~jchern/recvfrom.html 참고
	nRet = recvfrom(s,					// 소켓
				(LPSTR)&echoReply,		// 버퍼
				sizeof(ECHOREPLY),		// 버퍼의 크기
				0,						// 플래그
				(LPSOCKADDR)lpsaFrom,	// 받는 주소
				&nAddrLen);				// 주소 길의를 담는 자료를 가리키는 포인터

	// 되돌아 오는 값을 검사합니다.
	if(nRet == SOCKET_ERROR)
		ReportError("recvfrom() 함수 수행중 문제가 발생했습니다.");

	// 보낸 시간 값과 IP TTL 값을 받습니다.
	*pTTL = echoReply.ipHdr.TTL;
	return(echoReply.echoRequest.dwTime);

}

// ReportError() 함수
// 무슨 일 있습니까?
void ReportError(LPCSTR pWhere)
{
	fprintf(stderr,"\n%s WSAGetLastError 오류 코드 번호 : %d\n", WSAGetLastError());
}

// WaitforEchoReply() 함수
// select() 함수를 사용하여 언제 자료가 읽혀 질지 결정합니다.
int WaitForEchoReply(SOCKET s)
{
	struct timeval Timeout;
	fd_set readfds;

	readfds.fd_count = 1;
	readfds.fd_array[0] = s;
	Timeout.tv_sec = 5;
	Timeout.tv_usec = 0;

	// select(I/O 변화를 감지하는 소켓의 갯수, 읽기 상태 변화 감지할 소켓, 쓰기 상태 변화 감지할 소켓
	// 예외 상태 변화 감지할 소켓, 기다리기 위한 시간)
	// http://white.chungbuk.ac.kr/~jchern/select.html 참고
	return( select(1, &readfds, NULL, NULL, &Timeout) );

}

// 마이크 무유스(Mike Muuss) 의 in_cksum() 함수
// 원래 핑(ping) 코드와 그의 설명입니다.
// 참고 http://myhome.shinbiro.com/~eaglelee/rawsocket.txt
//
//* 저자 -
//* 마이크 무유스(Mike Muuss)
//* 미 육군 탄도 연구소(U. S. Army Ballistic Research Laboratory)
//* 1983년 12월
/*
 *		I N _ C K S U M
 *
 *
 *	인터넷 프로토콜 계열 헤더를 위한 체크섬 계산 루틴(C 버전)
 *

*/

u_short in_cksum(u_short *addr, int len)
{
	register int nleft = len;
	register u_short *w = addr;
	register u_short answer;
	register int sum = 0;

	/*
	 * 우리의 원래 알고리듬은 간단합니다. 32 비트 누산기(덧셈)를 써서,
	 * 우리는 16비트 단어를 순차적으로 더한 후 마지막에,
	 * 상위 16비트를 뒤로 접어 하위 16비트로 내립니다.
	 *
	 */

	while(nleft > 1){
		sum += *w++;
		nleft -= 2;

	}

	/* 필요하다면, 홀수 바이트 부분을 정리 합니다. */
	if(nleft == 1) {
		u_short u = 0;
		*(u_char*)(&u) = *(u_char*)w;
		sum += u;

	}

	/*
	 *	상위 16비트를 하위 16비트로 더합니다.
	 */

	sum = (sum >> 16) + (sum &0xffff); /* 상위 16비트를 하위 16비트로 더하기 */
	sum += (sum >> 16);					/* 자리 올림수를 더합니다. */
	answer = ~sum;						/* 16 비트로 자릅니다. */
	return (answer);

}
 

/*
    whois.c
    (c) feb 2000 by Gandalf
    다시 치기 http://cakel.tistory.com

*/

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <windows.h>
#include <winsock.h>

#define WIN32_LEAN_AND_MEAN    // define win32 only

void handle_error(void);
void write_file(char* buf);

int main(int argc, char** argv)
{
    WORD wVersionRequested;            // 소켓 dll 버전 정보
    WSADATA wsaData;                // 소켓 lib 생성화를 위한 자료
    SOCKET sock;                    // 소켓 정보
    const int BUF_LEN = 10000;        // 전송을 위한 크기 설정
    struct sockaddr_in address;        // 소켓 주소
    struct hostent* host;            // 호스트
    int err;                        // 문제 코드 잡기
    float socklib_ver;                // 소켓 DLL 버전
    char File_Buf[BUF_LEN];            // 파일 버퍼
    char DomainName[100];            // 사용자 도메인 이름
    char HostName[100];                // 사용자 호스트 이름
    time_t now;                        // 날짜와 시간

    if (argc < 2)                    // 실행 옵션을 가지고 있는지 확인
    {
            printf("\n사용법 : whois 도메인명 [whois.server] \n");
            exit(0);

    }

    strcpy(DomainName, argv[1]);    // 실행줄에서 도메인 이름 얻기
    strcat(DomainName, "\r\n");        // 서버가 예상하는  Carrage Return , Line Feed 를 넣어 준다.

    if (argc == 3)
            strcpy(HostName, argv[2]);    // 명령줄에서 호스트명을 얻습니다.
    else
            strcpy(HostName, "whois.internic.net");

    wVersionRequested = MAKEWORD(1, 1);

    // 만약 Winsock dll 호출을 해야한다면 WSAStartup 루틴을 불러와야 합니다.

    if( WSAStartup( wVersionRequested, &wsaData ) != 0 )
        handle_error();

    // 소켓 DLL 이 1.1 또는 이상인지 확인 합니다.
    socklib_ver = (float)(HIBYTE( wsaData.wVersion ) / 10.0);
    socklib_ver += LOBYTE( wsaData.wVersion );

    if ( socklib_ver < 1.1 )
    {
        printf("\n오류 : 소켓 라이브러리 버전이 1.1 이상이여야 합니다.\n");
        WSACleanup(); // 나가기 전에 정리
        exit(0);

    }

    // 현재 시간과 날짜를 화면에 기록합니다

    time(&now);
    sprintf(File_Buf, "Whois 세션 시작 %.24s.\n\n", ctime(&now));
    write_file(File_Buf);

/*
    소켓을 엽니다. AF_INET 파라메터는 윈도우즈에게 우리가 인터넷 요구한다는 것을
    알려줍니다. 다른 파라메터는 서로 다른 네트워크가  선택되었을 때를 위함입니다.
    예) NetBIOS, IPX 등. SOCK_STREAM 파라메터는 윈도우즈에게 TCP 를 UDP 보다더 원한다
    는 것을 알려줍니다. 그리고 마지막 파라메터는 항상 0 으로 정해서 윈도우즈에게
    기본적으로 전송할 프로토콜이 어떤건지 알려 줍니다 (예 PPP, IP)

*/

    if( sock = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP ) == INVALID_SOCKET )
        handle_error();

/*
    우리는 이제 주소 구조에 있는 몇몇 변수들을 생성해야 합니다. 다시 한번 윈도우즈에게
    우리는 인터넷을 쓰고 우리가 원격지에 있는 컴퓨터와 접속시 쓸 포트 번호를 말해줍니다.
    이번 경우 43 포트가 쓰이는데 이는 whois 를 위한 표준 포트 입니다. htons 루틴은
    인텔 계열 칩의 메모리 저장 방식을 다른 컴퓨터들과 맞게 하기 위해서 변환하는데 쓰입니다.
    이 표준은 모토로라 형식에 근간합니다.

*/

    address.sin_family = AF_INET;        // 인터넷
    address.sin_port = htons(43);        // whois 를 위한 포트 43

    // 로그 파일과 화면에 기록

    sprintf(File_Buf, "%s 에 접속합니다\n", HostName);
    write_file(File_Buf);

/*
    host 는 hostent 형으로 미리 정의된 구조체의 포인터입니다. 우리는 gethostbyname 에
    DNS 주소를 써서 hostent 구조체의 주소를 받습니다. 이는 우리가 IP 주소를 ISP 네임서버
    에서 받아올 수 있게 하기 위해서 입니다.

*/
    if( (host = gethostbyname(HostName)) == NULL )
        handle_error();

    if( (connect(sock, (struct sockaddr*)&address, sizeof(address)) ) < 0)
            handle_error();

    // 우리는 받은 IP 주소로 주소 구조체를 시작합니다.

    address.sin_addr.s_addr = *((unsigned long*) host->h_addr_list);

    // 이제 우리는 실제 whois 서버 자체로 접속할 준비가 되었습니다.

/*    우리는 이 시점에서 whois 서버에 접속을 해야 합니다.  그리고 domain 이름을 보내고
    응답을 기다립니다. send 와 recv 루틴이 항상 TCP 와 함께 사용됩니다. 이들은 헨드셰이킹
    을 하는데 sendto 와 recvfrom 루틴을 쓰는 UDP 프로토콜(헨드세이킹을 하지 않습니다)
    과 같습니다.
   
*/

    strcpy(File_Buf, DomainName);
    err = send(sock, File_Buf, strlen(File_Buf), 0);    // 도메인 이름을 보냅니다.
    err = recv(sock, File_Buf, BUF_LEN, 0);                // 첫번째 응답을 버립니다.
    err = recv(sock, File_Buf, BUF_LEN, 0);                // 돌아오는 질의의 결과를 받습니다.
    write_file(File_Buf);

    // 항상 나가기 전에 WSACleanup 을 수행합니다

    WSACleanup();        // 나가기 전에 청소

    exit(0);

    return 0;
}


void handle_error(void)
{

/*
    Error 는 WSAGetLastError 루틴을 써서 받아 오는 최근의 오류를 받습니다.
    이 교제 배우면서 우리는 좀더 무엇 때문에 이 오류가 나는지 배우게 될 것입니다

*/

    switch( WSAGetLastError() )
    {
        case WSANOTINITIALISED :
            printf("소켓을 시작 할수가 없습니다. \n");
            break;

        case WSAEAFNOSUPPORT :
            printf("이 주소 계열은 지원하지 않습니다.\n");
            break;

        case WSAEADDRNOTAVAIL :
            printf("이 주소는 로컬 컴퓨터에서 할당 받을수 없습니다.\n");
            break;

        case WSAECONNREFUSED :
            printf("이 시도는 의도적으로 거절당했습니다.\n");
            break;

        case WSAEDESTADDRREQ :
            printf("목적지 주소가 필요합니다.\n");
            break;
           
        case WSAEFAULT :
            printf("namelen 인수가 틀립니다.\n");
            break;

        case WSAEINVAL :
            printf("소켓이 아직 그 주소로 가지(bound) 않았습니다.\n");
            break;

        case WSAEISCONN :
            printf("소켓이 이미 연결 되었습니다.\n");
            break;

        case WSAEADDRINUSE :
            printf("그 주소는 이미 사용 중입니다.\n");
            break;

        case WSAEMFILE :
            printf("더 이상 기술자 파일이 존재 하지 않습니다\n");
            break;

        case WSAENOBUFS :
            printf("더 이상 버퍼 공간이 있지 않습니다. 소켓이 만들어 질수 없습니다.\n");
            break;

        case WSAEPROTONOSUPPORT :
            printf("명시한 프로토콜은 지원하지 않습니다.\n");
            break;

        case WSAEPROTOTYPE :
            printf("명시한 프로토콜은 이 소켓과 맞지 않은 형식입니다.\n");
            break;

        case WSAENETUNREACH :
            printf("지금 시간에는 이 호스트에서 네트워크에 접근 할수가 없습니다.\n");
            break;

        case WSAENOTSOCK :
            printf("기술자가 소켓이 아닙니다.\n");
            break;

        case WSAETIMEDOUT :
            printf("접속이 되지 않은 시도의 시간이 초과되었습니다.\n");
            break;

        case WSAENETDOWN :
            printf("네트워크 하부시스템이 망가졌습니다.\n");
            break;

        case WSAHOST_NOT_FOUND:
            printf("인증 응답 호스트가 발견되지 않았습니다.\n");
            break;

        case WSATRY_AGAIN :
            printf("비-인증 호스트가 없거나 서버가 실패(SERVERFAIL)했습니다.\n");
            break;;

        case WSANO_RECOVERY :
            printf("되돌릴수 없는 에러 입니다. FORMERR, REFUSED, NOTIMP.\n");
            break;
   
        case WSANO_DATA :
            printf("유요한 이름이지만 요구된 형식의 자료가 없습니다.\n");
            break;

        case WSAEINPROGRESS :
            printf("주소 막기 윈도우즈 소켓 작업이 진행 중입니다.\n");
            break;

        case WSAEINTR :
            printf("(막힌) 호출이 WSACancelBlockingCall() 함수에 의해 취소 되었습니다.\n");
            break;

        default :
            printf("알수 없는 오류 입니다.\n");
            break;

    }
            WSACleanup();
            exit(0);

}

void write_file(char *buf)
{
    // 화면과 파일에 동시에 출력합니다.
    FILE *fp = fopen("whoislog.txt", "a+");
    fprintf(fp, "%s\n", buf);
    fclose(fp);
    printf("%s\n",buf);
}