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