#include #define MAX_BUF_SIZE 1024
int main(int argc, char *argv[])
{
int sockfd;
int sin_size;
struct sockaddr_in dest;
char buf[MAX_BUF_SIZE + 1];
fd_set rfds;
struct timeval tv;
int retval, maxfd = -1;
if (argc != 3)
{
printf("usage: %s ip port\n for example: %s 127.0.0.1 8003\n", argv[0], argv[0]);
exit(1);
}
if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
fprintf(stderr, "socket error!\n");
exit(1);
}
bzero(&dest, sizeof(dest));
dest.sin_family = AF_INET;
dest.sin_port = htons(atoi(argv[2]));
if (inet_aton(argv[1], (struct in_addr *)&dest.sin_addr.s_addr) == 0)
{
fprintf(stderr, "%s error\n", argv[1]);
exit(1);
}
if (connect(sockfd, (struct sockaddr *)&dest, sizeof(struct sockaddr)) == -1)
{
fprintf(stderr, "connect error!\n");
exit(1);
}
printf("Be ready, can chat!\n");
FD_ZERO(&rfds);
maxfd = sockfd;
while (1)
{
FD_CLR(sockfd, &rfds);
FD_SET(sockfd, &rfds);
FD_SET(0, &rfds);
tv.tv_sec = 1;
tv.tv_usec = 0;
if ((retval = select(maxfd + 1, &rfds, NULL, NULL, &tv)) < 0)
{
if (errno == EINTR)
{
fprintf(stdout, "closed by signal, continue...\n");
continue;
}
else if (errno == EAGAIN)
continue;
}
else
{
fprintf(stderr, "select error!\n");
break;
}
}
else if (retval == 0)
{
continue;
}
if (FD_ISSET(sockfd, &rfds))
{
bzero(buf, MAX_BUF_SIZE + 1);
sin_size = recv(sockfd, buf, MAX_BUF_SIZE, 0);
if (sin_size > 0)
{
printf("receive message from server successfully: [%s]\ntotal %d bytes\n", buf, sin_size);
}
else
{
if (sin_size < 0)
printf("receive message failure! error code is %d, error message is %s\n",
errno, strerror(errno));
}
}
if (FD_ISSET(0, &rfds))
{
bzero(buf, MAX_BUF_SIZE + 1);
fgets(buf, MAX_BUF_SIZE, stdin);
if (!strncasecmp(buf, "quit", 4))
{
printf("quit chat!\n");
break;
}
sin_size = send(sockfd, buf, strlen(buf) - 1, 0);
if (sin_size < 0)
{
printf("message send failure: [%s], error code is %d, error message is %s\n",
buf, errno, strerror(errno));
break;
}
else
{
printf("message: %s\tsend to server successfully, total %d bytes\n", buf, sin_size);
}
}
}
close(sockfd);
return 0;
}