static void DiscardWire(WireType type, Stream stream)
{
switch (type)
{
case WireType.VARINT:
Varint.FromBase128(stream);
break;
case WireType.FIXED_64:
stream.Seek(8, SeekOrigin.Current);
break;
case WireType.LENGTH_DELIMITED:
ulong length = Varint.FromBase128(stream);
stream.Seek((long)length, SeekOrigin.Current);
break;
case WireType.FIXED_32:
stream.Seek(4, SeekOrigin.Current);
break;
default:
break;
}
}
static object ParseSingleVarint(IPrimitiveField.Types type, Stream stream, ulong max_len, out ulong length)
{
switch (type)
{
case IPrimitiveField.Types.Bool:
return(Varint.FromBase128(stream, max_len, out length) != 0);
case IPrimitiveField.Types.Int32:
return((int)Varint.FromBase128Signed(stream, max_len, out length));
case IPrimitiveField.Types.Int64:
return((long)Varint.FromBase128Signed(stream, max_len, out length));
case IPrimitiveField.Types.Uint32:
return((uint)Varint.FromBase128(stream, max_len, out length));
case IPrimitiveField.Types.Uint64:
return((ulong)Varint.FromBase128(stream, max_len, out length));
default:
throw new InvalidOperationException("This function can only parse values from varints");
}
}
public override void WriteSignedVarint(long number)
{
Varint.ToBase128(number, _stream);
}
static void ParseClass(IClass cls, Stream stream, long allowed_length = -1)
{
// How do I want to deal with invalid stuff?
if (allowed_length < 0)
{
allowed_length = stream.Length;
}
bool done = false;
try {
while (stream.Position != allowed_length && !done)
{
Tuple <int, WireType> fieldInfo = ReadFieldInfo(stream);
WireType wireType = fieldInfo.Item2;
int fieldNum = fieldInfo.Item1;
IField field = cls.Fields.Where((field) => { return(field.Number == fieldNum); }).SingleOrDefault();
if (field == null)
{
// Discard this field
DiscardWire(wireType, stream);
continue;
}
switch (field)
{
case IClassField classField:
DiscardIfNot(stream, WireType.LENGTH_DELIMITED, wireType, () => {
ulong length = Varint.FromBase128(stream);
AppendIfList(classField);
ParseClass(classField[classField.Count - 1], stream, stream.Position + (long)length);
});
break;
case IEnumField enumField:
Func <ulong, ulong> addOneEnum = (max_len) => {
AppendIfList(enumField);
enumField[enumField.Count - 1] = (int)Varint.FromBase128(stream, max_len, out ulong length);
return(length);
};
if (wireType == WireType.LENGTH_DELIMITED)
{
ulong length = Varint.FromBase128(stream);
for (ulong i = 0; i < length;)
{
i += addOneEnum(length - i);
}
}
else
{
DiscardIfNot(stream, WireType.VARINT, wireType, () => {
addOneEnum(10);
});
}
break;
case IPrimitiveField primitiveField:
IPrimitiveField.Types fieldType = primitiveField.Type;
switch (fieldType)
{
case IPrimitiveField.Types.Bool:
case IPrimitiveField.Types.Int32:
case IPrimitiveField.Types.Int64:
case IPrimitiveField.Types.Uint32:
case IPrimitiveField.Types.Uint64:
Func <ulong, ulong> addOnePrim = (max_len) => {
AppendIfList(primitiveField);
primitiveField[primitiveField.Count - 1] = ParseSingleVarint(fieldType, stream, max_len, out ulong length);
return(length);
};
if (wireType == WireType.LENGTH_DELIMITED)
{
ulong length = Varint.FromBase128(stream);
for (ulong i = 0; i < length;)
{
i += addOnePrim(length - i);
}
;
}
else
{
DiscardIfNot(stream, WireType.VARINT, wireType, () => {
addOnePrim(10);
});
}
break;
case IPrimitiveField.Types.Double:
DiscardIfNot(stream, WireType.FIXED_64, wireType, () => {
AppendIfList(primitiveField);
primitiveField[primitiveField.Count - 1] = new BinaryReader(stream).ReadDouble();
});
break;
case IPrimitiveField.Types.Float:
DiscardIfNot(stream, WireType.FIXED_32, wireType, () => {
AppendIfList(primitiveField);
primitiveField[primitiveField.Count - 1] = new BinaryReader(stream).ReadSingle();
});
break;
case IPrimitiveField.Types.String:
DiscardIfNot(stream, WireType.LENGTH_DELIMITED, wireType, () => {
ulong length = Varint.FromBase128(stream);
StringWriter stringWriter = new();
for (ulong i = 0; i < length; i++)
{
stringWriter.Write((char)stream.ReadByte());
}
AppendIfList(primitiveField);
primitiveField[primitiveField.Count - 1] = stringWriter.ToString();
});
break;
}
;
break;
}
}
} catch (EndOfStreamException) {
}
}
public static Tuple <int, WireType> ReadFieldInfo(Stream stream)
{
ulong num = Varint.FromBase128(stream);
return(new Tuple <int, WireType>((int)(num >> 3), (WireType)(num & 0x7)));
}
private static void SerializeClass(IClass cls, ProtoStream protostream)
{
foreach (IField field in cls.Fields)
{
if (field.IsTransient)
{
continue;
}
// Treat all lists as un-packed
for (int i = 0; i < field.Count; i++)
{
if (field is IEnumField enumField)
{
protostream.WriteFieldInfo(enumField.Number, WireType.VARINT);
Varint.ToBase128(enumField[i], protostream);
}
else if (field is IPrimitiveField primitiveField)
{
switch (primitiveField.Type)
{
case IPrimitiveField.Types.Bool:
protostream.WriteFieldInfo(primitiveField.Number, WireType.VARINT);
protostream.WriteVarint(((bool)primitiveField[i]) ? (ulong)1 : (ulong)0);
break;
case IPrimitiveField.Types.Int32:
protostream.WriteFieldInfo(primitiveField.Number, WireType.VARINT);
protostream.WriteSignedVarint((int)primitiveField[i]);
break;
case IPrimitiveField.Types.Int64:
protostream.WriteFieldInfo(primitiveField.Number, WireType.VARINT);
protostream.WriteSignedVarint((long)primitiveField[i]);
break;
case IPrimitiveField.Types.Uint32:
protostream.WriteFieldInfo(primitiveField.Number, WireType.VARINT);
protostream.WriteVarint((ulong)(uint)primitiveField[i]);
break;
case IPrimitiveField.Types.Uint64:
protostream.WriteFieldInfo(primitiveField.Number, WireType.VARINT);
protostream.WriteVarint((ulong)primitiveField[i]);
break;
case IPrimitiveField.Types.Double:
protostream.WriteFieldInfo(primitiveField.Number, WireType.FIXED_64);
byte[] doubleBytes = BitConverter.GetBytes((double)primitiveField[i]);
protostream.Write(doubleBytes, 0, 8);
break;
case IPrimitiveField.Types.Float:
protostream.WriteFieldInfo(primitiveField.Number, WireType.FIXED_32);
byte[] floatBytes = BitConverter.GetBytes((float)primitiveField[i]);
protostream.Write(floatBytes, 0, 4);
break;
case IPrimitiveField.Types.Bytes:
protostream.WriteFieldInfo(primitiveField.Number, WireType.LENGTH_DELIMITED);
byte[] bytes = (byte[])(primitiveField[i]);
protostream.WriteVarint((ulong)bytes.Length);
protostream.Write(bytes, 0, bytes.Length);
break;
case IPrimitiveField.Types.String:
protostream.WriteFieldInfo(primitiveField.Number, WireType.LENGTH_DELIMITED);
byte[] stringBytes = Encoding.UTF8.GetBytes((string)primitiveField[i]);
protostream.WriteVarint((ulong)stringBytes.Length);
protostream.Write(stringBytes, 0, stringBytes.Length);
break;
case IPrimitiveField.Types.External:
throw new NotSupportedException("External type must always be transient");
}
}
else if (field is IClassField classField)
{
using CountingProtoStream counting = new();
SerializeClass(classField[i], counting);
protostream.WriteFieldInfo(classField.Number, WireType.LENGTH_DELIMITED);
protostream.WriteVarint((ulong)counting.Length);
SerializeClass(classField[i], protostream);
}
}
}
}
public override void WriteSignedVarint(long number)
{
Position += Varint.Base128Length(number);
}
public override void WriteFieldInfo(int number, WireType type)
{
Position += Varint.Base128Length((ulong)((number << 3) | (int)type));
}
