valitydev/thrift
14
0
Fork 0
mirror of https://github.com/valitydev/thrift.git synced 2024-11-07 10:48:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

THRIFT-3235 C#: Limit recursion depth to 64

Browse Source 
Client: C#
Patch: Jens Geyer
This commit is contained in:
Jens Geyer 2015-07-09 23:02:46 +02:00
parent d47fcdd66d
commit 4018036980
3 changed files with 148 additions and 64 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
compiler/cpp/src/generate/t_csharp_generator.cc
View File

@ -909,6 +909,10 @@ void t_csharp_generator::generate_csharp_struct_reader(ofstream& out, t_struct*
indent(out) << "public void Read (TProtocol iprot)" << endl;
scope_up(out);
out << indent() << "iprot.IncrementRecursionDepth();" << endl;
out << indent() << "try" << endl;
scope_up(out);
const vector<t_field*>& fields = tstruct->get_members();
vector<t_field*>::const_iterator f_iter;
@ -977,6 +981,12 @@ void t_csharp_generator::generate_csharp_struct_reader(ofstream& out, t_struct*
}
}
scope_down(out);
out << indent() << "finally" << endl;
scope_up(out);
out << indent() << "iprot.DecrementRecursionDepth();" << endl;
scope_down(out);
indent_down();
indent(out) << "}" << endl << endl;
@ -986,6 +996,10 @@ void t_csharp_generator::generate_csharp_struct_writer(ofstream& out, t_struct*
out << indent() << "public void Write(TProtocol oprot) {" << endl;
indent_up();
out << indent() << "oprot.IncrementRecursionDepth();" << endl;
out << indent() << "try" << endl;
scope_up(out);
string name = tstruct->get_name();
const vector<t_field*>& fields = tstruct->get_sorted_members();
vector<t_field*>::const_iterator f_iter;
@ -1030,6 +1044,12 @@ void t_csharp_generator::generate_csharp_struct_writer(ofstream& out, t_struct*
indent(out) << "oprot.WriteFieldStop();" << endl;
indent(out) << "oprot.WriteStructEnd();" << endl;
scope_down(out);
out << indent() << "finally" << endl;
scope_up(out);
out << indent() << "oprot.DecrementRecursionDepth();" << endl;
scope_down(out);
indent_down();
indent(out) << "}" << endl << endl;
@ -1039,6 +1059,10 @@ void t_csharp_generator::generate_csharp_struct_result_writer(ofstream& out, t_s
indent(out) << "public void Write(TProtocol oprot) {" << endl;
indent_up();
out << indent() << "oprot.IncrementRecursionDepth();" << endl;
out << indent() << "try" << endl;
scope_up(out);
string name = tstruct->get_name();
const vector<t_field*>& fields = tstruct->get_sorted_members();
vector<t_field*>::const_iterator f_iter;
@ -1092,6 +1116,12 @@ void t_csharp_generator::generate_csharp_struct_result_writer(ofstream& out, t_s
out << endl << indent() << "oprot.WriteFieldStop();" << endl << indent()
<< "oprot.WriteStructEnd();" << endl;
scope_down(out);
out << indent() << "finally" << endl;
scope_up(out);
out << indent() << "oprot.DecrementRecursionDepth();" << endl;
scope_down(out);
indent_down();
indent(out) << "}" << endl << endl;
@ -1249,6 +1279,11 @@ void t_csharp_generator::generate_csharp_union_class(std::ofstream& out,
indent(out) << "}" << endl;
indent(out) << "public override void Write(TProtocol oprot) {" << endl;
indent_up();
out << indent() << "oprot.IncrementRecursionDepth();" << endl;
out << indent() << "try" << endl;
scope_up(out);
indent(out) << "TStruct struc = new TStruct(\"" << tunion->get_name() << "\");" << endl;
indent(out) << "oprot.WriteStructBegin(struc);" << endl;
@ -1264,6 +1299,13 @@ void t_csharp_generator::generate_csharp_union_class(std::ofstream& out,
indent(out) << "oprot.WriteFieldStop();" << endl;
indent(out) << "oprot.WriteStructEnd();" << endl;
indent_down();
scope_down(out);
out << indent() << "finally" << endl;
scope_up(out);
out << indent() << "oprot.DecrementRecursionDepth();" << endl;
scope_down(out);
indent(out) << "}" << endl;
indent_down();
@ -1987,6 +2029,11 @@ void t_csharp_generator::generate_csharp_union_reader(std::ofstream& out, t_stru
indent(out) << "public static " << tunion->get_name() << " Read(TProtocol iprot)" << endl;
scope_up(out);
out << indent() << "iprot.IncrementRecursionDepth();" << endl;
out << indent() << "try" << endl;
scope_up(out);
indent(out) << tunion->get_name() << " retval;" << endl;
indent(out) << "iprot.ReadStructBegin();" << endl;
indent(out) << "TField field = iprot.ReadFieldBegin();" << endl;
@ -2036,13 +2083,16 @@ void t_csharp_generator::generate_csharp_union_reader(std::ofstream& out, t_stru
// end of else for TStop
scope_down(out);
indent(out) << "iprot.ReadStructEnd();" << endl;
indent(out) << "return retval;" << endl;
indent_down();
scope_down(out);
out << indent() << "finally" << endl;
scope_up(out);
out << indent() << "iprot.DecrementRecursionDepth();" << endl;
scope_down(out);
indent(out) << "}" << endl << endl;
}

25
lib/csharp/src/Protocol/TProtocol.cs
View File

@ -29,11 +29,17 @@ namespace Thrift.Protocol
{
public abstract class TProtocol : IDisposable
{
private const int DEFAULT_RECURSION_DEPTH = 64;
protected TTransport trans;
protected int recursionLimit;
protected int recursionDepth;
protected TProtocol(TTransport trans)
{
this.trans = trans;
this.recursionLimit = DEFAULT_RECURSION_DEPTH;
this.recursionDepth = 0;
}
public TTransport Transport
@ -41,6 +47,25 @@ namespace Thrift.Protocol
get { return trans; }
}
public int RecursionLimit
{
get { return recursionLimit; }
set { recursionLimit = value; }
}
public void IncrementRecursionDepth()
{
if (recursionDepth < recursionLimit)
++recursionDepth;
else
throw new TProtocolException(TProtocolException.DEPTH_LIMIT, "Depth limit exceeded");
}
public void DecrementRecursionDepth()
{
--recursionDepth;
}
#region " IDisposable Support "
private bool _IsDisposed;

9
lib/csharp/src/Protocol/TProtocolUtil.cs
View File

@ -28,6 +28,9 @@ namespace Thrift.Protocol
public static class TProtocolUtil
{
public static void Skip(TProtocol prot, TType type)
{
prot.IncrementRecursionDepth();
try
{
switch (type)
{
@ -93,6 +96,12 @@ namespace Thrift.Protocol
prot.ReadListEnd();
break;
}
}
finally
{
prot.DecrementRecursionDepth();
}
}
}
}