select() в прокси-сервере
Я строю прокси-сервер в C, и я пытаюсь понять select() функция. У меня есть код, созданный таким образом, что соединение устанавливается с клиента, а затем веб-адрес извлекается, чтобы можно было установить другое соединение для подключения к реальному веб-серверу. Затем страница принимается прокси и передается обратно клиенту.
Я это понимаю select() позволит обрабатывать несколько клиентских запросов, но я не понимаю, как это помогает (или, скорее, как реализовать) второе подключение к веб-серверу. Из того, что я понимаю, мне больше не понадобится цикл while, чтобы продолжать получать данные с веб-сервера и передавать их обратно клиенту.
Нужен ли второй набор файловых дескрипторов для соединений с веб-сервером? Если я обрабатываю два или более клиентских запросов, как я могу убедиться, что они связаны друг с другом и правильно подключены к веб-серверу? Во всяком случае, я был бы признателен за любую помощь. Я прошел учебное пособие по сетевым технологиям и несколько других онлайн, но через пару дней я все еще не обдумал это.
void error(const char *msg)
{
perror(msg);
exit(1);
}
void *get_in_addr(struct sockaddr *sa)
{
if (sa->sa_family == AF_INET)
return &(((struct sockaddr_in*)sa)->sin_addr);
return &(((struct sockaddr_in6*)sa)->sin6_addr);
}
string getHostString(const char *buf);
int main(int argc, char *argv[])
{
int sockfd, newsockfd, portno;
int fdmax; //maximum file descriptor number
socklen_t clilen;
char buffer[256];
struct sockaddr_storage remoteaddr; //client address
char clientIP[INET6_ADDRSTRLEN];
//fd_set readfds, writefds, exceptfds;
fd_set masterfds, readfds;
struct timeval timeout;
int rc;
/*Set time limit. */
timeout.tv_sec = 3;
timeout.tv_usec = 0;
/*Create a descriptor set containing the sockets */
FD_ZERO(&readfds);
FD_ZERO(&masterfds);
/*FD_ZERO(&writefds);
FD_ZERO(&exceptfds);
FD_SET(newsockfd, &readfds);
rc = select(sizeof(readfds)*8, &readfds, NULL, NULL, &timeout);
if (rc==-1){
perror("select failed");
return -1;
}*/
struct sockaddr_in serv_addr, cli_addr;
int n;
if (argc < 2) {
fprintf(stderr,"ERROR, no port provided\n");
exit(1);
}
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0)
error("ERROR opening socket");
bzero((char *) &serv_addr, sizeof(serv_addr));
portno = atoi(argv[1]);
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = INADDR_ANY;
serv_addr.sin_port = htons(portno);
if (bind(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0)
error("ERROR on binding");
if((listen(sockfd,5)) == -1 )
error("Server-listen() error!!!");
printf("Server-listen() is OK...\n");
FD_SET(sockfd, &masterfds);
//keep track of the biggest file descriptor so far
fdmax = sockfd; //so far it's this one
for(;;){
readfds = masterfds;
if(select(fdmax+1, &readfds, NULL, NULL, NULL) == -1){
error("Server-select() error !");
}
printf("Server-select() is OK...\n");
for(int i = 0; i <= fdmax; i++){
printf("i: %d sockfd %d\n", i, sockfd);
if(FD_ISSET(i, &readfds)){
if(i == sockfd){ //sockfd is the listener
//following handles new connections
clilen = sizeof(cli_addr);
if((newsockfd = accept(sockfd,(struct sockaddr *) &cli_addr, &clilen)) == -1)
error("Server-accept() error!!!");
else{
printf("Server-accept() is OK...\n");
FD_SET(newsockfd, &masterfds); //add to master set
if(newsockfd > fdmax)
fdmax = newsockfd;
printf("selectserver: New connection from %s on "
"socket %d\n",
inet_ntop(remoteaddr.ss_family,
get_in_addr((struct sockaddr*)&cli_addr),
clientIP, INET6_ADDRSTRLEN),
newsockfd);
}
}
else{
//handle data from a client
//Here is where the FIRST read occurs
bzero(buffer,256);
//n = read(newsockfd,buffer,255);
n = read(i,buffer,255);
if (n < 0) error("ERROR reading from socket");
printf("Here is the message: \n\n%s\n\n",buffer);
//string hoststring(buffer);
string hoststring(getHostString(buffer));
int html_port = 80;
int html_socket;
printf("Prior to struct addrinfo\n");
struct addrinfo hints, *res;
memset(&hints, 0, sizeof hints);
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
hints.ai_flags = AI_PASSIVE;
printf("Prior to getaddrinfo()\n");
char *address = new char[hoststring.size() +1];
address[hoststring.size()] = 0;
memcpy(address, hoststring.c_str(), hoststring.size());
//getaddrinfo(token, (char *)html_port, &hints, &res);
getaddrinfo(address, "80", &hints, &res);
printf("We are past getaddrinfo()\n");
//if (html_socket = socket(PF_INET, SOCK_STREAM, 0) < 0){
if ((html_socket = socket(res->ai_family, res->ai_socktype, res->ai_protocol)) < 0){
printf("socket connection error\n");
}
printf("We are past socket()\n");
//char* address;
//address = new char[256];
//strncpy(address, "www.cs.ucr.edu", sizeof("www.cs.ucr.edu"));
printf("address: %s\n", address);
struct hostent * host;
if ((host = gethostbyname(address)) == NULL){
printf("Problem with gethostbyname()\n");
}
printf("We are past gethostbyname() and about to connect.\n");
if ( connect(html_socket, res->ai_addr, res->ai_addrlen) < 0)
printf("Unsuccessful completion of connect()");
int bytes_sent, bytes_recvd;
char recv_buff[1024];
bytes_sent = send(html_socket, buffer, 256, 0);
cout << "bytes_sent: " << bytes_sent << endl;
//do{
bytes_recvd = recv(html_socket, recv_buff, 1024, 0);
cout << "bytes_rcvd: " << bytes_recvd << endl;
cout << recv_buff << endl;
//FD_ZERO(&readfds);
//FD_SET(newsockfd, &writefds);
//n = write(newsockfd,"I got your message\n",20);
//n = write(newsockfd,recv_buff,bytes_recvd);
n = write(i,recv_buff,bytes_recvd);
if (n < 0) error("ERROR writing to socket");
bzero(recv_buff, 1024);
//}while(bytes_recvd !=0);
}
}
}
}
close(newsockfd);
close(sockfd);
return 0;
}
string getHostString(const char *buf){
string hoststring(buf);
hoststring = hoststring.substr(11);
cout << "hoststring: " << hoststring << endl;
int slashpos;
slashpos = hoststring.find("/");
int suffixendpos = hoststring.find("H") - slashpos;
string suffix = hoststring.substr(slashpos, suffixendpos);
hoststring = hoststring.substr(0, slashpos);
cout << "hoststring: " << hoststring << endl;
cout << "suffix: " << suffix << endl;
return hoststring;
}
2 ответа
Я знаю, что вы пытаетесь написать один сервер процессов, который обрабатывает несколько соединений в цикле и использует select() определить, когда дескриптор сокета имеет данные для чтения. Однако иногда другой подход на самом деле проще и масштабируемее.
Рассматривали ли вы использование multi-process сервер, на котором после каждого подключения к сокету вы fork() новый процесс для обработки запроса? Это устраняет необходимость беспокоиться о том, как select() работает и отображает запросы на правильный дескриптор сокета.
См. Обработка нескольких соединений здесь в качестве разумного примера.
В дополнение к использованию fork() Вы можете использовать потоки с POSIX pthreads библиотека, которая может принести вам немного эффективности. Вот хороший пример многопоточного сервера tcp/ip, который использует pthreads.
Вы читали справочные страницы select(2) и select_tut(2)?
Читали ли вы соответствующие главы, например, " Расширенное программирование в Linux" или " Расширенное программирование в Unix"?
На самом деле, из-за проблемы c10k и ограничения максимального дескриптора файла до 256 (или 1024), т.е. __FD_SETSIZE), select Системный вызов устарел, и вместо него следует использовать системный вызов poll(2).
Вы должны установить readfds внутри вашего for цикл, как раз перед select вызов, с явным FD_ZERO а также FD_SET (The fd_set Тип может быть массивом, поэтому вы не можете назначить его целиком.). select системный вызов может изменить его.
Не забудьте собрать с gcc -Wall -g и использовать отладчик. Вы также можете изучить исходный код существующих клиентских библиотек или прокси-серверов свободного программного обеспечения.