dsock.h

   1 /*
   2     Copyright 2004-2020 Chris Tallon
   3
   4     This file is part of VOMP.
   5
   6     VOMP is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     VOMP is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with VOMP.  If not, see <https://www.gnu.org/licenses/>.
  18 */
  19
  20 #ifndef DSOCK_H
  21 #define DSOCK_H
  22
  23 #ifndef WIN32
  24 #include <netinet/in.h>
  25 #include <netdb.h>
  26 #include <sys/socket.h>
  27 #include <unistd.h>
  28 #include <arpa/inet.h>
  29 #include <sys/time.h>
  30 #define SOCKET_ERROR 0
  31 #else
  32 #include <winsock2.h>
  33 #include <Ws2tcpip.h>
  34 #include <sys/timeb.h>
  35 #endif
  36
  37 #include <sys/types.h>
  38 #include <stdio.h>
  39 #include <stdlib.h>
  40 #include <string.h>
  41 #include <errno.h>
  42 #include "defines.h"
  43
  44 #define MAXBUFLEN 2000
  45
  46 class DatagramSocket
  47 {
  48   public:
  49     DatagramSocket(short);             // port
  50     ~DatagramSocket();
  51     int init();
  52     void shutdown();
  53 #ifdef WIN32
  54     unsigned char waitforMessage(unsigned char, SOCKET quitPipe = 0); // uchar =0-block =1-new wait =2-continue wait
  55 #else
  56     unsigned char waitforMessage(unsigned char, int quitPipe = 0); // uchar =0-block =1-new wait =2-continue wait
  57 #endif
  58     UINT getDataLength() const;
  59     const void* getData() const;         // returns a pointer to the data
  60     const char* getFromIPA() const;      // returns a pointer to from IP address
  61     short getFromPort() const;
  62     void send(const char *, short, char *, int); // send wants: IP Address ddn style, port, data, length of data
  63
  64   private:
  65     bool initted{};
  66     ULONG getIPNumber(ULONG num);
  67     ULONG iterate_ip{};
  68     const static char DSOCKDEBUG = 0;
  69     int socketnum;                  // Socket descriptor
  70     short myPort;                   // My port number
  71     struct sockaddr_in myAddr;      // My address
  72     struct sockaddr_in theirAddr;   // User address
  73     socklen_t addrlen;              // length of sockaddr struct
  74     char buf[MAXBUFLEN];            // main data buffer
  75     char fromIPA[20];               // from string (ip address)
  76     short fromPort;                 // which port user sent on
  77     int mlength;                    // length of message
  78     struct timeval tv;
  79     fd_set readfds;
  80 };
  81
  82 #endif