void DS::Stream::writeSafeString(const ST::string& value, DS::StringType format)
{
if (format == e_StringUTF16) {
ST::utf16_buffer buffer = value.to_utf16();
uint16_t length = value.size() & 0x0FFF;
ST::utf16_buffer work;
char16_t* data = work.create_writable_buffer(length);
memcpy(data, buffer.data(), length * sizeof(char16_t));
for (uint16_t i=0; i<length; ++i)
data[i] = ~data[i];
write<uint16_t>(length | 0xF000);
writeBytes(data, length * sizeof(char16_t));
write<char16_t>(0);
} else {
ST::char_buffer buffer = (format == e_StringUTF8) ? value.to_utf8()
: value.to_latin_1(ST::substitute_invalid);
uint16_t length = value.size() & 0x0FFF;
ST::char_buffer work;
char* data = work.create_writable_buffer(length);
memcpy(data, buffer.data(), length * sizeof(char));
for (uint16_t i=0; i<length; ++i)
data[i] = ~data[i];
write<uint16_t>(length | 0xF000);
writeBytes(data, length * sizeof(char));
}
}
ST::string DS::Stream::readSafeString(DS::StringType format)
{
uint16_t length = read<uint16_t>();
if (!(length & 0xF000))
read<uint16_t>(); // Discarded
length &= 0x0FFF;
if (format == e_StringUTF16) {
ST::utf16_buffer result;
char16_t* buffer = result.create_writable_buffer(length);
ssize_t bytes = readBytes(buffer, length * sizeof(char16_t));
read<char16_t>(); // redundant u'\0'
DS_DASSERT(bytes == static_cast<ssize_t>(length * sizeof(char16_t)));
if (length && (buffer[0] & 0x8000)) {
for (uint16_t i=0; i<length; ++i)
buffer[i] = ~buffer[i];
}
buffer[length] = 0;
return ST::string::from_utf16(result, ST::substitute_invalid);
} else {
ST::char_buffer result;
char* buffer = result.create_writable_buffer(length);
ssize_t bytes = readBytes(buffer, length * sizeof(char));
DS_DASSERT(bytes == static_cast<ssize_t>(length * sizeof(char)));
if (length && (buffer[0] & 0x80)) {
for (uint16_t i=0; i<length; ++i)
buffer[i] = ~buffer[i];
}
buffer[length] = 0;
return (format == e_StringUTF8)
? ST::string::from_utf8(result, ST::substitute_invalid)
: ST::string::from_latin_1(result);
}
}
void DS::Stream::writeString(const ST::string& value, DS::StringType format)
{
if (format == e_StringUTF16) {
ST::utf16_buffer buffer = value.to_utf16();
writeBytes(buffer.data(), buffer.size() * sizeof(char16_t));
} else {
ST::char_buffer buffer = (format == e_StringUTF8) ? value.to_utf8()
: value.to_latin_1(ST::substitute_invalid);
writeBytes(buffer.data(), buffer.size() * sizeof(char));
}
}
void cb_authServIpAddress(GateKeeper_Private& client)
{
START_REPLY(e_GateKeeperToCli_AuthServIpAddressReply);
// Trans ID
client.m_buffer.write<uint32_t>(DS::CryptRecvValue<uint32_t>(client.m_sock, client.m_crypt));
// Address
ST::utf16_buffer address = DS::Settings::AuthServerAddress();
client.m_buffer.write<uint16_t>(address.size());
client.m_buffer.writeBytes(address.data(), address.size() * sizeof(char16_t));
SEND_REPLY();
}
ST::string pnSocket::recvString(size_t maxlen)
{
uint16_t size;
if (recv(&size, sizeof(uint16_t)) <= 0)
size = 0;
if (size >= maxlen)
size = maxlen-1;
if (size > 0) {
ST::utf16_buffer str;
char16_t* buf = str.create_writable_buffer(size);
recv(buf, size * sizeof(char16_t));
return ST::string::from_utf16(str);
} else {
return ST::string();
}
}
ST::string DS::Stream::readString(size_t length, DS::StringType format)
{
if (format == e_StringUTF16) {
ST::utf16_buffer result;
char16_t* buffer = result.create_writable_buffer(length);
ssize_t bytes = readBytes(buffer, length * sizeof(char16_t));
DS_DASSERT(bytes == static_cast<ssize_t>(length * sizeof(char16_t)));
buffer[length] = 0;
return ST::string::from_utf16(result, ST::substitute_invalid);
} else {
ST::char_buffer result;
char* buffer = result.create_writable_buffer(length);
ssize_t bytes = readBytes(buffer, length * sizeof(char));
DS_DASSERT(bytes == static_cast<ssize_t>(length * sizeof(char)));
buffer[length] = 0;
return (format == e_StringUTF8) ? ST::string::from_utf8(result, ST::substitute_invalid)
: ST::string::from_latin_1(result);
}
}
