151 lines
4.7 KiB
C#
151 lines
4.7 KiB
C#
// This code is derived from jcifs smb client library <jcifs at samba dot org>
|
|
// Ported by J. Arturo <webmaster at komodosoft dot net>
|
|
//
|
|
// This library is free software; you can redistribute it and/or
|
|
// modify it under the terms of the GNU Lesser General Public
|
|
// License as published by the Free Software Foundation; either
|
|
// version 2.1 of the License, or (at your option) any later version.
|
|
//
|
|
// This library 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
|
|
// Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public
|
|
// License along with this library; if not, write to the Free Software
|
|
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
using SharpCifs.Dcerpc.Ndr;
|
|
|
|
namespace SharpCifs.Dcerpc
|
|
{
|
|
public abstract class DcerpcMessage : NdrObject
|
|
{
|
|
protected internal int Ptype = -1;
|
|
|
|
protected internal int Flags;
|
|
|
|
protected internal int Length;
|
|
|
|
protected internal int CallId;
|
|
|
|
protected internal int AllocHint;
|
|
|
|
protected internal int Result;
|
|
|
|
public virtual bool IsFlagSet(int flag)
|
|
{
|
|
return (Flags & flag) == flag;
|
|
}
|
|
|
|
public virtual void UnsetFlag(int flag)
|
|
{
|
|
Flags &= ~flag;
|
|
}
|
|
|
|
public virtual void SetFlag(int flag)
|
|
{
|
|
Flags |= flag;
|
|
}
|
|
|
|
public virtual DcerpcException GetResult()
|
|
{
|
|
if (Result != 0)
|
|
{
|
|
return new DcerpcException(Result);
|
|
}
|
|
return null;
|
|
}
|
|
|
|
internal virtual void Encode_header(NdrBuffer buf)
|
|
{
|
|
buf.Enc_ndr_small(5);
|
|
buf.Enc_ndr_small(0);
|
|
buf.Enc_ndr_small(Ptype);
|
|
buf.Enc_ndr_small(Flags);
|
|
buf.Enc_ndr_long(unchecked(0x00000010));
|
|
buf.Enc_ndr_short(Length);
|
|
buf.Enc_ndr_short(0);
|
|
buf.Enc_ndr_long(CallId);
|
|
}
|
|
|
|
/// <exception cref="SharpCifs.Dcerpc.Ndr.NdrException"></exception>
|
|
internal virtual void Decode_header(NdrBuffer buf)
|
|
{
|
|
if (buf.Dec_ndr_small() != 5 || buf.Dec_ndr_small() != 0)
|
|
{
|
|
throw new NdrException("DCERPC version not supported");
|
|
}
|
|
Ptype = buf.Dec_ndr_small();
|
|
Flags = buf.Dec_ndr_small();
|
|
if (buf.Dec_ndr_long() != unchecked(0x00000010))
|
|
{
|
|
throw new NdrException("Data representation not supported");
|
|
}
|
|
Length = buf.Dec_ndr_short();
|
|
if (buf.Dec_ndr_short() != 0)
|
|
{
|
|
throw new NdrException("DCERPC authentication not supported");
|
|
}
|
|
CallId = buf.Dec_ndr_long();
|
|
}
|
|
|
|
/// <exception cref="SharpCifs.Dcerpc.Ndr.NdrException"></exception>
|
|
public override void Encode(NdrBuffer buf)
|
|
{
|
|
int start = buf.GetIndex();
|
|
int allocHintIndex = 0;
|
|
buf.Advance(16);
|
|
if (Ptype == 0)
|
|
{
|
|
allocHintIndex = buf.GetIndex();
|
|
buf.Enc_ndr_long(0);
|
|
buf.Enc_ndr_short(0);
|
|
buf.Enc_ndr_short(GetOpnum());
|
|
}
|
|
Encode_in(buf);
|
|
Length = buf.GetIndex() - start;
|
|
if (Ptype == 0)
|
|
{
|
|
buf.SetIndex(allocHintIndex);
|
|
AllocHint = Length - allocHintIndex;
|
|
buf.Enc_ndr_long(AllocHint);
|
|
}
|
|
buf.SetIndex(start);
|
|
Encode_header(buf);
|
|
buf.SetIndex(start + Length);
|
|
}
|
|
|
|
/// <exception cref="SharpCifs.Dcerpc.Ndr.NdrException"></exception>
|
|
public override void Decode(NdrBuffer buf)
|
|
{
|
|
Decode_header(buf);
|
|
if (Ptype != 12 && Ptype != 2 && Ptype != 3 && Ptype != 13)
|
|
{
|
|
throw new NdrException("Unexpected ptype: " + Ptype);
|
|
}
|
|
if (Ptype == 2 || Ptype == 3)
|
|
{
|
|
AllocHint = buf.Dec_ndr_long();
|
|
buf.Dec_ndr_short();
|
|
buf.Dec_ndr_short();
|
|
}
|
|
if (Ptype == 3 || Ptype == 13)
|
|
{
|
|
Result = buf.Dec_ndr_long();
|
|
}
|
|
else
|
|
{
|
|
Decode_out(buf);
|
|
}
|
|
}
|
|
|
|
public abstract int GetOpnum();
|
|
|
|
/// <exception cref="SharpCifs.Dcerpc.Ndr.NdrException"></exception>
|
|
public abstract void Encode_in(NdrBuffer dst);
|
|
|
|
/// <exception cref="SharpCifs.Dcerpc.Ndr.NdrException"></exception>
|
|
public abstract void Decode_out(NdrBuffer src);
|
|
}
|
|
}
|