108 lines
2.6 KiB
C++
108 lines
2.6 KiB
C++
/* Copyright: (c) Kayne Ruse 2013
|
|
*
|
|
* This software is provided 'as-is', without any express or implied
|
|
* warranty. In no event will the authors be held liable for any damages
|
|
* arising from the use of this software.
|
|
*
|
|
* Permission is granted to anyone to use this software for any purpose,
|
|
* including commercial applications, and to alter it and redistribute it
|
|
* freely, subject to the following restrictions:
|
|
*
|
|
* 1. The origin of this software must not be misrepresented; you must not
|
|
* claim that you wrote the original software. If you use this software
|
|
* in a product, an acknowledgment in the product documentation would be
|
|
* appreciated but is not required.
|
|
*
|
|
* 2. Altered source versions must be plainly marked as such, and must not be
|
|
* misrepresented as being the original software.
|
|
*
|
|
* 3. This notice may not be removed or altered from any source
|
|
* distribution.
|
|
*/
|
|
#ifndef NETWORK_HPP_
|
|
#define NETWORK_HPP_
|
|
|
|
#include <cstdint>
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#define _WIN32_WINNT 0x501
|
|
#include <winsock2.h>
|
|
#include <ws2tcpip.h>
|
|
|
|
void NetworkInit();
|
|
void NetworkQuit();
|
|
|
|
class TCPSocket {
|
|
public:
|
|
TCPSocket();
|
|
TCPSocket(const char* ip, int port);
|
|
~TCPSocket();
|
|
|
|
/* param 1: ip address to connect to
|
|
* param 2: port to open the server socket on
|
|
*/
|
|
void Open(const char* ip, int port);
|
|
void Close();
|
|
|
|
/* Send() and Receive()
|
|
* param 1: data to be sent/received
|
|
* param 2: length/maxlength of the data
|
|
* param 3: flags to the internal function
|
|
* return:
|
|
* the amount of data sent/received (not necessarily the same value as len/maxlen)
|
|
*/
|
|
int Send(const void* data, int len, int flags = 0);
|
|
int Recv(void* data, int maxlen, int flags = 0);
|
|
private:
|
|
SOCKET sock;
|
|
friend class TCPServerSocket;
|
|
};
|
|
|
|
class TCPServerSocket {
|
|
public:
|
|
TCPServerSocket();
|
|
TCPServerSocket(int port);
|
|
~TCPServerSocket();
|
|
|
|
void Open(int port);
|
|
void Close();
|
|
|
|
int Accept(TCPSocket*, int uSeconds = 0);
|
|
private:
|
|
SOCKET sock;
|
|
};
|
|
|
|
//TODO: Write the UDP systems
|
|
//
|
|
//class UDPRemote {
|
|
//public:
|
|
// UDPRemote();
|
|
// UDPRemote(const char* ip, int port);
|
|
//
|
|
// /* param 1: ip of the remote to connect to, null to clear
|
|
// * param 2: port of the remote to connect to
|
|
// */
|
|
// void Set(const char* ip, int port);
|
|
// //TODO: Get?
|
|
//private:
|
|
// sockaddr addr;
|
|
// friend class UDPSocket;
|
|
//};
|
|
//
|
|
//class UDPSocket {
|
|
//public:
|
|
// UDPSocket();
|
|
// UDPSocket(int port);
|
|
// ~UDPSocket();
|
|
//
|
|
// int Open(int port);
|
|
// void Close();
|
|
//
|
|
// int Send(const void* data, int len, UDPRemote* rem, int flags = 0);
|
|
// int Recv(void* data, int maxlen, UDPRemote* rem, int flags = 0);
|
|
//private:
|
|
// SOCKET sock;
|
|
//};
|
|
|
|
#endif
|