ndr.h

 
#ifndef CHANNELS_RDPEAR_NDR_H_
#define CHANNELS_RDPEAR_NDR_H_
 
#include <winpr/stream.h>
#include <freerdp/api.h>
 
#define NDR_PTR_NULL (0UL)
 
#define NDR_SIMPLE_TYPE_DECL(LOWER, UPPER)                                                        \
  BOOL ndr_read_##LOWER(NdrContext* context, wStream* s, UPPER* v);                             \
  BOOL ndr_read_##LOWER##_(NdrContext* context, wStream* s, const void* hints, void* v);        \
  BOOL ndr_write_##LOWER(NdrContext* context, wStream* s, UPPER v);                             \
  BOOL ndr_write_##LOWER##_(NdrContext* context, wStream* s, const void* hints, const void* v); \
  extern const NdrMessageDescr ndr_##LOWER##_descr_s;                                           \
  NdrMessageType ndr_##LOWER##_descr(void)
 
#define NDR_ARRAY_OF_TYPE_DECL(TYPE, UPPERTYPE)                                               \
  BOOL ndr_read_##TYPE##Array(NdrContext* context, wStream* s, const void* hints, void* v); \
  BOOL ndr_write_##TYPE##Array(NdrContext* context, wStream* s, const void* hints,          \
                               const void* v);                                              \
  void ndr_destroy_##TYPE##Array(NdrContext* context, const void* hints, void* obj);        \
  extern const NdrMessageDescr ndr_##TYPE##Array_descr_s;                                   \
  NdrMessageType ndr_##TYPE##Array_descr(void);                                             \
                                                                                              \
  BOOL ndr_read_##TYPE##VaryingArray(NdrContext* context, wStream* s, const void* hints,    \
                                     void* v);                                              \
  BOOL ndr_write_##TYPE##VaryingArray(NdrContext* context, wStream* s, const void* hints,   \
                                      const void* v);                                       \
  extern const NdrMessageDescr ndr_##TYPE##VaryingArray_descr_s;                            \
  NdrMessageType ndr_##TYPE##VaryingArray_descr(void)
 
