blob: 81ca7e368d64560dca828e7ff841331a93e8c480 [file] [log] [blame]
/*
* Copyright 2013 The Emscripten Authors. All rights reserved.
* Emscripten is available under two separate licenses, the MIT license and the
* University of Illinois/NCSA Open Source License. Both these licenses can be
* found in the LICENSE file.
*/
#include <errno.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <assert.h>
#ifdef __EMSCRIPTEN__
#include <emscripten.h>
#endif
#include "test_sockets_msg.h"
#define MESSAGE "0123456789"
int sockfd;
msg_t readmsg;
msg_t writemsg;
void finish(int result) {
close(sockfd);
#ifdef __EMSCRIPTEN__
REPORT_RESULT(result);
emscripten_force_exit(result);
#else
exit(result);
#endif
}
void main_loop() {
static int state = 0;
static int readPos = 0;
static int writePos = 0;
int selectRes;
ssize_t transferAmount;
fd_set sett;
switch (state) {
case 0:
// writing 10 bytes to the server
// since the socket in the read file descriptors has no available data,
// select should tell us 0 handles are ready
FD_ZERO(&sett);
FD_SET(sockfd, &sett);
selectRes = select(64, &sett, NULL, NULL, NULL);
if (selectRes != 0) {
printf("case 0: read select != 0 (%d)\n", selectRes);
finish(EXIT_FAILURE);
}
// the socket in the write file descriptors has to result in either a 0 or 1
// the connection either is setting up or is established and writing is possible
FD_ZERO(&sett);
FD_SET(sockfd, &sett);
selectRes = select(64, NULL, &sett, NULL, NULL);
if (selectRes == -1) {
printf("case 0: write select == -1\n");
finish(EXIT_FAILURE);
} else if (selectRes == 0) {
return;
}
// send a single byte
transferAmount = do_msg_write(sockfd, &writemsg, writePos, 1, NULL, 0);
if (transferAmount != -1) writePos += transferAmount;
// after 10 bytes switch to next state
if (writePos >= writemsg.length) {
state = 1;
}
break;
case 1:
// wait until we can read one byte to make sure the server
// has sent the data and then closed the connection
FD_ZERO(&sett);
FD_SET(sockfd, &sett);
selectRes = select(64, &sett, NULL, NULL, NULL);
if (selectRes == -1) {
printf("case 1: read selectRes == -1\n");
finish(EXIT_FAILURE);
} else if (selectRes == 0) {
return;
}
// read a single byte
transferAmount = do_msg_read(sockfd, &readmsg, readPos, 1, NULL, NULL);
if (transferAmount == 0) {
perror("server closed");
finish(EXIT_FAILURE);
} else if (transferAmount != -1) {
readPos += transferAmount;
}
// if successfully reading 1 byte, switch to next state
if (readPos >= 1) {
state = 2;
}
break;
case 2:
// calling select with the socket in the write file descriptors should
// succeed, but the socket should not set in the set.
FD_ZERO(&sett);
FD_SET(sockfd, &sett);
selectRes = select(64, NULL, &sett, NULL, NULL);
if (selectRes != 0 || FD_ISSET(sockfd, &sett)) {
printf("case 2: write selectRes != 0 || FD_ISSET(sockfd, &sett)\n");
finish(EXIT_FAILURE);
}
// calling select with the socket in the read file descriptors
// has to succeed because there is still data in the inQueue
FD_ZERO(&sett);
FD_SET(sockfd, &sett);
selectRes = select(64, &sett, NULL, NULL, NULL);
if (selectRes != 1) {
printf("case 2: read selectRes != 1\n");
finish(EXIT_FAILURE);
} else if (selectRes == 0) {
return;
}
// read a single byte
transferAmount = do_msg_read(sockfd, &readmsg, readPos, 1, NULL, NULL);
if (transferAmount == 0) {
perror("server closed");
finish(EXIT_FAILURE);
} else if (transferAmount != -1) {
readPos += transferAmount;
}
// with 10 bytes read the inQueue is empty => switch state
if (readPos >= readmsg.length) {
state = 3;
}
break;
case 3:
// calling select with the socket in the read file descriptors
// should succeed
FD_ZERO(&sett);
FD_SET(sockfd, &sett);
selectRes = select(64, &sett, NULL, NULL, NULL);
if (selectRes != 1) {
printf("case 3: read selectRes != 1\n");
finish(EXIT_FAILURE);
}
// but recv should return 0 signaling the remote
// end has closed the connection.
transferAmount = do_msg_read(sockfd, &readmsg, readPos, 0, NULL, NULL);
if (transferAmount) {
printf("case 3: read != 0\n");
finish(EXIT_FAILURE);
}
// report back success, the 266 is just an arbitrary value without
// deeper meaning
finish(266);
break;
default:
printf("Impossible state!\n");
finish(EXIT_FAILURE);
break;
}
return;
}
// This test checks for an intended asymmetry in the behavior of the select function.
// Scenario: the client sends data to the server. After 10 received bytes the
// server sends 10 bytes on its own and immediately afterwards closes the connection.
// This mimics a typical connect-request-response-disconnect situation.
// After the server closed the connection select calls with the socket in the write file
// descriptors have to fail as the tcp connection is already down and there is no way
// anymore to send data.
// Select calls with the socket in the read file descriptor list still have to succeed
// as there are still 10 bytes to read from the inQueue. So, for the same socket the
// select call behaves differently depending on whether the socket is listed in the
// read or write file descriptors.
int main() {
struct sockaddr_in addr;
int res;
memset(&readmsg, 0, sizeof(msg_t));
memset(&writemsg, 0, sizeof(msg_t));
writemsg.length = strlen(MESSAGE) + 1;
writemsg.buffer = malloc(writemsg.length);
strncpy(writemsg.buffer, MESSAGE, writemsg.length);
sockfd = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sockfd == -1) {
perror("cannot create socket");
finish(EXIT_FAILURE);
}
fcntl(sockfd, F_SETFL, O_NONBLOCK);
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_port = htons(SOCKK);
if (inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr) != 1) {
perror("inet_pton failed");
finish(EXIT_FAILURE);
}
// This call should succeed (even if the server port is closed)
res = connect(sockfd, (struct sockaddr *)&addr, sizeof(addr));
if (res == -1 && errno != EINPROGRESS) {
perror("connect failed");
finish(EXIT_FAILURE);
}
#ifdef __EMSCRIPTEN__
emscripten_set_main_loop(main_loop, 0, 0);
#else
while (1) main_loop();
#endif
return EXIT_SUCCESS;
}