#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#define DATALENG 1024
#define
PORT 80 /*
웹서비스 포트 */
#define SERVERNUM 3 /*
리얼서버의 숫자 */
#define INTERVAL 5 /*
리얼서버 검사 주기(초 단위) */
int TcpSend(int sockfd,char *send_buf)
{
if (send(sockfd,
send_buf, strlen(send_buf), 0) == -1)
{
perror("TCP send");
exit(1);
}
return
1;
}
int TcpRecv(int sockfd,char *recv_buf)
{
int
recv_num;
if((recv_num=recv(sockfd,recv_buf,DATALENG,0))==-1)
{ perror("TCP
recv");
exit(1);
}
return
1;
}
int TcpConnect(char *host,int port,char
*send_buf, char *recv_buf)
{
int
sockfd;
struct hostent *he;
struct sockaddr_in their_addr;
if ((he=gethostbyname(host))
== NULL) { /* get the host info */
herror("gethostbyname");
exit(1);
}
if ((sockfd
= socket(AF_INET, SOCK_STREAM, 0)) == -1) {
perror("TCP
socket");
exit(1);
}
their_addr.sin_family
= AF_INET;
their_addr.sin_port
= htons(port);
their_addr.sin_addr
= *((struct in_addr *)he->h_addr);
bzero(&(their_addr.sin_zero),
8);
/* 리얼서버가 fail 됬으면 다음의 if
문에서 걸려서 리턴되게 된다 고칠 필요없다*/
if
(connect(sockfd, (struct sockaddr *)&their_addr,sizeof(struct
sockaddr)) == -1) {
close(sockfd);
return
0;
}
TcpSend(sockfd,send_buf);
TcpRecv(sockfd,recv_buf);
close(sockfd);
return
1;
}
/***** MAIN START ****/
int main()
{
char send_buf[100], recv_buf[DATALENG];
char OffCommand[SERVERNUM][100];
char OnCommand[SERVERNUM][100];
char addr[SERVERNUM][15];
int ServerOn[SERVERNUM],temp,i;
/* 웹서버에게 페이지를 요청하는 문장이다
GET /index.html 은 요청하는 페이지의이름이다. 다른
이름과 경로로 바꾸어 적어주어도 무방하다.
User-Agent:
는 요청프로그램을 적어주는 부분으로 적당히 적어주면
된다.
필자의 경우 이 프로그램을 MMD 프로젝트의
코드에서 따온 것이므로 미니 MMD 라고 적었다.
Host:
부분은 요청하는 머신의 이름이다 일반적으로 로드밸선서의
별명을 적으면 된다*/
strcpy(send_buf,"GET /index.html
HTTP/1.0＼nUser-Agent: MiniMMD＼nHost: medusa＼nAccept:
*/*＼n＼n");
/* 리얼서버에 대한 정보를 하나씩 적어준다.
addr 은 리얼서버의 이름이며, OffCommand 는 리얼서버가
응답이 없을 때 수행해줄 명령이다.
이 명령은 210.123.54.230:80
으로 들어온 웹서비스 요청을 192.168.1.1:80 로 전송하지
않게 해준다.
OnCommand 는 리얼서버가 다시 살아나면
수행해줄 명령이다.*/
/* real server 192.168.1.1
*/
strcpy(addr[0],"node1");
strcpy(OffCommand[0],"/sbin/ipvsadm
-d -t 210.123.54.230:80 -R 192.168.1.1:80");
strcpy(OnCommand[0],"/sbin/ipvsadm -a -t 210.123.54.230:80
-R 192.168.1.1:80");
ServerOn[0]=1;
/* real server 192.168.1.3 */
strcpy(addr[1],"node3");
strcpy(OffCommand[1],"/sbin/ipvsadm
-d -t 210.123.54.230:80 -R 192.168.1.3:80");
strcpy(OnCommand[1],"/sbin/ipvsadm -a -t 210.123.54.230:80
-R 192.168.1.3:80");
ServerOn[1]=1;
/* real server 192.168.1.4 */
strcpy(addr[2],"node4");
strcpy(OffCommand[2],"/sbin/ipvsadm
-d -t 210.123.54.230:80 -R 192.168.1.4:80");
strcpy(OnCommand[2],"/sbin/ipvsadm -a -t 210.123.54.230:80
-R 192.168.1.4:80");
ServerOn[2]=1;
while(1) {
/*
모든 리얼서버에 대해서 용청을 보내고 응답을 받기 위한
시도를 한다 */
for(i=0; i<SERVERNUM
; i++) {
printf("%s
checking.... ",addr[i]);
fflush(stdout);
/* send_buf 의 내용(웹페이지
요청)을 주어진 주소(addr[i] port) 로 보내고 응답을
recv_buf
에 받는다 */
temp=TcpConnect(addr[i],PORT,send_buf,recv_buf);
if(temp == 0) /*
real server fail */
{
printf(" Fail＼n");
fflush(stdout);
if(ServerOn[i]
== 1)
{
ServerOn[i]=temp;
/*
리얼서버가 죽었으면 더이상 리얼서버로 웹페이지 요청을
보내지 않도록 명령을
수행한다
*/
system(OffCommand[i]);
}
}
else
{ printf("
OK!!＼n");
fflush(stdout);
if(ServerOn[i] == 0)
{ ServerOn[i]=temp;
/* 리얼서버가 살아났으면 리얼서버로 웹페이지 요청을
다시 보내도록 명령을 수행한다 */
system(OnCommand[i]);
}
}
}
/* 적당한 시간마다 리얼서버를 체크한다 */
sleep(INTERVAL);
}
return 0;
} |
