static int client_recv(int *fd, char *buf)
{
int len, offset = 0;
while (1) {
errno = 0;
len = sock_recv_poll(*fd, buf, server_msg_size, offset);
/* socket closed or msg is not valid */
if (len < 1 || (offset + len) > server_msg_size ||
(buf[0] != start_byte && buf[0] != start_fin_byte)) {
if (!errno)
errno = ENOMSG;
break;
}
offset += len;
if (buf[offset - 1] != end_byte)
continue;
if (verbose) {
tst_resm_hexd(TINFO, buf, offset,
"msg recv from sock %d:", *fd);
}
/* recv last msg, close socket */
if (buf[0] == start_fin_byte)
break;
return 0;
}
shutdown(*fd, SHUT_WR);
SAFE_CLOSE(cleanup, *fd);
*fd = -1;
return (errno) ? -1 : 0;
}
static void check_file_data(const char exp_buf[], size_t size)
{
char rbuf[size];
tst_resm(TINFO, "reading the file, compare with expected buffer");
SAFE_LSEEK(cleanup, fd, 0, SEEK_SET);
SAFE_READ(cleanup, 1, fd, rbuf, size);
if (memcmp(exp_buf, rbuf, size)) {
if (verbose) {
tst_resm_hexd(TINFO, exp_buf, size, "expected:");
tst_resm_hexd(TINFO, rbuf, size, "but read:");
}
tst_brkm(TFAIL, cleanup, "not expected file data");
}
}
void *server_fn(void *cfd)
{
int client_fd = (intptr_t) cfd;
int num_requests = 0, offset = 0;
/* Reply will be constructed from first client request */
char *send_msg = NULL;
int send_msg_size = 0;
char recv_msg[max_msg_len];
setsockopt(client_fd, SOL_SOCKET, SO_LINGER, &clo, sizeof(clo));
ssize_t recv_len;
while (1) {
recv_len = sock_recv_poll(client_fd, recv_msg,
max_msg_len, offset);
if (recv_len == 0)
break;
if (recv_len < 0 || (offset + recv_len) > max_msg_len ||
(recv_msg[0] != start_byte &&
recv_msg[0] != start_fin_byte)) {
tst_resm(TFAIL, "recv failed, sock '%d'", client_fd);
goto out;
}
offset += recv_len;
if (recv_msg[offset - 1] != end_byte) {
/* msg is not complete, continue recv */
continue;
}
/* client asks to terminate */
if (recv_msg[0] == start_fin_byte)
goto out;
if (verbose) {
tst_resm_hexd(TINFO, recv_msg, offset,
"msg recv from sock %d:", client_fd);
}
/* if we send reply for the first time, construct it here */
if (!send_msg) {
send_msg_size = parse_client_request(recv_msg);
if (send_msg_size < 0) {
tst_resm(TFAIL, "wrong msg size '%d'",
send_msg_size);
goto out;
}
send_msg = make_server_reply(send_msg_size);
}
/*
* It will tell client that server is going
* to close this connection.
*/
if (++num_requests >= server_max_requests)
send_msg[0] = start_fin_byte;
if (send(client_fd, send_msg, send_msg_size,
MSG_NOSIGNAL) == -1) {
tst_resm(TFAIL | TERRNO, "send failed");
goto out;
}
offset = 0;
if (num_requests >= server_max_requests) {
/* max reqs, close socket */
shutdown(client_fd, SHUT_WR);
break;
}
}
free(send_msg);
SAFE_CLOSE(cleanup, client_fd);
return NULL;
out:
free(send_msg);
SAFE_CLOSE(cleanup, client_fd);
cleanup();
tst_exit();
}
void *client_fn(LTP_ATTRIBUTE_UNUSED void *arg)
{
char buf[server_msg_size];
int cfd, i;
intptr_t err = 0;
/* connect & send requests */
cfd = client_connect_send(client_msg, client_msg_size);
if (cfd == -1) {
err = errno;
goto out;
}
if (client_recv(&cfd, buf)) {
err = errno;
goto out;
}
for (i = 1; i < client_max_requests; ++i) {
/* check connection, it can be closed */
int ret = 0;
if (cfd != -1)
ret = recv(cfd, buf, 1, MSG_DONTWAIT);
if (ret == 0) {
/* try to reconnect and send */
if (cfd != -1)
SAFE_CLOSE(cleanup, cfd);
cfd = client_connect_send(client_msg, client_msg_size);
if (cfd == -1) {
err = errno;
goto out;
}
if (client_recv(&cfd, buf)) {
err = errno;
break;
}
continue;
} else if (ret > 0) {
err = EMSGSIZE;
break;
}
if (verbose) {
tst_resm_hexd(TINFO, client_msg, client_msg_size,
"try to send msg[%d]", i);
}
if (send(cfd, client_msg, client_msg_size,
MSG_NOSIGNAL) != client_msg_size) {
err = ECOMM;
break;
}
if (client_recv(&cfd, buf)) {
err = errno;
break;
}
}
if (cfd != -1)
SAFE_CLOSE(cleanup, cfd);
out:
return (void *) err;
}
