492 lines
9.8 KiB
C++
492 lines
9.8 KiB
C++
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <math.h>
|
||
#include <stdarg.h>
|
||
#include "Sys.h"
|
||
|
||
/******************************** Object ********************************/
|
||
|
||
// 输出对象的字符串表示方式
|
||
String& Object::ToStr(String& str) const
|
||
{
|
||
const char* name = typeid(*this).name();
|
||
while(*name >= '0' && *name <= '9') name++;
|
||
|
||
str.Set(name);
|
||
|
||
return str;
|
||
}
|
||
|
||
String Object::ToString() const
|
||
{
|
||
String str;
|
||
ToStr(str);
|
||
|
||
return str;
|
||
}
|
||
|
||
void Object::Show(bool newLine) const
|
||
{
|
||
//String str;
|
||
// 为了减少堆分配，采用较大的栈缓冲区
|
||
char cs[0x200];
|
||
String str(cs, ArrayLength(cs));
|
||
str.SetLength(0);
|
||
ToStr(str);
|
||
str.Show(newLine);
|
||
}
|
||
|
||
/******************************** ByteArray ********************************/
|
||
|
||
// 字符串转为字节数组
|
||
ByteArray::ByteArray(String& str) : Array(0)
|
||
{
|
||
char* p = str.GetBuffer();
|
||
Set((byte*)p, str.Length());
|
||
}
|
||
|
||
// 不允许修改，拷贝
|
||
ByteArray::ByteArray(const String& str) : Array(0)
|
||
{
|
||
char* p = ((String&)str).GetBuffer();
|
||
Copy((byte*)p, str.Length());
|
||
}
|
||
|
||
// 重载等号运算符，使用外部指针、内部长度，用户自己注意安全
|
||
ByteArray& ByteArray::operator=(const byte* data)
|
||
{
|
||
Set(data, Length());
|
||
|
||
return *this;
|
||
}
|
||
|
||
// 显示十六进制数据，指定分隔字符
|
||
String& ByteArray::ToHex(String& str, char sep, int newLine) const
|
||
{
|
||
byte* buf = GetBuffer();
|
||
|
||
// 拼接在字符串后面
|
||
int k = str.Length();
|
||
for(int i=0; i < Length(); i++, buf++)
|
||
{
|
||
byte b = *buf >> 4;
|
||
str.SetAt(k++, b > 9 ? ('A' + b - 10) : ('0' + b));
|
||
b = *buf & 0x0F;
|
||
str.SetAt(k++, b > 9 ? ('A' + b - 10) : ('0' + b));
|
||
|
||
if(i < Length() - 1)
|
||
{
|
||
if(newLine > 0 && (i + 1) % newLine == 0)
|
||
{
|
||
str.SetAt(k++, '\r');
|
||
str.SetAt(k++, '\n');
|
||
}
|
||
else if(sep != '\0')
|
||
str.SetAt(k++, sep);
|
||
}
|
||
}
|
||
//str.SetAt(k, '\0');
|
||
//str.SetLength(k);
|
||
|
||
return str;
|
||
}
|
||
|
||
// 显示十六进制数据，指定分隔字符
|
||
String ByteArray::ToHex(char sep, int newLine) const
|
||
{
|
||
String str;
|
||
|
||
return ToHex(str, sep, newLine);
|
||
}
|
||
|
||
String& ByteArray::ToStr(String& str) const
|
||
{
|
||
return ToHex(str, '-', 0x20);
|
||
}
|
||
|
||
// 显示对象。默认显示ToString
|
||
void ByteArray::Show(bool newLine) const
|
||
{
|
||
/*// 每个字节后面带一个横杠，有换行的时候两个字符，不带横杠
|
||
int len = Length() * 2;
|
||
if(sep != '\0') len += Length();
|
||
if(newLine > 0) len += Length() / newLine;*/
|
||
|
||
// 采用栈分配然后复制，避免堆分配
|
||
char cs[512];
|
||
String str(cs, ArrayLength(cs));
|
||
// 清空字符串，变成0长度，因为ToHex内部是附加
|
||
str.Clear();
|
||
|
||
// 如果需要的缓冲区超过512，那么让它自己分配好了
|
||
ToHex(str, '-', 0x20);
|
||
|
||
str.Show(newLine);
|
||
}
|
||
|
||
/******************************** String ********************************/
|
||
|
||
// 输出对象的字符串表示方式
|
||
String& String::ToStr(String& str) const
|
||
{
|
||
// 把当前字符串复制到目标字符串后面
|
||
str.Copy(*this, str.Length());
|
||
|
||
return (String&)*this;
|
||
}
|
||
|
||
// 输出对象的字符串表示方式
|
||
String String::ToString() const
|
||
{
|
||
return *this;
|
||
}
|
||
|
||
// 清空已存储数据。长度放大到最大容量
|
||
String& String::Clear()
|
||
{
|
||
Array::Clear();
|
||
|
||
_Length = 0;
|
||
|
||
return *this;
|
||
}
|
||
|
||
String& String::Append(char ch)
|
||
{
|
||
Copy(&ch, 1, Length());
|
||
|
||
return *this;
|
||
}
|
||
|
||
String& String::Append(const char* str, int len)
|
||
{
|
||
Copy(str, 0, Length());
|
||
|
||
return *this;
|
||
}
|
||
|
||
char* _itoa(int value, char* string, int radix)
|
||
{
|
||
char tmp[33];
|
||
char* tp = tmp;
|
||
int i;
|
||
unsigned v;
|
||
int sign;
|
||
char* sp;
|
||
if (radix > 36 || radix <= 1) return 0;
|
||
|
||
sign = (radix == 10 && value < 0);
|
||
if (sign)
|
||
v = -value;
|
||
else
|
||
v = (unsigned)value;
|
||
while (v || tp == tmp)
|
||
{
|
||
i = v % radix;
|
||
v = v / radix;
|
||
if (i < 10)
|
||
*tp++ = i+'0';
|
||
else
|
||
*tp++ = i + 'A' - 10;
|
||
}
|
||
|
||
sp = string;
|
||
if (sign)
|
||
*sp++ = '-';
|
||
while (tp > tmp)
|
||
*sp++ = *--tp;
|
||
*sp = 0;
|
||
|
||
return string;
|
||
}
|
||
|
||
String& String::Append(int value)
|
||
{
|
||
char ch[16];
|
||
_itoa(value, ch, 10);
|
||
|
||
return Append(ch);
|
||
}
|
||
|
||
String& String::Append(byte bt)
|
||
{
|
||
int k = Length();
|
||
//SetLength(k + 2);
|
||
|
||
byte b = bt >> 4;
|
||
SetAt(k++, b > 9 ? ('A' + b - 10) : ('0' + b));
|
||
b = bt & 0x0F;
|
||
SetAt(k++, b > 9 ? ('A' + b - 10) : ('0' + b));
|
||
|
||
return *this;
|
||
}
|
||
|
||
String& String::Append(ByteArray& bs)
|
||
{
|
||
//assert_param2(false, "未实现");
|
||
//Copy(bs.ToHex(), Length());
|
||
bs.ToHex(*this);
|
||
|
||
return *this;
|
||
}
|
||
|
||
// 调试输出字符串
|
||
void String::Show(bool newLine) const
|
||
{
|
||
if(!Length()) return;
|
||
|
||
// C格式字符串以0结尾
|
||
char* p = GetBuffer();
|
||
if(!IN_ROM_SECTION(p))
|
||
p[Length()] = 0;
|
||
|
||
debug_printf("%s", GetBuffer());
|
||
if(newLine) debug_printf("\r\n");
|
||
}
|
||
|
||
// 格式化字符串，输出到现有字符串后面。方便我们连续格式化多个字符串
|
||
String& String::Format(const char* format, ...)
|
||
{
|
||
va_list ap;
|
||
|
||
//const char* fmt = format.GetBuffer();
|
||
va_start(ap, format);
|
||
|
||
// 无法准确估计长度，大概乘以2处理
|
||
int len = Length();
|
||
CheckCapacity(len + (strlen(format) << 1), len);
|
||
|
||
char* p = GetBuffer();
|
||
len = vsnprintf(p + len, Capacity() - len, format, ap);
|
||
_Length += len;
|
||
|
||
va_end(ap);
|
||
|
||
return *this;
|
||
}
|
||
|
||
String& String::Concat(const Object& obj)
|
||
{
|
||
//Copy(str, Length());
|
||
//Object& obj2 = (Object&)obj;
|
||
//obj2.ToStr(*this);
|
||
|
||
return obj.ToStr(*this);
|
||
}
|
||
|
||
String& String::Concat(const char* str, int len)
|
||
{
|
||
Copy(str, 0, Length());
|
||
|
||
return *this;
|
||
}
|
||
|
||
String& String::operator+=(const Object& obj)
|
||
{
|
||
return this->Concat(obj);
|
||
}
|
||
|
||
String& String::operator+=(const char* str)
|
||
{
|
||
return this->Concat(str);
|
||
}
|
||
|
||
String& operator+(String& str1, const Object& obj)
|
||
{
|
||
return str1.Concat(obj);
|
||
}
|
||
|
||
String& operator+(String& str1, const char* str2)
|
||
{
|
||
return str1.Concat(str2);
|
||
}
|
||
|
||
/*String operator+(const char* str1, const char* str2)
|
||
{
|
||
String str(str1);
|
||
return str + str2;
|
||
}*/
|
||
|
||
String operator+(const char* str, const Object& obj)
|
||
{
|
||
String s;
|
||
s = str;
|
||
s += obj;
|
||
return s;
|
||
}
|
||
|
||
String operator+(const Object& obj, const char* str)
|
||
{
|
||
String s;
|
||
obj.ToStr(s);
|
||
s += str;
|
||
return s;
|
||
}
|
||
|
||
/******************************** IPAddress ********************************/
|
||
/* IP地址 */
|
||
|
||
const IPAddress IPAddress::Any(0, 0, 0, 0);
|
||
const IPAddress IPAddress::Broadcast(255, 255, 255, 255);
|
||
|
||
IPAddress::IPAddress(byte ip1, byte ip2, byte ip3, byte ip4)
|
||
{
|
||
Value = (ip4 << 24) + (ip3 << 16) + (ip2 << 8) + ip1;
|
||
}
|
||
|
||
bool IPAddress::IsAny() const { return Value == 0; }
|
||
|
||
bool IPAddress::IsBroadcast() const { return Value == 0xFFFFFFFF; }
|
||
|
||
uint IPAddress::GetSubNet(const IPAddress& mask) const
|
||
{
|
||
return Value & mask.Value;
|
||
}
|
||
|
||
// 重载索引运算符[]，让它可以像数组一样使用下标索引。
|
||
byte& IPAddress::operator[](int i)
|
||
{
|
||
assert_param(i >= 0 && i < 4);
|
||
|
||
return ((byte*)&Value)[i];
|
||
}
|
||
|
||
// 字节数组
|
||
byte* IPAddress::ToArray() const
|
||
{
|
||
return (byte*)&Value;
|
||
}
|
||
|
||
String& IPAddress::ToStr(String& str) const
|
||
{
|
||
byte* ips = (byte*)&Value;
|
||
|
||
str.Format("%d.%d.%d.%d", ips[0], ips[1], ips[2], ips[3]);
|
||
|
||
return str;
|
||
}
|
||
|
||
/*void IPAddress::Show()
|
||
{
|
||
byte* ips = (byte*)&Value;
|
||
debug_printf("%d.%d.%d.%d", ips[0], ips[1], ips[2], ips[3]);
|
||
}*/
|
||
|
||
/******************************** IPEndPoint ********************************/
|
||
|
||
const IPEndPoint IPEndPoint::Any(IPAddress::Any, 0);
|
||
|
||
IPEndPoint::IPEndPoint()
|
||
{
|
||
Address = IPAddress::Any;
|
||
Port = 0;
|
||
}
|
||
|
||
IPEndPoint::IPEndPoint(const IPAddress& addr, ushort port)
|
||
{
|
||
Address = addr;
|
||
Port = port;
|
||
}
|
||
|
||
String& IPEndPoint::ToStr(String& str) const
|
||
{
|
||
//str = Address.ToString();
|
||
Address.ToStr(str);
|
||
|
||
char ss[7];
|
||
int len = sprintf(ss, ":%d", Port);
|
||
str.Copy(ss, len, str.Length());
|
||
|
||
return str;
|
||
}
|
||
|
||
/*void IPEndPoint::Show()
|
||
{
|
||
byte* ips = (byte*)&Address.Value;
|
||
debug_printf("%d.%d.%d.%d:%d", ips[0], ips[1], ips[2], ips[3], Port);
|
||
}*/
|
||
|
||
bool operator==(const IPEndPoint& addr1, const IPEndPoint& addr2)
|
||
{
|
||
return addr1.Port == addr2.Port && addr1.Address == addr2.Address;
|
||
}
|
||
bool operator!=(const IPEndPoint& addr1, const IPEndPoint& addr2)
|
||
{
|
||
return addr1.Port != addr2.Port || addr1.Address != addr2.Address;
|
||
}
|
||
|
||
/******************************** MacAddress ********************************/
|
||
/* MAC地址 */
|
||
|
||
#define MAC_MASK 0xFFFFFFFFFFFFull
|
||
|
||
const MacAddress MacAddress::Empty(0);
|
||
const MacAddress MacAddress::Full(MAC_MASK);
|
||
|
||
MacAddress::MacAddress(ulong v)
|
||
{
|
||
//v4 = v;
|
||
//v2 = v >> 32;
|
||
Value = v & MAC_MASK;
|
||
}
|
||
|
||
// 是否广播地址，全0或全1
|
||
bool MacAddress::IsBroadcast() const { return Value == Empty.Value || Value == Full.Value; }
|
||
|
||
MacAddress& MacAddress::operator=(ulong v)
|
||
{
|
||
//v4 = v;
|
||
//v2 = v >> 32;
|
||
|
||
// 下面这个写法很好，一条汇编指令即可完成，但是会覆盖当前结构体后两个字节
|
||
//*(ulong*)this = v;
|
||
|
||
// 下面的写法需要5条汇编指令，先放入内存，再分两次读写
|
||
/*uint* p = (uint*)&v;
|
||
v4 = *p++;
|
||
v2 = *(ushort*)p;*/
|
||
|
||
Value = v & MAC_MASK;
|
||
|
||
return *this;
|
||
}
|
||
|
||
// 重载索引运算符[]，让它可以像数组一样使用下标索引。
|
||
byte& MacAddress::operator[](int i)
|
||
{
|
||
assert_param(i >= 0 && i < 6);
|
||
|
||
return ((byte*)&Value)[i];
|
||
}
|
||
|
||
// 字节数组
|
||
byte* MacAddress::ToArray() const
|
||
{
|
||
return (byte*)&Value;
|
||
}
|
||
|
||
/*bool MacAddress::operator==(MacAddress& addr1, MacAddress& addr2)
|
||
{
|
||
return addr1.v4 == addr2.v4 && addr1.v2 == addr2.v2;
|
||
}
|
||
|
||
bool MacAddress::operator!=(MacAddress& addr1, MacAddress& addr2)
|
||
{
|
||
return addr1.v4 != addr2.v4 || addr1.v2 != addr2.v2;
|
||
}*/
|
||
|
||
String& MacAddress::ToStr(String& str) const
|
||
{
|
||
byte* macs = (byte*)&Value;
|
||
|
||
str.Format("%02X-%02X-%02X-%02X-%02X-%02X", macs[0], macs[1], macs[2], macs[3], macs[4], macs[5]);
|
||
|
||
return str;
|
||
}
|
||
|
||
/*void MacAddress::Show()
|
||
{
|
||
byte* macs = (byte*)&Value;
|
||
debug_printf("%02X-%02X-%02X-%02X-%02X-%02X", macs[0], macs[1], macs[2], macs[3], macs[4], macs[5]);
|
||
}*/
|