← Back to branch summary

~ubuntu-branches/ubuntu/vivid/emscripten/vivid

« back to all changes in this revision

Viewing changes to tests/sockets/test_sockets_echo_client.c

  • Committer: Package Import Robot
  • Author(s): Sylvestre Ledru
  • Date: 2013-09-20 22:44:35 UTC
  • mfrom: (4.1.1 sid)
  • Revision ID: package-import@ubuntu.com-20130920224435-apuwj4fsl3fqv1a6
Tags: 1.5.6~20130920~6010666-1
http://bugs.debian.org/723129
* New snapshot release
* Update the list of supported architectures to the same as libv8
  (Closes: #723129)
* emlibtool has been removed from upstream.
* Fix warning syntax-error-in-dep5-copyright
* Refresh of the patches

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/sockets/test_sockets_echo_client.c
 
1
#include <errno.h>
 
2
#include <sys/types.h>
 
3
#include <sys/socket.h>
 
4
#include <netinet/in.h>
 
5
#include <arpa/inet.h>
 
6
#include <stdio.h>
 
7
#include <stdlib.h>
 
8
#include <string.h>
 
9
#include <unistd.h>
 
10
#include <fcntl.h>
 
11
#include <sys/ioctl.h>
 
12
#include <assert.h>
 
13
#if EMSCRIPTEN
 
14
#include <emscripten.h>
 
15
#endif
 
16
 
 
17
#include "test_sockets_msg.h"
 
18
 
 
19
// message to send to the server
 
20
#ifndef MESSAGE
 
21
#define MESSAGE "pingtothepong"
 
22
#endif
 
23
 
 
24
typedef enum {
 
25
  MSG_READ,
 
26
  MSG_WRITE
 
27
} msg_state_t;
 
28
 
 
29
typedef struct {
 
30
  int fd;
 
31
  msg_t msg;
 
32
  msg_state_t state;
 
33
} server_t;
 
34
 
 
35
server_t server;
 
36
msg_t echo_msg;
 
37
int echo_read;
 
38
int echo_wrote;
 
39
 
 
40
void finish(int result) {
 
41
  if (server.fd) {
 
42
    close(server.fd);
 
43
    server.fd = 0;
 
44
  }
 
45
#if EMSCRIPTEN
 
46
  REPORT_RESULT();
 
47
#endif
 
48
  exit(result);
 
49
}
 
50
 
 
51
void main_loop(void *arg) {
 
52
  static char out[1024*2];
 
53
  static int pos = 0;
 
54
  fd_set fdr;
 
55
  fd_set fdw;
 
56
  int res;
 
57
 
 
58
  // make sure that server.fd is ready to read / write
 
59
  FD_ZERO(&fdr);
 
60
  FD_ZERO(&fdw);
 
61
  FD_SET(server.fd, &fdr);
 
62
  FD_SET(server.fd, &fdw);
 
63
  res = select(64, &fdr, &fdw, NULL, NULL);
 
64
  if (res == -1) {
 
65
    perror("select failed");
 
66
    finish(EXIT_FAILURE);
 
67
  }
 
68
 
 
69
  if (server.state == MSG_READ) {
 
70
    if (!FD_ISSET(server.fd, &fdr)) {
 
71
      return;
 
72
    }
 
73
 
 
74
    // as a test, confirm with ioctl that we have data available
 
75
    // after selecting
 
76
    int available;
 
77
    res = ioctl(server.fd, FIONREAD, &available);
 
78
    assert(res != -1);
 
79
    assert(available);
 
80
 
 
81
    res = do_msg_read(server.fd, &server.msg, echo_read, 0, NULL, NULL);
 
82
    if (res == 0) {
 
83
      perror("server closed");
 
84
      finish(EXIT_FAILURE);
 
85
    } else if (res != -1) {
 
86
      echo_read += res;
 
87
    }
 
88
 
 
89
    // once we've read the entire message, validate it
 
90
    if (echo_read >= server.msg.length) {
 
91
      assert(!strcmp(server.msg.buffer, MESSAGE));
 
92
      finish(EXIT_SUCCESS);
 
93
    }
 
94
  } else {
 
95
    if (!FD_ISSET(server.fd, &fdw)) {
 
96
      return;
 
97
    }
 
98
 
 
99
    res = do_msg_write(server.fd, &echo_msg, echo_wrote, 0, NULL, 0);
 
100
    if (res == 0) {
 
101
      perror("server closed");
 
102
      finish(EXIT_FAILURE);
 
103
    } else if (res != -1) {
 
104
      echo_wrote += res;
 
105
    }
 
106
 
 
107
    // once we're done writing the message, read it back
 
108
    if (echo_wrote >= echo_msg.length) {
 
109
      server.state = MSG_READ;
 
110
    }
 
111
  }
 
112
}
 
113
 
 
114
int main() {
 
115
  struct sockaddr_in addr;
 
116
  int res;
 
117
 
 
118
  memset(&server, 0, sizeof(server_t));
 
119
  server.state = MSG_WRITE;
 
120
 
 
121
  // setup the message we're going to echo
 
122
  memset(&echo_msg, 0, sizeof(msg_t));
 
123
  echo_msg.length = strlen(MESSAGE) + 1;
 
124
  echo_msg.buffer = malloc(echo_msg.length);
 
125
  strncpy(echo_msg.buffer, MESSAGE, echo_msg.length);
 
126
 
 
127
  // create the socket and set to non-blocking
 
128
#if !USE_UDP
 
129
  server.fd = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
 
130
#else
 
131
  server.fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
 
132
#endif
 
133
  if (server.fd == -1) {
 
134
    perror("cannot create socket");
 
135
    finish(EXIT_FAILURE);
 
136
  }
 
137
  fcntl(server.fd, F_SETFL, O_NONBLOCK);
 
138
 
 
139
  // connect the socket
 
140
  memset(&addr, 0, sizeof(addr));
 
141
  addr.sin_family = AF_INET;
 
142
  addr.sin_port = htons(SOCKK);
 
143
  if (inet_pton(AF_INET, "127.0.0.1", &addr.sin_addr) != 1) {
 
144
    perror("inet_pton failed");
 
145
    finish(EXIT_FAILURE);
 
146
  }
 
147
  
 
148
  res = connect(server.fd, (struct sockaddr *)&addr, sizeof(addr));
 
149
  if (res == -1 && errno != EINPROGRESS) {
 
150
    perror("connect failed");
 
151
    finish(EXIT_FAILURE);
 
152
  }
 
153
 
 
154
#if EMSCRIPTEN
 
155
  emscripten_set_main_loop(main_loop, 0, 0);
 
156
#else
 
157
  while (1) main_loop(NULL);
 
158
#endif
 
159
 
 
160
  return EXIT_SUCCESS;
 
161
}
 
 
b'\\ No newline at end of file'