tinc-vpn.org Git - tinc/blob - test/splice.c

   1 /*
   2     splice.c -- Splice two outgoing tinc connections together
   3     Copyright (C) 2018 Guus Sliepen <guus@tinc-vpn.org>
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation; either version 2 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License along
  16     with this program; if not, write to the Free Software Foundation, Inc.,
  17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 */
  19
  20 #include "../src/system.h"
  21
  22 #ifdef HAVE_MINGW
  23 static const char *winerror(int err) {
  24         static char buf[1024], *ptr;
  25
  26         ptr = buf + snprintf(buf, sizeof(buf), "(%d) ", err);
  27
  28         if(!FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
  29                           NULL, err, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), ptr, sizeof(buf) - (ptr - buf), NULL)) {
  30                 strncpy(buf, "(unable to format errormessage)", sizeof(buf));
  31         };
  32
  33         if((ptr = strchr(buf, '\r'))) {
  34                 *ptr = '\0';
  35         }
  36
  37         return buf;
  38 }
  39
  40 #define strerror(x) ((x)>0?strerror(x):winerror(GetLastError()))
  41 #define sockerrno WSAGetLastError()
  42 #define sockstrerror(x) winerror(x)
  43 #else
  44 #define sockerrno errno
  45 #define sockstrerror(x) strerror(x)
  46 #endif
  47
  48 int main(int argc, char *argv[]) {
  49         if(argc < 7) {
  50                 fprintf(stderr, "Usage: %s name1 host1 port1 name2 host2 port2 [protocol]\n", argv[0]);
  51                 return 1;
  52         }
  53
  54         const char *protocol;
  55
  56         if(argc >= 8) {
  57                 protocol = argv[7];
  58         } else {
  59                 protocol = "17.7";
  60         }
  61
  62 #ifdef HAVE_MINGW
  63         static struct WSAData wsa_state;
  64
  65         if(WSAStartup(MAKEWORD(2, 2), &wsa_state)) {
  66                 return 1;
  67         }
  68
  69 #endif
  70         int sock[2];
  71         char buf[1024];
  72
  73         struct addrinfo *ai, hint;
  74         memset(&hint, 0, sizeof(hint));
  75
  76         hint.ai_family = AF_UNSPEC;
  77         hint.ai_socktype = SOCK_STREAM;
  78         hint.ai_protocol = IPPROTO_TCP;
  79         hint.ai_flags = 0;
  80
  81         for (int i = 0; i < 2; i++) {
  82                 if(getaddrinfo(argv[2 + 3 * i], argv[3 + 3 * i], &hint, &ai) || !ai) {
  83                         fprintf(stderr, "getaddrinfo() failed: %s\n", sockstrerror(sockerrno));
  84                         return 1;
  85                 }
  86
  87                 sock[i] = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol);
  88
  89                 if(sock[i] == -1) {
  90                         fprintf(stderr, "Could not create socket: %s\n", sockstrerror(sockerrno));
  91                         return 1;
  92                 }
  93
  94                 if(connect(sock[i], ai->ai_addr, ai->ai_addrlen)) {
  95                         fprintf(stderr, "Could not connect to %s: %s\n", argv[i + 3 * i], sockstrerror(sockerrno));
  96                         return 1;
  97                 }
  98
  99                 fprintf(stderr, "Connected to %s\n", argv[1 + 3 * i]);
 100
 101                 /* Pretend to be the other one */
 102                 int len = snprintf(buf, sizeof buf, "0 %s %s\n", argv[4 - 3 * i], protocol);
 103                 if (send(sock[i], buf, len, 0) != len) {
 104                         fprintf(stderr, "Error sending data to %s: %s\n", argv[1 + 3 * i], sockstrerror(sockerrno));
 105                         return 1;
 106                 }
 107
 108                 /* Ignore the response */
 109                 do {
 110                         if (recv(sock[i], buf, 1, 0) != 1) {
 111                                 fprintf(stderr, "Error reading data from %s: %s\n", argv[1 + 3 * i], sockstrerror(sockerrno));
 112                                 return 1;
 113                         }
 114                 } while(*buf != '\n');
 115         }
 116
 117         fprintf(stderr, "Splicing...\n");
 118
 119         int nfds = (sock[0] > sock[1] ? sock[0] : sock[1]) + 1;
 120
 121         while(true) {
 122                 fd_set fds;
 123                 FD_ZERO(&fds);
 124                 FD_SET(sock[0], &fds);
 125                 FD_SET(sock[1], &fds);
 126
 127                 if(select(nfds, &fds, NULL, NULL, NULL) <= 0) {
 128                         return 1;
 129                 }
 130
 131                 for(int i = 0; i < 2; i++ ) {
 132                         if(FD_ISSET(sock[i], &fds)) {
 133                                 ssize_t len = recv(sock[i], buf, sizeof buf, 0);
 134
 135                                 if(len < 0) {
 136                                         fprintf(stderr, "Error while reading from %s: %s\n", argv[1 + i * 3], sockstrerror(sockerrno));
 137                                         return 1;
 138                                 }
 139
 140                                 if(len == 0) {
 141                                         fprintf(stderr, "Connection closed by %s\n", argv[1 + i * 3]);
 142                                         return 0;
 143                                 }
 144
 145                                 if(send(sock[i ^ 1], buf, len, 0) != len) {
 146                                         fprintf(stderr, "Error while writing to %s: %s\n", argv[4 - i * 3], sockstrerror(sockerrno));
 147                                         return 1;
 148                                 }
 149                         }
 150                 }
 151         }
 152
 153         return 0;
 154 }