Changed code to support Windows libraries and functions as well
This commit is contained in:
21
easysock.cpp
21
easysock.cpp
@@ -5,7 +5,7 @@
|
|||||||
#include <cerrno>
|
#include <cerrno>
|
||||||
#include <fcntl.h>
|
#include <fcntl.h>
|
||||||
|
|
||||||
#ifndef _WIN_32
|
#ifndef _WIN32
|
||||||
const int INVALID_SOCKET = -1;
|
const int INVALID_SOCKET = -1;
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
@@ -27,10 +27,8 @@ int sock_quit(void) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* Function to create a socket - Accepts IP version(4 or 6), protocol
|
/* Function to create a socket - Accepts IP version(4 or 6), protocol
|
||||||
type (PROTO_TCP or PROTO_UDP) and a flag to indicate whether the socket
|
type (PROTO_TCP or PROTO_UDP). */
|
||||||
should be set in blocking mode or not. This flag is ONLY FOR TCP. It does
|
SOCKET create_socket(int network, char transport) {
|
||||||
nothing if the protocol is UDP.*/
|
|
||||||
int create_socket(int network, char transport, bool is_blocking) {
|
|
||||||
sock_init();
|
sock_init();
|
||||||
int domain;
|
int domain;
|
||||||
int type;
|
int type;
|
||||||
@@ -57,11 +55,6 @@ int create_socket(int network, char transport, bool is_blocking) {
|
|||||||
int set_opt = 1;
|
int set_opt = 1;
|
||||||
setsockopt(newSock, SOL_SOCKET, SO_REUSEADDR, (char *)&set_opt, sizeof(set_opt));
|
setsockopt(newSock, SOL_SOCKET, SO_REUSEADDR, (char *)&set_opt, sizeof(set_opt));
|
||||||
|
|
||||||
if (is_blocking && transport == ES_TCP) {
|
|
||||||
int flags = fcntl(newSock, F_GETFL);
|
|
||||||
flags |= O_NONBLOCK;
|
|
||||||
fcntl(newSock,F_SETFL,flags);
|
|
||||||
}
|
|
||||||
return newSock;
|
return newSock;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -90,8 +83,8 @@ int create_addr(int network, char* address, int port,struct sockaddr* dest) {
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
int create_local (int network, char transport, char* address, int port,struct sockaddr* addr_struct, bool is_blocking) {
|
SOCKET create_local (int network, char transport, char* address, int port,struct sockaddr* addr_struct) {
|
||||||
int socket = create_socket(network,transport, is_blocking);
|
int socket = create_socket(network,transport);
|
||||||
if (socket < 0) {
|
if (socket < 0) {
|
||||||
return (-1 * errno);
|
return (-1 * errno);
|
||||||
}
|
}
|
||||||
@@ -116,7 +109,7 @@ int create_local (int network, char transport, char* address, int port,struct so
|
|||||||
return socket;
|
return socket;
|
||||||
}
|
}
|
||||||
|
|
||||||
int create_remote (int network,char transport,char* address,int port,struct sockaddr* remote_addr_struct, bool is_blocking) {
|
SOCKET create_remote (int network,char transport,char* address,int port,struct sockaddr* remote_addr_struct) {
|
||||||
|
|
||||||
struct addrinfo hints; /* Used to tell getaddrinfo what kind of address we want */
|
struct addrinfo hints; /* Used to tell getaddrinfo what kind of address we want */
|
||||||
struct addrinfo* results; /* Used by getaddrinfo to store the addresses */
|
struct addrinfo* results; /* Used by getaddrinfo to store the addresses */
|
||||||
@@ -142,7 +135,7 @@ int create_remote (int network,char transport,char* address,int port,struct sock
|
|||||||
create_addr(network,address,port,remote_addr_struct);
|
create_addr(network,address,port,remote_addr_struct);
|
||||||
}
|
}
|
||||||
|
|
||||||
int socket = create_socket(network,transport, is_blocking);
|
int socket = create_socket(network,transport);
|
||||||
if (socket < 0) {
|
if (socket < 0) {
|
||||||
return (-1 * errno);
|
return (-1 * errno);
|
||||||
}
|
}
|
||||||
|
@@ -1,10 +1,12 @@
|
|||||||
#ifndef EASYSOCK_HPP_
|
#ifndef EASYSOCK_HPP_
|
||||||
#define EASYSOCK_HPP_
|
#define EASYSOCK_HPP_
|
||||||
|
|
||||||
#ifdef _WIN_32
|
#ifdef _WIN32
|
||||||
#include <winsock2.h>
|
#include <winsock2.h>
|
||||||
#include <Ws2tcpip.h>
|
#include <winsock.h>
|
||||||
#else
|
#include <ws2tcpip.h>
|
||||||
|
#endif
|
||||||
|
#ifdef linux
|
||||||
#include <sys/types.h>
|
#include <sys/types.h>
|
||||||
#include <sys/socket.h>
|
#include <sys/socket.h>
|
||||||
#include <netdb.h>
|
#include <netdb.h>
|
||||||
@@ -12,7 +14,7 @@
|
|||||||
#include <arpa/inet.h>
|
#include <arpa/inet.h>
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifndef _WIN_32
|
#ifndef _WIN32
|
||||||
typedef int SOCKET;
|
typedef int SOCKET;
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
@@ -25,13 +27,12 @@ const char ES_UDP = 'U';
|
|||||||
a layer 3 - network layer - integer, which must be '4' for IPv4
|
a layer 3 - network layer - integer, which must be '4' for IPv4
|
||||||
and 6 for IPv6;
|
and 6 for IPv6;
|
||||||
a layer 4 - transport layer - character, which must be 'T' for
|
a layer 4 - transport layer - character, which must be 'T' for
|
||||||
TCP or 'U' for UDP; and
|
TCP or 'U' for UDP.
|
||||||
a bool that indicates whether the socket should be blocking or non-blocking.
|
|
||||||
|
|
||||||
|
It creates a _blocking_ socket, and returns the created socket, or -1
|
||||||
|
if the socket creation failed.*/
|
||||||
|
|
||||||
It returns the created socket, or -1 if the socket creation failed.*/
|
SOCKET create_socket(int network, char transport);
|
||||||
|
|
||||||
SOCKET create_socket(int network, char transport, bool is_blocking = false);
|
|
||||||
|
|
||||||
|
|
||||||
/* This function fills in the sockaddr struct 'dest' based on the given information.
|
/* This function fills in the sockaddr struct 'dest' based on the given information.
|
||||||
@@ -43,7 +44,7 @@ and dest is a pointer to the sockaddr struct that will be filled in.
|
|||||||
The function returns with -202 if the network parameter contained neither '4'
|
The function returns with -202 if the network parameter contained neither '4'
|
||||||
nor '6'. */
|
nor '6'. */
|
||||||
|
|
||||||
SOCKET create_addr(int network, char* address, int port,struct sockaddr* dest);
|
int create_addr(int network, char* address, int port,struct sockaddr* dest);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@@ -53,7 +54,7 @@ same as above.
|
|||||||
|
|
||||||
It prints the error returned by 'bind' if something went wrong, and returns ( -1 * errno ).*/
|
It prints the error returned by 'bind' if something went wrong, and returns ( -1 * errno ).*/
|
||||||
|
|
||||||
SOCKET create_local (int network, char transport, char* address, int port,struct sockaddr* addr_struct, bool is_blocking = false);
|
SOCKET create_local (int network, char transport, char* address, int port,struct sockaddr* addr_struct);
|
||||||
|
|
||||||
|
|
||||||
/* This function utilizes the same functions as 'create_local' but _connects_ to the
|
/* This function utilizes the same functions as 'create_local' but _connects_ to the
|
||||||
@@ -62,7 +63,7 @@ as above. This function needs an empty 'sockaddr *' structure passed to it, whic
|
|||||||
|
|
||||||
If something goes wrong, this function returns with ( -1 * errno ). */
|
If something goes wrong, this function returns with ( -1 * errno ). */
|
||||||
|
|
||||||
SOCKET create_remote (int network,char transport,char* address,int port,struct sockaddr* remote_addr_struct, bool is_blocking = false);
|
SOCKET create_remote (int network,char transport,char* address,int port,struct sockaddr* remote_addr_struct);
|
||||||
|
|
||||||
/* check_ip_ver - This function checks if the given string is an IPv4 address (returns 4),
|
/* check_ip_ver - This function checks if the given string is an IPv4 address (returns 4),
|
||||||
IPv6 address (returns 6) or neither (returns -1). */
|
IPv6 address (returns 6) or neither (returns -1). */
|
||||||
|
@@ -2,7 +2,13 @@
|
|||||||
#define _SOCK_CLASS
|
#define _SOCK_CLASS
|
||||||
|
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <sys/socket.h>
|
#ifdef linux
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#endif
|
||||||
|
#ifdef _WIN32
|
||||||
|
#include <winsock2.h>
|
||||||
|
#include <ws2tcpip.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
/* Global constants - can be used by children classes as return values, and by any clients to check what type the socket is of */
|
/* Global constants - can be used by children classes as return values, and by any clients to check what type the socket is of */
|
||||||
const int SOCK_CLIENT = 'C';
|
const int SOCK_CLIENT = 'C';
|
||||||
|
15
main.cpp
15
main.cpp
@@ -1,3 +1,18 @@
|
|||||||
|
#if defined(_WIN32)
|
||||||
|
#define NOGDI // All GDI defines and routines
|
||||||
|
#define NOUSER // All USER defines and routines
|
||||||
|
#define WIN32_LEAN_AND_MEAN
|
||||||
|
#include <windows.h> // or any library that uses Windows.h
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#if defined(_WIN32) // raylib uses these names as function parameters
|
||||||
|
#undef near
|
||||||
|
#undef far
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <cmath>
|
#include <cmath>
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
|
@@ -1,9 +1,15 @@
|
|||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include <arpa/inet.h>
|
#ifdef linux
|
||||||
|
#include <arpa/inet.h>
|
||||||
|
#endif
|
||||||
|
#ifdef _WIN32
|
||||||
|
#include <winsock2.h>
|
||||||
|
#endif
|
||||||
#include "includes/serialization.h"
|
#include "includes/serialization.h"
|
||||||
|
|
||||||
|
|
||||||
/* Takes in float values, casts them to uint16_t and creates a Serial_Data struct */
|
/* Takes in float values, casts them to uint16_t and creates a Serial_Data struct */
|
||||||
Serial_Data Serial_create_data(float pad_x, float pad_y, float ball_x, float ball_y) {
|
Serial_Data Serial_create_data(float pad_x, float pad_y, float ball_x, float ball_y) {
|
||||||
Serial_Data data;
|
Serial_Data data;
|
||||||
|
@@ -1,4 +1,9 @@
|
|||||||
#include <sys/socket.h>
|
#ifdef linux
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#endif
|
||||||
|
#ifdef _WIN32
|
||||||
|
#include <winsock2.h>
|
||||||
|
#endif
|
||||||
#include <fcntl.h>
|
#include <fcntl.h>
|
||||||
#include "includes/sock.hpp"
|
#include "includes/sock.hpp"
|
||||||
#include "includes/server.hpp"
|
#include "includes/server.hpp"
|
||||||
@@ -15,7 +20,6 @@ Server::~Server() {
|
|||||||
/* Sends given data through the peer socket - This method is overriden from the
|
/* Sends given data through the peer socket - This method is overriden from the
|
||||||
base method, because a different socket must be used. In the server's case, the
|
base method, because a different socket must be used. In the server's case, the
|
||||||
'peer' socket i.e. the socket returned after calling 'accept', must be used. */
|
'peer' socket i.e. the socket returned after calling 'accept', must be used. */
|
||||||
|
|
||||||
void Server::sendAll(std::string to_send) {
|
void Server::sendAll(std::string to_send) {
|
||||||
Sock::sendAll(to_send);
|
Sock::sendAll(to_send);
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user