src/socket.h

   1 /*
   2 Minetest-c55
   3 Copyright (C) 2010 celeron55, Perttu Ahola <celeron55@gmail.com>
   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 #ifndef SOCKET_HEADER
  21 #define SOCKET_HEADER
  22
  23 #ifdef _WIN32
  24         #define WIN32_LEAN_AND_MEAN
  25         // Without this some of the network functions are not found on mingw
  26         #ifndef _WIN32_WINNT
  27                 #define _WIN32_WINNT 0x0501
  28         #endif
  29         #include <windows.h>
  30         #include <winsock2.h>
  31         #include <ws2tcpip.h>
  32         #ifdef _MSC_VER
  33                 #pragma comment(lib, "wsock32.lib")
  34         #endif
  35 typedef SOCKET socket_t;
  36 typedef int socklen_t;
  37 #else
  38         #include <sys/socket.h>
  39         #include <netinet/in.h>
  40         #include <fcntl.h>
  41         #include <netdb.h>
  42         #include <unistd.h>
  43 typedef int socket_t;
  44 #endif
  45
  46 #include <ostream>
  47 #include "exceptions.h"
  48 #include "constants.h"
  49
  50 class SocketException : public BaseException
  51 {
  52 public:
  53         SocketException(const char *s):
  54                 BaseException(s)
  55         {
  56         }
  57 };
  58
  59 class ResolveError : public BaseException
  60 {
  61 public:
  62         ResolveError(const char *s):
  63                 BaseException(s)
  64         {
  65         }
  66 };
  67
  68 class SendFailedException : public BaseException
  69 {
  70 public:
  71         SendFailedException(const char *s):
  72                 BaseException(s)
  73         {
  74         }
  75 };
  76
  77 void sockets_init();
  78 void sockets_cleanup();
  79
  80 class Address
  81 {
  82 public:
  83         Address();
  84         Address(unsigned int address, unsigned short port);
  85         Address(unsigned int a, unsigned int b,
  86                         unsigned int c, unsigned int d,
  87                         unsigned short port);
  88         bool operator==(Address &address);
  89         bool operator!=(Address &address);
  90         void Resolve(const char *name);
  91         unsigned int getAddress() const;
  92         unsigned short getPort() const;
  93         void setAddress(unsigned int address);
  94         void setAddress(unsigned int a, unsigned int b,
  95                         unsigned int c, unsigned int d);
  96         void setPort(unsigned short port);
  97         void print(std::ostream *s) const;
  98         void print() const;
  99 private:
 100         unsigned int m_address;
 101         unsigned short m_port;
 102 };
 103
 104 class UDPSocket
 105 {
 106 public:
 107         UDPSocket();
 108         ~UDPSocket();
 109         void Bind(unsigned short port);
 110         //void Close();
 111         //bool IsOpen();
 112         void Send(const Address & destination, const void * data, int size);
 113         // Returns -1 if there is no data
 114         int Receive(Address & sender, void * data, int size);
 115         int GetHandle(); // For debugging purposes only
 116         void setTimeoutMs(int timeout_ms);
 117         // Returns true if there is data, false if timeout occurred
 118         bool WaitData(int timeout_ms);
 119 private:
 120         int m_handle;
 121         int m_timeout_ms;
 122 };
 123
 124 #endif
 125