NMS using ucd snmp

저도 같은 프로젝트하는데 ^^

#include "ping.h"

char send_buf[1500];			//보내는 버퍼
char recv_buf[1500];			//받는 버퍼
int g_id = 0;					//getpid()값 저장. 내가 보낸 ping만 받기위해
struct sockaddr_in dest_addr;	//목적지 소켓구조체
struct sockaddr_in src_addr;	//내 소켓구조체
pthread_t p_thread[2];			//쓰레드 id, [0]번 send, [1]번 recv

unsigned short icmp_cksum(u_short *addr, int len)		//ip체크섬 무시 -_-;
{
	register int nleft = len;
	register u_short *w = addr;
	register int sum = 0;
	u_short answer = 0;

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

	if(nleft==1){
		*(u_char *)(&answer) = *(u_char *)w;
		sum == answer;
	}

	sum = (sum>>16) + (sum&0xffff);
	sum += (sum>>16);
	answer = ~sum;
	return(answer);
}

void sig_alrm()			//ping 보내기를 마친지 sleep(x)초 후에 호출됨
{
	pthread_detach(p_thread[1]);		//[1](recv) 쓰레드를 죽인다
}

int init()		//ping 전체모듈을 준비하는 부분
{
	int i;
	connect_db();		//db 연결

	g_id = getpid();	//내가 보낸 ping을 확인하기위해 getpid() 저장
	init_ping_node();	//ping_node 생성
	get_addr();			//db에서 ip를 읽어오는 부분

	for(i=0; i<9; i++)
		insert_ping_node(ipaddr[i]);	//읽어온 ip 저장

	signal(SIGALRM, &sig_alrm);			//signal 선언

	memset(&dest_addr, 0, sizeof(struct sockaddr_in));		//목적지 구조체 초기화

	dest_addr.sin_family = AF_INET;
	dest_addr.sin_port = 0;

	return 0;
}

void init_ping_node(void)		//ping 노드 초기화하는 부분
{
	ping_head = (PING_NODE *)malloc(sizeof(PING_NODE));
	ping_tail = (PING_NODE *)malloc(sizeof(PING_NODE));
	ping_head->next = ping_tail;
	ping_head->prev = ping_head;
	ping_tail->next = ping_tail;
	ping_tail->prev = ping_head;
}

PING_NODE *insert_ping_node(char str[])		//새로운 노드 삽입
{
	PING_NODE *s;

	s = (PING_NODE *)malloc(sizeof(PING_NODE));

	strcpy(s->addr, str);		//str로 넘어온놈 ip에 저장
	s->check = 'N';				//기본 셋팅 N

	//새로운 노드는 가장 뒤에 붙인다.
	ping_tail->prev->next = s;
	s->prev = ping_tail->prev;
	s->next = ping_tail;
	ping_tail->prev = s;

	return s;
}

void *send_ping(void *arg)		//ping 보내는 쓰레드, 인자값으로 보내는 소켓을 받는다
{
	struct icmphdr *icmp;
	int sockfd = *((int *)arg);		//보내는 소켓을 sockfd에 저장
	static int g_seq = 0;			//icmp 시퀀스 넘버
	PING_NODE *s;

	for(s=ping_head->next; s!=ping_tail; s=s->next)		//head 다음부터 tail전까지 모든 ip에 핑을 보낸다
	{
		icmp = (struct icmphdr *)send_buf;
		icmp->type = ICMP_ECHO;
		icmp->code = 0;
		icmp->un.echo.id = g_id;				//내 pid
		icmp->un.echo.sequence = g_seq++;
		icmp->checksum = 0;
		icmp->checksum = icmp_cksum((unsigned short *)icmp, 16);		//체크섬

		dest_addr.sin_addr.s_addr = inet_addr(s->addr);			//목적지 ip가 계속 바뀐다
		if(sendto(sockfd, send_buf, 16, 0, (struct sockaddr *)&dest_addr, sizeof(dest_addr))<=0)
			perror("sendto: ");
	}

	sleep(1);		//ping를 다 send 한후 1초 기다린다
	sig_alrm();		//signal 호출
}

void recv_icmp(char *buf, int totlen)	//받은 ping 을 처리하는 부분
{
	int iplen, icmplen;
	struct iphdr *ip;
	struct icmphdr *icmp;
	int count=0;
	PING_NODE *s;

	s=ping_head->next;

	ip = (struct iphdr *)buf;
	iplen = ip->ihl*4;
	icmp = (struct icmphdr *)(buf + iplen);

	if(icmp->type == ICMP_ECHOREPLY){	//ping에 대한 reply 인가?
		if(icmp->un.echo.id == g_id)	//identified가 내 pid와 같은가?
		{
			for(count=0; count<icmp->un.echo.sequence; count++)	//시퀀스만큼 이동
				s=s->next;
			
			if(s!=ping_tail)	//찾은 부분이 끝이 아니면 check를 Y로 체크해서 살아있다고 표시
				s->check = 'Y';
		}
	}
}

void *recv_ping(void *arg)		//ping을 받는 쓰레드, 인자값으로 받는 소켓을 받는다
{
	int len, sz;
	int sockfd = *((int *)arg);	//받는 소켓을 sockfd에 저장

	memset(recv_buf, '\0', sizeof(recv_buf));

	while(1)
	{
		if((recvfrom(sockfd, recv_buf, sizeof(recv_buf), 0, (struct sockaddr *)&src_addr, &len))<=0)
			perror("recvfrom: ");

		recv_icmp(recv_buf, sz);		//받은 데이터를 처리부분에 넘긴다
	}
}

int ping(void)
{
	int thr_id;
	int status;
	int sendfd, recvfd;
	
	if((sendfd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP))<0){		//ping을 보내는 소켓을 만든다
		fprintf(stderr, "ERROR: Opening raw socket. (need UID 0)\n");
		exit(0);
	}

	if((recvfd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP))<0){		//ping을 받는 소켓을 만든다
		fprintf(stderr, "ERROR: Opening raw socket. (need UID 0)\n");
		exit(0);
	}	

	init();			//전체 준비부분 실행

	thr_id = pthread_create(&p_thread[0], NULL, send_ping, (void *)&sendfd);		//ping을 보내는 쓰레드 실행
	thr_id = pthread_create(&p_thread[1], NULL, recv_ping, (void *)&recvfd);		//ping을 받는 쓰레드 실행

	//ping을 보내는 쓰레드의 종료를 처리 (detach로 죽이기 때문에 처리 안됨)
	pthread_join(p_thread[0], (void *)&status);
	pthread_join(p_thread[1], (void *)&status);		//ping을 받는 쓰레드의 종요를 처리(detach로 죽이기때문에 안됨 -_-;

}

몇군데 문제점이 있습니다.

일단 여러개 가져다가 고쳐쓰다보니 signal도 의미가 없어지고 -_-;

pthread_detach(p_thread[1]) 가 쓰레드 죽이는건줄 알았는데... 죽던말던 신경 안쓴다는거라네요.

즉 p_thread[1]은 계속 receive에서 블록되있습니다.

지금까지는 문제가 없어서... 제가 갑자기 UI로 옮겨가서 놔둔상태이지만, 발표회때 문제생기면 ㅡㅡㅋ