Здравствуйте. Надеюсь на вашу помощь. Пишу программу сервер с использованием вызова select(). Программа реализует чат между клиентами. Проблема в следующем. Если один из клиентов неожиданно завершил соединение, то сервер падает, и соединение с другими клиентами разрывается. Как решить эту проблему?
Вот код. #include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>
//#include <sys/types.h>
#include <sys/wait.h>
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <netdb.h>
#include <signal.h>
#define BACKDOOR_PORT 8080
#define BUFS 1024
pid_t pid;
void init_daemon(void);
void do_back_dooring(void);
int accept_client(int fd);
int do_the_job(int fd);
int main(int argc, char *argv[])
{
pid = fork(); //породили дочерний процесс
if (pid < 0)
{
perror("fork() error");
exit(1);
}
if (pid > 0) // родительский процесс.
{
printf("Starting...\n");
exit(0); //завершили родительский процесс.
}
init_daemon();// становимся "демоном"
while(1)
{
do_back_dooring();
}
}
int tcp_socket;
struct sockaddr_in tcp_socket_side;
fd_set main_set;
int max_fd;
int x;
void init_daemon(void)
{
int ret;
tcp_socket = socket(PF_INET, SOCK_STREAM, 0);//создали сокет
x = tcp_socket;
if (tcp_socket == -1)
{
perror("socket() error");
exit(1);
}
else
{
printf("socket()_OK!\n");
}
struct sockaddr_in socket_addr;
socket_addr.sin_family = AF_INET;
socket_addr.sin_port = htons(BACKDOOR_PORT);
socket_addr.sin_addr.s_addr = INADDR_ANY;
memset( &(socket_addr.sin_zero),0,sizeof(socket_addr.sin_zero));
ret = bind(tcp_socket, (struct sockaddr*)&socket_addr,sizeof(struct sockaddr));//задали сокету имя
if (ret == -1)
{
perror("bind() error");
exit(1);
}
else
{
printf("bind()_OK!\n");
}
ret = listen(tcp_socket, 1000);//перевели сокет в слушающий режим.
if (ret == -1)
{
perror("listen() error");
exit(1);
}
else
{
printf("listen()_OK!\n");
printf("Server is ready to accept clients\n");
}
#ifndef DEBUG
/* закрыли стандартные устройства ввода - вывода */
close(STDIN_FILENO);
close(STDOUT_FILENO);
close(STDERR_FILENO);
setsid();
#endif
//становимся демоном
FD_ZERO(&main_set);
FD_SET(tcp_socket, &main_set);
max_fd = tcp_socket;
}
fd_set temp_set;
char *prompt = "::command://";
char *selecterr = "select()_error!\n";
char *error = "error\n";
int fd;
char *msg;
char buf[BUFS];
void do_back_dooring(void) //функция проверки сокетов...
{
int ret;
int i;
memcpy(&temp_set, &main_set, sizeof(fd_set));
ret = select(max_fd+1, &temp_set, NULL, NULL, NULL);
if (ret < 0)
{
write(fd, selecterr, strlen(selecterr));
FD_ZERO(&temp_set);
return;
}
for (fd = 0; fd <= max_fd; fd++)
{
if (FD_ISSET(fd, &temp_set))
{
if (fd == tcp_socket) //появился новый клиент.
{
accept_client(fd);
}
else
{
if (!do_the_job(fd)) //исполняем команды клиента
{
FD_CLR(fd, &main_set);
close(fd);
}
for (i = 0; i < tcp_socket; i++)
{
if (FD_ISSET(i, &main_set))
{
write(i, msg, strlen(msg));
}
}
write(fd, prompt, strlen(prompt));
}
}
}
}
char *hi = "Welcome! Please enter a key.\n";
char *invkey = "Invalid key, connection closed\n";
char *ok = "Congratulations!\n";
//char buf[BUFS];
int accept_client(int back_door) //принимаем соединение от клиента.
{
int ret;
int err;
struct sockaddr_in client;
int addr_size = sizeof(struct sockaddr_in);
ret = accept(back_door, (struct sockaddr *)&client, &addr_size);
if (ret == -1)
{
return 0;
}
write(ret, hi, strlen(hi));
write(ret, prompt, strlen(prompt));
memset(buf, 0, BUFS);
if (read(ret, buf, BUFS) == -1)
{
return 0;
}
if ( !strncmp(buf, "PASSWORD", 8))
{
write(ret, ok, strlen(ok));
memset(buf, 0, BUFS);
if (ret > max_fd)
{
max_fd = ret;
}
FD_SET(ret, &main_set);
write(ret, prompt, strlen(prompt));
}
else
{
write(ret, invkey, strlen(invkey));
close(ret);
}
return 0;
}
char *bye = "Good bye!\n";
void sig_child_handler(int sign)
{
return;
}
char *prog = buf;
char *args[13];
int do_the_job(int fd) //исполняем команды от клиента.
{
int ret;
int i;
pid_t pid;
int child_status;
char *tok;
int argn;
memset(buf, 0, BUFS); //первые BUFS байт заполнили buf нулями. Очистили буфер.
ret = read(fd, buf, BUFS);// прочитали то, что прислал клиент. из fd в buf.
if (ret == -1)
{
write(fd, error, strlen(error));
return 0;
}
if (!strncmp(buf, "STOP", 4))//сравнивает buf и STOP, и, если они равны, то завершаем соединение.
{
write(fd, bye, strlen(bye));
memset(buf, 0, BUFS);
return 0;
}
msg = buf;
//write(fd, msg, strlen(msg));
//return 1;
/*signal(SIGCHLD, sig_child_handler);
ret = sleep(5);
if (ret == 0)
{
kill(pid, SIGKILL);
signal(SIGCHLD, SIG_IGN);
waitpid(pid, &child_status, 0);
}*/
return 1;
}