mirror of
https://github.com/mod-playerbots/azerothcore-wotlk.git
synced 2025-11-29 17:38:24 +08:00
165 lines
3.3 KiB
C++
165 lines
3.3 KiB
C++
/**
|
|
@file NetMessage.cpp
|
|
|
|
@maintainer Morgan McGuire, morgan@cs.brown.edu
|
|
@created 2005-02-06
|
|
@edited 2005-02-06
|
|
*/
|
|
#include "G3D/platform.h"
|
|
#include "G3D/NetAddress.h"
|
|
#include "G3D/BinaryInput.h"
|
|
#include "G3D/BinaryOutput.h"
|
|
#include "G3D/Array.h"
|
|
#include "G3D/stringutils.h"
|
|
#include "G3D/System.h"
|
|
#include "G3D/NetworkDevice.h"
|
|
|
|
#if defined(G3D_LINUX) || defined(G3D_OSX)
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
#include <netdb.h>
|
|
#include <netinet/tcp.h>
|
|
#define _alloca alloca
|
|
|
|
# ifndef SOCKADDR_IN
|
|
# define SOCKADDR_IN struct sockaddr_in
|
|
# endif
|
|
# ifndef SOCKET
|
|
# define SOCKET int
|
|
# endif
|
|
|
|
// SOCKADDR_IN is supposed to be defined in NetAddress.h
|
|
#ifndef SOCKADDR_IN
|
|
# error Network headers included in wrong order
|
|
#endif
|
|
#endif
|
|
|
|
|
|
namespace G3D {
|
|
|
|
NetAddress::NetAddress() {
|
|
System::memset(&addr, 0, sizeof(addr));
|
|
}
|
|
|
|
void NetAddress::init(uint32 host, uint16 port) {
|
|
if ((host != 0) || (port != 0)) {
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_port = htons(port);
|
|
if (host == 0) {
|
|
host = INADDR_ANY;
|
|
}
|
|
addr.sin_addr.s_addr = htonl(host);
|
|
} else {
|
|
System::memset(&addr, 0, sizeof(addr));
|
|
}
|
|
}
|
|
|
|
|
|
NetAddress::NetAddress(
|
|
const std::string& hostname,
|
|
uint16 port) {
|
|
init(hostname, port);
|
|
}
|
|
|
|
|
|
void NetAddress::init(
|
|
const std::string& hostname,
|
|
uint16 port) {
|
|
|
|
uint32 addr;
|
|
|
|
if (hostname == "") {
|
|
addr = INADDR_NONE;
|
|
} else {
|
|
addr = inet_addr(hostname.c_str());
|
|
}
|
|
|
|
// The address wasn't in numeric form, resolve it
|
|
if (addr == INADDR_NONE) {
|
|
// Get the IP address of the server and store it in host
|
|
struct hostent* host = gethostbyname(hostname.c_str());
|
|
|
|
if (host == NULL) {
|
|
init(0, 0);
|
|
return;
|
|
}
|
|
|
|
System::memcpy(&addr, host->h_addr_list[0], host->h_length);
|
|
}
|
|
|
|
if (addr != INADDR_NONE) {
|
|
addr = ntohl(addr);
|
|
}
|
|
init(addr, port);
|
|
}
|
|
|
|
|
|
NetAddress::NetAddress(uint32 hostip, uint16 port) {
|
|
init(hostip, port);
|
|
}
|
|
|
|
|
|
NetAddress NetAddress::broadcastAddress(uint16 port) {
|
|
return NetAddress(NetworkDevice::instance()->broadcastAddressArray()[0], port);
|
|
}
|
|
|
|
|
|
NetAddress::NetAddress(const std::string& hostnameAndPort) {
|
|
|
|
Array<std::string> part = stringSplit(hostnameAndPort, ':');
|
|
|
|
debugAssert(part.length() == 2);
|
|
init(part[0], atoi(part[1].c_str()));
|
|
}
|
|
|
|
|
|
NetAddress::NetAddress(const SOCKADDR_IN& a) {
|
|
addr = a;
|
|
}
|
|
|
|
|
|
NetAddress::NetAddress(const struct in_addr& addr, uint16 port) {
|
|
#ifdef G3D_WIN32
|
|
init(ntohl(addr.S_un.S_addr), port);
|
|
#else
|
|
init(htonl(addr.s_addr), port);
|
|
#endif
|
|
}
|
|
|
|
|
|
void NetAddress::serialize(class BinaryOutput& b) const {
|
|
b.writeUInt32(ip());
|
|
b.writeUInt16(port());
|
|
}
|
|
|
|
|
|
void NetAddress::deserialize(class BinaryInput& b) {
|
|
uint32 i;
|
|
uint16 p;
|
|
|
|
i = b.readUInt32();
|
|
p = b.readUInt16();
|
|
|
|
init(i, p);
|
|
}
|
|
|
|
|
|
bool NetAddress::ok() const {
|
|
return addr.sin_family != 0;
|
|
}
|
|
|
|
|
|
std::string NetAddress::ipString() const {
|
|
return format("%s", inet_ntoa(*(in_addr*)&(addr.sin_addr)));
|
|
}
|
|
|
|
|
|
std::string NetAddress::toString() const {
|
|
return ipString() + format(":%d", ntohs(addr.sin_port));
|
|
}
|
|
|
|
}
|