#ifdef __cplusplus
extern "C"
{
#endif
 
  typedef struct NdrContext_s NdrContext;
 
  typedef UINT32 ndr_refid;
 
  typedef BOOL (*NDR_READER_FN)(NdrContext* context, wStream* s, const void* hints, void* target);
  typedef BOOL (*NDR_WRITER_FN)(NdrContext* context, wStream* s, const void* hints,
                                const void* obj);
  typedef void (*NDR_DESTROY_FN)(NdrContext* context, const void* hints, void* obj);
  typedef void (*NDR_DUMP_FN)(wLog* logger, UINT32 lvl, size_t indentLevel, const void* obj);
 
  typedef enum
  {
    NDR_ARITY_SIMPLE,
    NDR_ARITY_ARRAYOF,
    NDR_ARITY_VARYING_ARRAYOF,
  } NdrTypeArity;
 
  typedef struct
  {
    NdrTypeArity arity;
    size_t itemSize;
    NDR_READER_FN readFn;
    NDR_WRITER_FN writeFn;
    NDR_DESTROY_FN destroyFn;
    NDR_DUMP_FN dumpFn;
  } NdrMessageDescr;
 
  typedef const NdrMessageDescr* NdrMessageType;
 
  typedef enum
  {
    NDR_NOT_POINTER,
    NDR_POINTER_NON_NULL,
    NDR_POINTER
  } NdrPointerType;
 
  typedef struct
  {
    const char* name;
    size_t structOffset;
    NdrPointerType pointerType;
    ssize_t hintsField;
    NdrMessageType typeDescr;
  } NdrFieldStruct;
 
  typedef struct
  {
    const char* name;
    size_t nfields;
    const NdrFieldStruct* fields;
  } NdrStructDescr;
 
  typedef struct
  {
    ndr_refid ptrId;
    const char* name;
    void* hints;
    void* target;
    NdrMessageType msg;
  } NdrDeferredEntry;
 
  void ndr_context_free(NdrContext* context);
 
  static INLINE void ndr_context_destroy(NdrContext** pcontext)
  {
    WINPR_ASSERT(pcontext);
    ndr_context_free(*pcontext);
    *pcontext = NULL;
  }
 
  WINPR_ATTR_MALLOC(ndr_context_free, 1)
  NdrContext* ndr_context_new(BOOL bigEndianDrep, BYTE version);
 
  void ndr_context_reset(NdrContext* context);
 
  WINPR_ATTR_MALLOC(ndr_context_free, 1)
  NdrContext* ndr_context_copy(const NdrContext* src);
 
  WINPR_ATTR_MALLOC(ndr_context_free, 1)
  NdrContext* ndr_read_header(wStream* s);
 
  BOOL ndr_write_header(NdrContext* context, wStream* s);
 
  NDR_SIMPLE_TYPE_DECL(uint8, UINT8);
  NDR_SIMPLE_TYPE_DECL(uint16, UINT16);
  NDR_SIMPLE_TYPE_DECL(uint32, UINT32);
  NDR_SIMPLE_TYPE_DECL(uint64, UINT64);
 
  NDR_ARRAY_OF_TYPE_DECL(uint8, BYTE);
  NDR_ARRAY_OF_TYPE_DECL(uint16, UINT16);
 
  BOOL ndr_skip_bytes(NdrContext* context, wStream* s, size_t nbytes);
  BOOL ndr_read_align(NdrContext* context, wStream* s, size_t sz);
  BOOL ndr_write_align(NdrContext* context, wStream* s, size_t sz);
 
  BOOL ndr_read_pickle(NdrContext* context, wStream* s);
  BOOL ndr_write_pickle(NdrContext* context, wStream* s);
 
  BOOL ndr_read_constructed(NdrContext* context, wStream* s, wStream* target);
  BOOL ndr_write_constructed(NdrContext* context, wStream* s, wStream* payload);
 
  BOOL ndr_start_constructed(NdrContext* context, wStream* s);
  BOOL ndr_end_constructed(NdrContext* context, wStream* s);
 
  BOOL ndr_read_wchar(NdrContext* context, wStream* s, WCHAR* ptr);
 
  typedef struct
  {
    UINT32 length;
    UINT32 maxLength;
  } NdrVaryingArrayHints;
 
  BOOL ndr_read_uconformant_varying_array(NdrContext* context, wStream* s,
                                          const NdrVaryingArrayHints* hints,
                                          NdrMessageType itemType, void* ptarget);
  BOOL ndr_write_uconformant_varying_array(NdrContext* context, wStream* s,
                                           const NdrVaryingArrayHints* hints,
                                           NdrMessageType itemType, const void* src);
 
  typedef struct
  {
    UINT32 count;
  } NdrArrayHints;
 
  BOOL ndr_read_uconformant_array(NdrContext* context, wStream* s, const NdrArrayHints* hints,
                                  NdrMessageType itemType, void* vtarget);
  BOOL ndr_write_uconformant_array(NdrContext* context, wStream* s, UINT32 len,
                                   NdrMessageType itemType, const BYTE* ptr);
 
  BOOL ndr_struct_read_fromDescr(NdrContext* context, wStream* s, const NdrStructDescr* descr,
                                 void* target);
  BOOL ndr_struct_write_fromDescr(NdrContext* context, wStream* s, const NdrStructDescr* descr,
                                  const void* src);
  void ndr_struct_dump_fromDescr(wLog* logger, UINT32 lvl, size_t identLevel,
                                 const NdrStructDescr* descr, const void* obj);
  void ndr_struct_destroy(NdrContext* context, const NdrStructDescr* descr, void* pptr);
 
  ndr_refid ndr_pointer_refid(const void* ptr);
  BOOL ndr_read_refpointer(NdrContext* context, wStream* s, UINT32* refId);
  BOOL ndr_context_allocatePtr(NdrContext* context, const void* ptr, ndr_refid* prefId,
                               BOOL* pnewPtr);
 
  BOOL ndr_read_pointedMessageEx(NdrContext* context, wStream* s, ndr_refid ptrId,
                                 NdrMessageType descr, void* hints, void** target);
 
  BOOL ndr_push_deferreds(NdrContext* context, NdrDeferredEntry* deferreds, size_t ndeferred);
  BOOL ndr_treat_deferred_read(NdrContext* context, wStream* s);
  BOOL ndr_treat_deferred_write(NdrContext* context, wStream* s);
 
#ifdef __cplusplus
}
#endif
 
#endif /* CHANNELS_RDPEAR_NDR_H_ */