98 lines
2.5 KiB
C++
98 lines
2.5 KiB
C++
/*
|
|
Copyright (c) 2013 yvt, learn_more
|
|
|
|
This file is part of OpenSpades.
|
|
|
|
OpenSpades is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
OpenSpades is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with OpenSpades. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
#include <enet/enet.h>
|
|
#include <regex>
|
|
|
|
#include "ServerAddress.h"
|
|
|
|
namespace spades {
|
|
|
|
// unsigned long version of atol, cross-platform (including MSVC)
|
|
uint32_t ServerAddress::ParseIntegerAddress(const std::string &str) {
|
|
uint32_t vl = 0;
|
|
for (size_t i = 0; i < str.size(); i++) {
|
|
char c = str[i];
|
|
if (c >= '0' && c <= '9') {
|
|
vl *= 10;
|
|
vl += (uint32_t)(c - '0');
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
|
|
return vl;
|
|
}
|
|
|
|
ServerAddress::ServerAddress(std::string address, ProtocolVersion version)
|
|
: mAddress(address), mVersion(version) {
|
|
static std::regex v075regex {"(.*):0?\\.?75"};
|
|
static std::regex v076regex {"(.*):0?\\.?76"};
|
|
|
|
std::smatch matchResult;
|
|
|
|
if (std::regex_match(address, matchResult, v075regex)) {
|
|
version = ProtocolVersion::v075;
|
|
mAddress = matchResult[1];
|
|
} else if (std::regex_match(address, matchResult, v076regex)) {
|
|
version = ProtocolVersion::v076;
|
|
mAddress = matchResult[1];
|
|
}
|
|
}
|
|
|
|
std::string ServerAddress::StripProtocol(const std::string &address) {
|
|
if (address.find("aos:///") == 0) {
|
|
return address.substr(7);
|
|
} else if (address.find("aos://") == 0) {
|
|
return address.substr(6);
|
|
}
|
|
return address;
|
|
}
|
|
|
|
ENetAddress ServerAddress::GetENetAddress() const {
|
|
std::string address = StripProtocol(mAddress);
|
|
|
|
ENetAddress addr;
|
|
size_t pos = address.find(':');
|
|
if (pos == std::string::npos) {
|
|
// addr = address;
|
|
addr.port = 32887;
|
|
} else {
|
|
addr.port = atoi(address.substr(pos + 1).c_str());
|
|
address = address.substr(0, pos);
|
|
}
|
|
|
|
if (address.find('.') != std::string::npos) {
|
|
enet_address_set_host(&addr, address.c_str());
|
|
} else {
|
|
addr.host = ParseIntegerAddress(address);
|
|
}
|
|
return addr;
|
|
}
|
|
|
|
std::string ServerAddress::ToString(bool includeProtocol) const {
|
|
if (!includeProtocol) {
|
|
return StripProtocol(mAddress);
|
|
}
|
|
return mAddress;
|
|
}
|
|
|
|
}; // namespace spades
|