TCPを用いたチャットプログラム
Posted: 2017年7月17日(月) 12:33
TCPを用いたチャットプログラムで、サーバ側でforkにより複数のクライアントと接続するまでは上手く行きましたが、子プロセスと親プロセスの間にパイプを繋ぎ、子プロセスに届いたメッセージを子プロセス間で共有しようとすると、2つ目のクライアントを接続後、recv時にBad file descriptorなどが発生し中々上手く行きません。
どのあたりが原因なのでしょうか?
<TCPサーバ>
#include <stdio.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#define MAXPENDING 5
#define RCVBUFSIZE 32
void DieWithError(char *errorMessage);
void HandleTCPClient(int clntSocket);
int main(int argc, char *argv[])
{
int servSock;
int clntSock;
int maxDescriptor;
fd_set sockSet;
int running=1;
struct sockaddr_in echoServAddr;
struct sockaddr_in echoClntAddr;
unsigned short echoServPort;
unsigned int clntLen;
pid_t pid;
int fds[2];
int fd2[2];
char echoBuffer[RCVBUFSIZE];
char andBuffer[RCVBUFSIZE];
int recvMsgSize;
if (argc != 2)
{
fprintf(stderr, "Usage: %s <Server Port>\n", argv[0]);
exit(1);
}
maxDescriptor = -1;
echoServPort = atoi(argv[1]);
if (pipe(fds) < 0) {
perror("fds");
return 1;
}
if (pipe(fd2) < 0) {
perror("fd2");
return 1;
}
if ((servSock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
DieWithError("socket() failed");
memset(&echoServAddr, 0, sizeof(echoServAddr));
echoServAddr.sin_family = AF_INET;
echoServAddr.sin_addr.s_addr = htonl(INADDR_ANY);
echoServAddr.sin_port = htons(echoServPort);
if (bind(servSock, (struct sockaddr *) &echoServAddr, sizeof(echoServAddr)) < 0)
DieWithError("bind() failed");
if (listen(servSock, MAXPENDING) < 0)
DieWithError("listen() failed");
if (servSock > maxDescriptor)
maxDescriptor = servSock;
while(running)
{
FD_ZERO(&sockSet);
FD_SET(STDIN_FILENO, &sockSet);
FD_SET(servSock, &sockSet);
if (select(maxDescriptor + 1, &sockSet, NULL, NULL,NULL) == 0)
printf("wait\n");
else
{
if (FD_ISSET(STDIN_FILENO, &sockSet))
{
printf("Shutting down server\n");
getchar();
running = 0;
}
if (FD_ISSET(servSock, &sockSet))
{
clntLen = sizeof(echoClntAddr);
if ((clntSock = accept(servSock, (struct sockaddr *) &echoClntAddr,
&clntLen)) < 0)
DieWithError("accept() failed");
printf("Handling client %s\n", inet_ntoa(echoClntAddr.sin_addr));
if((pid=fork())==0){
close(fd2[1]);
close(fds[0]);
close(servSock);
if ((recvMsgSize = recv(clntSock, echoBuffer, RCVBUFSIZE, 0)) < 0)
DieWithError("recv() failed");
if(write(fds[1], echoBuffer, recvMsgSize) < 0) {
perror("write()");
return -1;
}
if(read(fd2[0], echoBuffer, RCVBUFSIZE) < 0) {
perror("read()");
return -1;
}
while (recvMsgSize > 0)
{
if (send(clntSock, echoBuffer, recvMsgSize, 0) != recvMsgSize)
DieWithError("send() failed");
if ((recvMsgSize = recv(clntSock, echoBuffer, RCVBUFSIZE, 0)) < 0)
DieWithError("recv() failed");
if(write(fds[1], echoBuffer, recvMsgSize) < 0) {
perror("write()");
return -1;
}
if(read(fd2[0], echoBuffer, RCVBUFSIZE) < 0) {
perror("read()");
return -1;
}
}
printf("koend\n");
close(fd2[0]);
close(fds[1]);
close(clntSock);
_exit(1);
}else if(pid>0){
close(fds[1]);
close(fd2[0]);
if(read(fds[0], echoBuffer, RCVBUFSIZE) < 0) {
perror("read()");
return -1;
}
if(write(fd2[1], echoBuffer, recvMsgSize) < 0) {
perror("write()");
return -1;
}
close(fds[0]);
close(fd2[1]);
close(clntSock);
clntSock=-1;
printf("oya\n");
}else{
close(clntSock);
clntSock=-1;
}
}
}
}
}
どのあたりが原因なのでしょうか?
<TCPサーバ>
#include <stdio.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#define MAXPENDING 5
#define RCVBUFSIZE 32
void DieWithError(char *errorMessage);
void HandleTCPClient(int clntSocket);
int main(int argc, char *argv[])
{
int servSock;
int clntSock;
int maxDescriptor;
fd_set sockSet;
int running=1;
struct sockaddr_in echoServAddr;
struct sockaddr_in echoClntAddr;
unsigned short echoServPort;
unsigned int clntLen;
pid_t pid;
int fds[2];
int fd2[2];
char echoBuffer[RCVBUFSIZE];
char andBuffer[RCVBUFSIZE];
int recvMsgSize;
if (argc != 2)
{
fprintf(stderr, "Usage: %s <Server Port>\n", argv[0]);
exit(1);
}
maxDescriptor = -1;
echoServPort = atoi(argv[1]);
if (pipe(fds) < 0) {
perror("fds");
return 1;
}
if (pipe(fd2) < 0) {
perror("fd2");
return 1;
}
if ((servSock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
DieWithError("socket() failed");
memset(&echoServAddr, 0, sizeof(echoServAddr));
echoServAddr.sin_family = AF_INET;
echoServAddr.sin_addr.s_addr = htonl(INADDR_ANY);
echoServAddr.sin_port = htons(echoServPort);
if (bind(servSock, (struct sockaddr *) &echoServAddr, sizeof(echoServAddr)) < 0)
DieWithError("bind() failed");
if (listen(servSock, MAXPENDING) < 0)
DieWithError("listen() failed");
if (servSock > maxDescriptor)
maxDescriptor = servSock;
while(running)
{
FD_ZERO(&sockSet);
FD_SET(STDIN_FILENO, &sockSet);
FD_SET(servSock, &sockSet);
if (select(maxDescriptor + 1, &sockSet, NULL, NULL,NULL) == 0)
printf("wait\n");
else
{
if (FD_ISSET(STDIN_FILENO, &sockSet))
{
printf("Shutting down server\n");
getchar();
running = 0;
}
if (FD_ISSET(servSock, &sockSet))
{
clntLen = sizeof(echoClntAddr);
if ((clntSock = accept(servSock, (struct sockaddr *) &echoClntAddr,
&clntLen)) < 0)
DieWithError("accept() failed");
printf("Handling client %s\n", inet_ntoa(echoClntAddr.sin_addr));
if((pid=fork())==0){
close(fd2[1]);
close(fds[0]);
close(servSock);
if ((recvMsgSize = recv(clntSock, echoBuffer, RCVBUFSIZE, 0)) < 0)
DieWithError("recv() failed");
if(write(fds[1], echoBuffer, recvMsgSize) < 0) {
perror("write()");
return -1;
}
if(read(fd2[0], echoBuffer, RCVBUFSIZE) < 0) {
perror("read()");
return -1;
}
while (recvMsgSize > 0)
{
if (send(clntSock, echoBuffer, recvMsgSize, 0) != recvMsgSize)
DieWithError("send() failed");
if ((recvMsgSize = recv(clntSock, echoBuffer, RCVBUFSIZE, 0)) < 0)
DieWithError("recv() failed");
if(write(fds[1], echoBuffer, recvMsgSize) < 0) {
perror("write()");
return -1;
}
if(read(fd2[0], echoBuffer, RCVBUFSIZE) < 0) {
perror("read()");
return -1;
}
}
printf("koend\n");
close(fd2[0]);
close(fds[1]);
close(clntSock);
_exit(1);
}else if(pid>0){
close(fds[1]);
close(fd2[0]);
if(read(fds[0], echoBuffer, RCVBUFSIZE) < 0) {
perror("read()");
return -1;
}
if(write(fd2[1], echoBuffer, recvMsgSize) < 0) {
perror("write()");
return -1;
}
close(fds[0]);
close(fd2[1]);
close(clntSock);
clntSock=-1;
printf("oya\n");
}else{
close(clntSock);
clntSock=-1;
}
}
}
}
}