dsock.h

   1 /*
   2     Copyright 2004-2005 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, write to the Free Software
  18     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19 */
  20
  21 #ifndef DSOCK_H
  22 #define DSOCK_H
  23
  24 #ifndef WIN32
  25 #include <netinet/in.h>
  26 #include <netdb.h>
  27 #include <sys/socket.h>
  28 #include <unistd.h>
  29 #include <arpa/inet.h>
  30 #include <sys/time.h>
  31 #define SOCKET_ERROR 0
  32 #else
  33 #include <winsock2.h>
  34 #include <Ws2tcpip.h>
  35 #include <sys/timeb.h>
  36 #endif
  37
  38 #include <sys/types.h>
  39 #include <stdio.h>
  40 #include <stdlib.h>
  41 #include <string.h>
  42 #include <errno.h>
  43 #include "defines.h"
  44
  45 #define MAXBUFLEN 2000
  46
  47 class DatagramSocket
  48 {
  49   public:
  50     DatagramSocket(short);             // port
  51     ~DatagramSocket();
  52     int init();
  53     void shutdown();
  54     unsigned char waitforMessage(unsigned char); // int =0-block =1-new wait =2-continue wait
  55     int getDataLength(void) const;
  56     char *getData(void);               // returns a pointer to the data
  57     char *getFromIPA(void);            // returns a pointer to from IP address
  58     short getFromPort(void) const;
  59     void send(char *, short, char *, int); // send wants: IP Address ddn style, port,
  60                                            // data, length of data
  61   private:
  62     bool initted;
  63     ULONG getIPNumber(ULONG num);
  64     ULONG iterate_ip;
  65     const static char DSOCKDEBUG = 0;
  66     int socketnum;                  // Socket descriptor
  67     short myPort;                   // My port number
  68     struct sockaddr_in myAddr;      // My address
  69     struct sockaddr_in theirAddr;   // User address
  70     socklen_t addrlen;              // length of sockaddr struct
  71     char buf[MAXBUFLEN];            // main data buffer
  72     char fromIPA[20];               // from string (ip address)
  73     short fromPort;                 // which port user sent on
  74     int mlength;                    // length of message
  75     struct timeval tv;
  76     fd_set readfds;
  77 };
  78
  79 #endif