#ifndef _SMBSRV_NDR_H
#define _SMBSRV_NDR_H
#include <sys/types.h>
#include <sys/uio.h>
#include <stdlib.h>
#include <string.h>
#include <smb/wintypes.h>
#include <libmlrpc/ndrtypes.ndl>
#include <libmlrpc/rpcpdu.ndl>
#ifdef __cplusplus
extern "C" {
#endif
#define NDR_F_NONE 0x0000
#define NDR_F_PARAMS_MASK 0x00FF
#define NDR_F_SIZE_IS 0x0001
#define NDR_F_LENGTH_IS 0x0002
#define NDR_F_SWITCH_IS 0x0004
#define NDR_F_IS_STRING 0x0008
#define NDR_F_IS_POINTER 0x0010
#define NDR_F_IS_REFERENCE 0x0020
#define NDR_F_DIMENSION_IS 0x0040
#define NDR_F_WHENCE_MASK 0x00F0
#define NDR_F_BACKPTR 0x0010
#define NDR_F_OUTER 0x0020
#define NDR_F_TOPMOST 0x0040
#define NDR_F_TYPEOP_MASK 0x0F00
#define NDR_F_ARRAY 0x0100
#define NDR_F_POINTER 0x0200
#define NDR_F_STRING 0x0300
#define NDR_F_UNION 0x0400
#define NDR_F_STRUCT 0x0500
#define NDR_F_OPERATION 0x0600
#define NDR_F_INTERFACE 0x0700
#define NDR_F_CONFORMANT 0x1000
#define NDR_F_VARYING 0x2000
#define NDR_F_FAKE 0x4000
struct ndr_heap;
struct ndr_stream;
struct ndr_reference;
typedef uint16_t ndr_wchar_t;
typedef struct ndr_typeinfo {
unsigned char version;
unsigned char alignment;
unsigned short type_flags;
int (*ndr_func)(struct ndr_reference *);
unsigned short pdu_size_fixed_part;
unsigned short pdu_size_variable_part;
unsigned short c_size_fixed_part;
unsigned short c_size_variable_part;
} ndr_typeinfo_t;
typedef struct ndr_reference {
struct ndr_reference *next;
struct ndr_reference *enclosing;
struct ndr_stream *stream;
ndr_typeinfo_t *ti;
char *name;
unsigned long pdu_offset;
char *datum;
char **backptr;
unsigned short outer_flags;
unsigned short inner_flags;
unsigned short type_flags;
unsigned short packed_alignment;
unsigned long size_is;
unsigned long strlen_is;
unsigned long switch_is;
unsigned long dimension_is;
unsigned long pdu_end_offset;
} ndr_ref_t;
typedef struct ndr_stream_ops {
char *(*ndo_malloc)(struct ndr_stream *, unsigned, ndr_ref_t *);
int (*ndo_free)(struct ndr_stream *, char *, ndr_ref_t *);
int (*ndo_grow_pdu)(struct ndr_stream *, unsigned long, ndr_ref_t *);
int (*ndo_pad_pdu)(struct ndr_stream *, unsigned long,
unsigned long, ndr_ref_t *);
int (*ndo_get_pdu)(struct ndr_stream *, unsigned long,
unsigned long, char *, int, ndr_ref_t *);
int (*ndo_put_pdu)(struct ndr_stream *, unsigned long,
unsigned long, char *, int, ndr_ref_t *);
void (*ndo_tattle)(struct ndr_stream *, char *, ndr_ref_t *);
void (*ndo_tattle_error)(struct ndr_stream *, ndr_ref_t *);
int (*ndo_reset)(struct ndr_stream *);
void (*ndo_destruct)(struct ndr_stream *);
} ndr_stream_ops_t;
#define NDS_MALLOC(NDS, LEN, REF) \
(*(NDS)->ndo->ndo_malloc)(NDS, LEN, REF)
#define NDS_GROW_PDU(NDS, WANT_END_OFF, REF) \
(*(NDS)->ndo->ndo_grow_pdu)(NDS, WANT_END_OFF, REF)
#define NDS_PAD_PDU(NDS, PDU_OFFSET, N_BYTES, REF) \
(*(NDS)->ndo->ndo_pad_pdu)(NDS, PDU_OFFSET, N_BYTES, REF)
#define NDS_GET_PDU(NDS, PDU_OFFSET, N_BYTES, BUF, SWAP, REF) \
(*(NDS)->ndo->ndo_get_pdu)(NDS, PDU_OFFSET, N_BYTES, BUF, SWAP, REF)
#define NDS_PUT_PDU(NDS, PDU_OFFSET, N_BYTES, BUF, SWAP, REF) \
(*(NDS)->ndo->ndo_put_pdu)(NDS, PDU_OFFSET, N_BYTES, BUF, SWAP, REF)
#define NDS_TATTLE(NDS, WHAT, REF) \
(*(NDS)->ndo->ndo_tattle)(NDS, WHAT, REF)
#define NDS_TATTLE_ERROR(NDS, WHAT, REF) \
(*(NDS)->ndo->ndo_tattle_error)(NDS, REF)
#define NDS_RESET(NDS) (*(NDS)->ndo->ndo_reset)(NDS)
#define NDS_DESTRUCT(NDS) (*(NDS)->ndo->ndo_destruct)(NDS)
typedef struct ndr_stream {
unsigned long pdu_size;
unsigned long pdu_max_size;
unsigned long pdu_base_offset;
unsigned long pdu_scan_offset;
unsigned long pdu_body_offset;
unsigned long pdu_body_size;
unsigned long pdu_hdr_size;
unsigned char *pdu_base_addr;
ndr_stream_ops_t *ndo;
unsigned char m_op;
unsigned char dir;
unsigned char swap;
unsigned char flags;
short error;
short error_ref;
ndr_ref_t *outer_queue_head;
ndr_ref_t **outer_queue_tailp;
ndr_ref_t *outer_current;
struct ndr_heap *heap;
} ndr_stream_t;
#define NDR_M_OP_NONE 0x00
#define NDR_M_OP_MARSHALL 0x01
#define NDR_M_OP_UNMARSHALL 0x02
#define NDR_DIR_NONE 0x00
#define NDR_DIR_IN 0x10
#define NDR_DIR_OUT 0x20
#define NDR_MODE_CALL_SEND (NDR_M_OP_MARSHALL + NDR_DIR_IN)
#define NDR_MODE_CALL_RECV (NDR_M_OP_UNMARSHALL + NDR_DIR_IN)
#define NDR_MODE_RETURN_SEND (NDR_M_OP_MARSHALL + NDR_DIR_OUT)
#define NDR_MODE_RETURN_RECV (NDR_M_OP_UNMARSHALL + NDR_DIR_OUT)
#define NDR_MODE_BUF_ENCODE NDR_MODE_CALL_SEND
#define NDR_MODE_BUF_DECODE NDR_MODE_RETURN_RECV
#define NDR_MODE_TO_M_OP(MODE) ((MODE) & 0x0F)
#define NDR_MODE_TO_DIR(MODE) ((MODE) & 0xF0)
#define NDR_M_OP_AND_DIR_TO_MODE(M_OP, DIR) ((M_OP)|(DIR))
#define NDR_MODE_MATCH(NDS, MODE) \
(NDR_M_OP_AND_DIR_TO_MODE((NDS)->m_op, (NDS)->dir) == (MODE))
#define NDR_IS_FIRST_FRAG(F) ((F) & NDR_PFC_FIRST_FRAG)
#define NDR_IS_LAST_FRAG(F) ((F) & NDR_PFC_LAST_FRAG)
#define NDR_IS_SINGLE_FRAG(F) \
(NDR_IS_FIRST_FRAG((F)) && NDR_IS_LAST_FRAG((F)))
#define NDS_F_NONE 0x00
#define NDS_F_NOTERM 0x01
#define NDS_F_NONULL 0x02
#define NDS_SETF(S, F) ((S)->flags |= (F))
#define NDS_CLEARF(S, F) ((S)->flags &= ~(F))
#define NDR_ERR_MALLOC_FAILED -1
#define NDR_ERR_M_OP_INVALID -2
#define NDR_ERR_UNDERFLOW -3
#define NDR_ERR_GROW_FAILED -4
#define NDR_ERR_PAD_FAILED -5
#define NDR_ERR_OUTER_HEADER_BAD -6
#define NDR_ERR_SWITCH_VALUE_ILLEGAL -7
#define NDR_ERR_SWITCH_VALUE_INVALID -8
#define NDR_ERR_SWITCH_VALUE_MISSING -9
#define NDR_ERR_SIZE_IS_MISMATCH_PDU -10
#define NDR_ERR_SIZE_IS_MISMATCH_AFTER -11
#define NDR_ERR_SIZE_IS_UNEXPECTED -12
#define NDR_ERR_SIZE_IS_DUPLICATED -13
#define NDR_ERR_OUTER_PARAMS_MISMATCH -14
#define NDR_ERR_ARRAY_VARLEN_ILLEGAL -15
#define NDR_ERR_ARRAY_UNION_ILLEGAL -16
#define NDR_ERR_OUTER_PARAMS_BAD -17
#define NDR_ERR_OUTER_UNION_ILLEGAL -18
#define NDR_ERR_TOPMOST_UNION_ILLEGAL -19
#define NDR_ERR_TOPMOST_VARLEN_ILLEGAL -20
#define NDR_ERR_INNER_PARAMS_BAD -21
#define NDR_ERR_UNIMPLEMENTED -22
#define NDR_ERR_NOT_AN_INTERFACE -23
#define NDR_ERR_STRLEN -24
#define NDR_ERR_STRING_SIZING -25
#define NDR_ERR_BOUNDS_CHECK -26
#define NDR_SET_ERROR(REF, ERROR) \
((REF)->stream->error = (ERROR), \
(REF)->stream->error_ref = __LINE__, \
NDS_TATTLE_ERROR((REF)->stream, 0, REF))
#define NDR_TATTLE(REF, WHAT) \
(*(REF)->stream->ndo->ndo_tattle)((REF)->stream, WHAT, REF)
#define MEMBER_STR(MEMBER) #MEMBER
#define NDR_DIR_IS_IN (encl_ref->stream->dir == NDR_DIR_IN)
#define NDR_DIR_IS_OUT (encl_ref->stream->dir == NDR_DIR_OUT)
#define NDR_MEMBER_PTR_WITH_ARG(TYPE, MEMBER, OFFSET, \
ARGFLAGS, ARGMEM, ARGVAL) { \
myref.pdu_offset = encl_ref->pdu_offset + (OFFSET); \
myref.name = MEMBER_STR(MEMBER); \
myref.datum = (char *)val->MEMBER; \
myref.inner_flags = ARGFLAGS; \
myref.ti = &ndt_##TYPE; \
myref.ARGMEM = ARGVAL; \
if (!ndr_inner(&myref)) \
return (0); \
}
#define NDR_MEMBER_PTR_WITH_DIMENSION(TYPE, MEMBER, OFFSET, SIZE_IS) \
NDR_MEMBER_PTR_WITH_ARG(TYPE, MEMBER, OFFSET, \
NDR_F_DIMENSION_IS, dimension_is, SIZE_IS)
#define NDR_MEMBER_WITH_ARG(TYPE, MEMBER, OFFSET, \
ARGFLAGS, ARGMEM, ARGVAL) { \
myref.pdu_offset = encl_ref->pdu_offset + (OFFSET); \
myref.name = MEMBER_STR(MEMBER); \
myref.datum = (char *)&val->MEMBER; \
myref.inner_flags = ARGFLAGS; \
myref.ti = &ndt_##TYPE; \
myref.ARGMEM = ARGVAL; \
if (!ndr_inner(&myref)) \
return (0); \
}
#define NDR_MEMBER(TYPE, MEMBER, OFFSET) \
NDR_MEMBER_WITH_ARG(TYPE, MEMBER, OFFSET, \
NDR_F_NONE, size_is, 0)
#define NDR_MEMBER_ARR_WITH_SIZE_IS(TYPE, MEMBER, OFFSET, SIZE_IS) \
NDR_MEMBER_WITH_ARG(TYPE, MEMBER, OFFSET, \
NDR_F_SIZE_IS, size_is, SIZE_IS)
#define NDR_MEMBER_ARR_WITH_DIMENSION(TYPE, MEMBER, OFFSET, SIZE_IS) \
NDR_MEMBER_WITH_ARG(TYPE, MEMBER, OFFSET, \
NDR_F_DIMENSION_IS, dimension_is, SIZE_IS)
#define NDR_MEMBER_PTR_WITH_SIZE_IS(TYPE, MEMBER, OFFSET, SIZE_IS) \
NDR_MEMBER_WITH_ARG(TYPE, MEMBER, OFFSET, \
NDR_F_SIZE_IS+NDR_F_IS_POINTER, size_is, SIZE_IS)
#define NDR_MEMBER_PTR(TYPE, MEMBER, OFFSET) \
NDR_MEMBER_WITH_ARG(TYPE, MEMBER, OFFSET, \
NDR_F_IS_POINTER, size_is, 0)
#define NDR_MEMBER_WITH_SWITCH_IS(TYPE, MEMBER, OFFSET, SWITCH_IS) \
NDR_MEMBER_WITH_ARG(TYPE, MEMBER, OFFSET, \
NDR_F_SWITCH_IS, switch_is, SWITCH_IS)
#define NDR_TOPMOST_MEMBER_WITH_ARG(TYPE, MEMBER, \
ARGFLAGS, ARGMEM, ARGVAL) { \
myref.pdu_offset = -1; \
myref.name = MEMBER_STR(MEMBER); \
myref.datum = (char *)&val->MEMBER; \
myref.inner_flags = ARGFLAGS; \
myref.ti = &ndt_##TYPE; \
myref.ARGMEM = ARGVAL; \
if (!ndr_topmost(&myref)) \
return (0); \
}
#define NDR_TOPMOST_MEMBER(TYPE, MEMBER) \
NDR_TOPMOST_MEMBER_WITH_ARG(TYPE, MEMBER, \
NDR_F_NONE, size_is, 0)
#define NDR_TOPMOST_MEMBER_ARR_WITH_SIZE_IS(TYPE, MEMBER, SIZE_IS) \
NDR_TOPMOST_MEMBER_WITH_ARG(TYPE, MEMBER, \
NDR_F_SIZE_IS, size_is, SIZE_IS)
#define NDR_TOPMOST_MEMBER_ARR_WITH_DIMENSION(TYPE, MEMBER, SIZE_IS) \
NDR_TOPMOST_MEMBER_WITH_ARG(TYPE, MEMBER, \
NDR_F_DIMENSION_IS, dimension_is, SIZE_IS)
#define NDR_TOPMOST_MEMBER_PTR_WITH_SIZE_IS(TYPE, MEMBER, SIZE_IS) \
NDR_TOPMOST_MEMBER_WITH_ARG(TYPE, MEMBER, \
NDR_F_SIZE_IS+NDR_F_IS_POINTER, size_is, SIZE_IS)
#define NDR_TOPMOST_MEMBER_PTR(TYPE, MEMBER) \
NDR_TOPMOST_MEMBER_WITH_ARG(TYPE, MEMBER, \
NDR_F_IS_POINTER, size_is, 0)
#define NDR_TOPMOST_MEMBER_REF(TYPE, MEMBER) \
NDR_TOPMOST_MEMBER_WITH_ARG(TYPE, MEMBER, \
NDR_F_IS_REFERENCE, size_is, 0)
#define NDR_TOPMOST_MEMBER_REF_WITH_SIZE_IS(TYPE, MEMBER, SIZE_IS) \
NDR_TOPMOST_MEMBER_WITH_ARG(TYPE, MEMBER, \
NDR_F_SIZE_IS+NDR_F_IS_REFERENCE, size_is, SIZE_IS)
#define NDR_TOPMOST_MEMBER_WITH_SWITCH_IS(TYPE, MEMBER, SWITCH_IS) \
NDR_TOPMOST_MEMBER_WITH_ARG(TYPE, MEMBER, \
NDR_F_SWITCH_IS, switch_is, SWITCH_IS)
#define NDR_PARAMS_MEMBER_WITH_ARG(TYPE, MEMBER, ARGFLAGS, \
ARGMEM, ARGVAL) { \
myref.pdu_offset = encl_ref->pdu_offset; \
myref.name = MEMBER_STR(MEMBER); \
myref.datum = (char *)&val->MEMBER; \
myref.inner_flags = ARGFLAGS; \
myref.ti = &ndt_##TYPE; \
myref.ARGMEM = ARGVAL; \
if (!ndr_params(&myref)) \
return (0); \
}
#define NDR_PARAMS_MEMBER(TYPE, MEMBER) \
NDR_PARAMS_MEMBER_WITH_ARG(TYPE, MEMBER, \
NDR_F_NONE, size_is, 0)
#define NDR_STRING_DIM 1
#define NDR_ANYSIZE_DIM 1
int ndo_process(struct ndr_stream *, ndr_typeinfo_t *, char *);
int ndo_operation(struct ndr_stream *, ndr_typeinfo_t *, int opnum, char *);
void ndo_printf(struct ndr_stream *, ndr_ref_t *, const char *, ...);
void ndo_trace(const char *);
void ndo_fmt(struct ndr_stream *, ndr_ref_t *, char *);
int ndr_params(ndr_ref_t *);
int ndr_topmost(ndr_ref_t *);
int ndr_run_outer_queue(struct ndr_stream *);
int ndr_outer(ndr_ref_t *);
int ndr_outer_fixed(ndr_ref_t *);
int ndr_outer_fixed_array(ndr_ref_t *);
int ndr_outer_conformant_array(ndr_ref_t *);
int ndr_outer_conformant_construct(ndr_ref_t *);
int ndr_size_is(ndr_ref_t *);
int ndr_outer_string(ndr_ref_t *);
int ndr_outer_peek_sizing(ndr_ref_t *, unsigned, unsigned long *);
int ndr_outer_poke_sizing(ndr_ref_t *, unsigned, unsigned long *);
int ndr_outer_align(ndr_ref_t *);
int ndr_outer_grow(ndr_ref_t *, unsigned);
int ndr_inner(ndr_ref_t *);
int ndr_inner_pointer(ndr_ref_t *);
int ndr_inner_reference(ndr_ref_t *);
int ndr_inner_array(ndr_ref_t *);
size_t ndr_mbstowcs(struct ndr_stream *, ndr_wchar_t *, const char *, size_t);
void nds_bswap(void *src, void *dst, size_t len);
#ifdef __cplusplus
}
#endif
#endif