vrmodel

annotate src/inpclient.cc @ 5:9e260c091f75

find changesets by author, revision, files, or words in the commit message
wrote some server code in the test input tool
author John Tsiombikas <nuclear@member.fsf.org>
date Mon, 15 Sep 2014 10:57:31 +0300
parents a32b151fb3c6
children
rev   line source
nuclear@2 1 #include <stdio.h>
nuclear@2 2 #include <stdlib.h>
nuclear@2 3 #include <string.h>
nuclear@2 4 #include <errno.h>
nuclear@2 5 #include <unistd.h>
nuclear@2 6 #include <sys/types.h>
nuclear@2 7 #include <sys/socket.h>
nuclear@2 8 #include <arpa/inet.h>
nuclear@2 9 #include <sys/time.h>
nuclear@2 10 #include <sys/select.h>
nuclear@2 11 #include "inpclient.h"
nuclear@2 12 #include "proto.h"
nuclear@2 13
nuclear@2 14 static int discover(struct sockaddr_in *srv_sa);
nuclear@3 15 static int send_request(int s, struct message *msg, long timeout);
nuclear@2 16
nuclear@2 17 static int opt_dport = DEF_PORT; /* discover broadcast port */
nuclear@4 18 static long opt_timeout = 5000; /* 10 sec discovery timeout */
nuclear@2 19
nuclear@2 20 static int sock = -1;
nuclear@2 21
nuclear@2 22
nuclear@2 23 int netinp_start()
nuclear@2 24 {
nuclear@2 25 struct message msg;
nuclear@2 26 struct sockaddr_in srv_sa;
nuclear@2 27
nuclear@2 28 if(sock == -1) {
nuclear@2 29 if((sock = discover(&srv_sa)) == -1) {
nuclear@2 30 return -1;
nuclear@2 31 }
nuclear@2 32
nuclear@2 33 if(connect(sock, (struct sockaddr*)&srv_sa, sizeof srv_sa) == -1) {
nuclear@2 34 perror("failed to connect socket");
nuclear@2 35 close(sock);
nuclear@2 36 return -1;
nuclear@2 37 }
nuclear@2 38 }
nuclear@2 39
nuclear@2 40 memset(&msg, 0, sizeof msg);
nuclear@2 41 msg.magic = MAGIC;
nuclear@3 42 msg.type = REQ_START;
nuclear@4 43 if(send_request(sock, &msg, opt_timeout) == -1) {
nuclear@4 44 return -1;
nuclear@4 45 }
nuclear@4 46 return msg.magic == MAGIC && msg.type == RSP_OK ? 0 : -1;
nuclear@2 47 }
nuclear@2 48
nuclear@2 49 int netinp_stop()
nuclear@2 50 {
nuclear@3 51 struct message msg;
nuclear@3 52
nuclear@3 53 if(sock == -1) {
nuclear@3 54 return -1;
nuclear@3 55 }
nuclear@3 56
nuclear@3 57 memset(&msg, 0, sizeof msg);
nuclear@3 58 msg.magic = MAGIC;
nuclear@3 59 msg.type = REQ_STOP;
nuclear@4 60
nuclear@4 61 if(send_request(sock, &msg, opt_timeout) == -1) {
nuclear@4 62 return -1;
nuclear@4 63 }
nuclear@4 64 return msg.magic == MAGIC && msg.type == RSP_OK ? 0 : -1;
nuclear@3 65 }
nuclear@3 66
nuclear@4 67 int netinp_read(float *pos, unsigned int *bnmask)
nuclear@3 68 {
nuclear@3 69 struct message msg;
nuclear@3 70
nuclear@3 71 if(sock == -1) {
nuclear@3 72 return -1;
nuclear@3 73 }
nuclear@3 74
nuclear@3 75 if(recv(sock, &msg, sizeof msg, 0) == -1) {
nuclear@3 76 perror("netinp_read failed");
nuclear@3 77 return -1;
nuclear@3 78 }
nuclear@3 79
nuclear@4 80 if(msg.magic != MAGIC || msg.type != MAGIC) {
nuclear@4 81 fprintf(stderr, "not enough magic in the message\n");
nuclear@4 82 return -1;
nuclear@4 83 }
nuclear@4 84
nuclear@4 85 pos[0] = msg.data.event.x;
nuclear@4 86 pos[1] = msg.data.event.y;
nuclear@4 87 pos[2] = msg.data.event.z;
nuclear@4 88 *bnmask = msg.data.event.bnmask;
nuclear@3 89 return 0;
nuclear@3 90 }
nuclear@3 91
nuclear@3 92 int netinp_reset()
nuclear@3 93 {
nuclear@4 94 struct message msg;
nuclear@4 95
nuclear@4 96 if(sock == -1) {
nuclear@4 97 return -1;
nuclear@4 98 }
nuclear@4 99
nuclear@4 100 memset(&msg, 0, sizeof msg);
nuclear@4 101 msg.magic = MAGIC;
nuclear@4 102 msg.type = REQ_STOP;
nuclear@4 103
nuclear@4 104 if(send_request(sock, &msg, opt_timeout) == -1) {
nuclear@4 105 return -1;
nuclear@4 106 }
nuclear@4 107 return msg.magic == MAGIC && msg.type == RSP_OK ? 0 : -1;
nuclear@2 108 }
nuclear@2 109
nuclear@2 110 static int discover(struct sockaddr_in *srv_sa)
nuclear@2 111 {
nuclear@2 112 int s, true_val = 1;
nuclear@2 113 struct sockaddr_in sa;
nuclear@2 114 long timeout = opt_timeout;
nuclear@2 115 struct message msg;
nuclear@2 116 struct timeval tv0, tv, tv_timeout;
nuclear@2 117
nuclear@2 118 if((s = socket(PF_INET, SOCK_DGRAM, 0)) == -1) {
nuclear@2 119 perror("failed to create discover datagram socket");
nuclear@2 120 return -1;
nuclear@2 121 }
nuclear@2 122 setsockopt(s, SOL_SOCKET, SO_BROADCAST, &true_val, sizeof true_val);
nuclear@2 123
nuclear@2 124 do {
nuclear@2 125 fd_set rdset;
nuclear@2 126
nuclear@2 127 msg.magic = MAGIC;
nuclear@2 128 msg.type = REQ_DISCOVER;
nuclear@2 129
nuclear@2 130 memset(&sa, 0, sizeof sa);
nuclear@2 131 sa.sin_family = AF_INET;
nuclear@2 132 sa.sin_port = htons(opt_dport);
nuclear@2 133 sa.sin_addr.s_addr = 0xffffffff;
nuclear@2 134
nuclear@2 135 if(sendto(s, &msg, sizeof msg, 0, (struct sockaddr*)&sa, sizeof sa) == -1) {
nuclear@2 136 perror("failed to send discovery bcast dgram");
nuclear@2 137 close(s);
nuclear@2 138 return -1;
nuclear@2 139 }
nuclear@2 140
nuclear@2 141 tv_timeout.tv_sec = timeout / 1000;
nuclear@2 142 tv_timeout.tv_usec = (timeout % 1000) * 1000;
nuclear@2 143
nuclear@2 144 FD_ZERO(&rdset);
nuclear@2 145 FD_SET(s, &rdset);
nuclear@2 146
nuclear@2 147 gettimeofday(&tv0, 0);
nuclear@2 148
nuclear@2 149 while(select(s + 1, &rdset, 0, 0, &tv_timeout) == -1 && errno == EINTR);
nuclear@2 150
nuclear@2 151 gettimeofday(&tv, 0);
nuclear@2 152 timeout -= (tv.tv_sec - tv0.tv_sec) * 1000 + (tv.tv_usec - tv0.tv_usec) / 1000;
nuclear@2 153
nuclear@2 154 if(FD_ISSET(s, &rdset)) {
nuclear@2 155 socklen_t sa_size = sizeof *srv_sa;
nuclear@2 156 if(recvfrom(s, &msg, sizeof msg, 0, (struct sockaddr*)srv_sa, &sa_size) == -1) {
nuclear@2 157 perror("failed to receive datagram\n");
nuclear@2 158 }
nuclear@2 159 if(msg.magic == MAGIC && msg.type == RSP_OK) {
nuclear@4 160 printf("found input server \"%s\" at: %s:%d\n", msg.data.raw,
nuclear@2 161 inet_ntoa(srv_sa->sin_addr), ntohs(srv_sa->sin_port));
nuclear@2 162 return s;
nuclear@2 163 }
nuclear@2 164 }
nuclear@2 165
nuclear@2 166 } while(timeout > 0);
nuclear@2 167
nuclear@2 168 close(s);
nuclear@2 169 return -1;
nuclear@2 170 }
nuclear@2 171
nuclear@3 172 static int send_request(int s, struct message *msg, long timeout)
nuclear@2 173 {
nuclear@3 174 fd_set rdset;
nuclear@3 175 struct timeval tv, tv0, tv_timeout;
nuclear@3 176 struct message resp;
nuclear@2 177
nuclear@3 178 do {
nuclear@3 179 if(send(s, msg, sizeof *msg, 0) == -1) {
nuclear@3 180 perror("failed to send message");
nuclear@3 181 return -1;
nuclear@3 182 }
nuclear@3 183
nuclear@3 184 tv_timeout.tv_sec = timeout / 1000;
nuclear@3 185 tv_timeout.tv_usec = (timeout % 1000) * 1000;
nuclear@3 186
nuclear@3 187 FD_ZERO(&rdset);
nuclear@3 188 FD_SET(s, &rdset);
nuclear@3 189
nuclear@3 190 gettimeofday(&tv0, 0);
nuclear@3 191 while(select(s + 1, &rdset, 0, 0, &tv_timeout) == -1 && errno == EINTR);
nuclear@3 192 gettimeofday(&tv, 0);
nuclear@3 193
nuclear@3 194 if(FD_ISSET(s, &rdset)) {
nuclear@3 195 if(recv(s, &resp, sizeof resp, 0) == -1) {
nuclear@3 196 perror("failed to recieve message");
nuclear@3 197 return -1;
nuclear@3 198 }
nuclear@3 199 if(resp.magic != MAGIC) {
nuclear@3 200 fprintf(stderr, "got invalid response\n");
nuclear@3 201 return -1;
nuclear@3 202 }
nuclear@3 203 memcpy(msg, &resp, sizeof resp);
nuclear@3 204 return 0;
nuclear@3 205 }
nuclear@3 206
nuclear@3 207 timeout -= (tv.tv_sec - tv0.tv_sec) * 1000 + (tv.tv_usec - tv0.tv_usec) / 1000;
nuclear@3 208 } while(timeout > 0);
nuclear@3 209
nuclear@3 210 return -1;
nuclear@2 211 }