r20278: make it possible to use oid strings in the drsuapi instead of uint8_t arrays

with ber encoded oid's

metze
(This used to be commit bc13b3690166b29df246cc4778b378e9cf2a22d7)

1 files changed, 106 insertions, 0 deletions

diff --git a/source4/librpc/ndr/ndr_drsuapi.c b/source4/librpc/ndr/ndr_drsuapi.c
index 614d567e03..37c78a2c80 100644
--- a/source4/librpc/ndr/ndr_drsuapi.c
+++ b/source4/librpc/ndr/ndr_drsuapi.c
@@ -24,6 +24,7 @@
 #include "includes.h"
 #include "librpc/gen_ndr/ndr_drsuapi.h"
 #include "librpc/gen_ndr/ndr_misc.h"
+#include "libcli/util/asn_1.h"
 
 void ndr_print_drsuapi_DsReplicaObjectListItem(struct ndr_print *ndr, const char *name, 
 					       const struct drsuapi_DsReplicaObjectListItem *r)
@@ -62,3 +63,108 @@ void ndr_print_drsuapi_DsReplicaObjectListItemEx(struct ndr_print *ndr, const ch
 		ndr_print_drsuapi_DsReplicaObjectListItemEx(ndr, "next_object", r->next_object);
 	}
 }
+
+#define _ASN1_PUSH_CHECK(call) do { \
+	BOOL _status; \
+	_status = call; \
+	if (_status != True) { \
+		return ndr_push_error(ndr, NDR_ERR_SUBCONTEXT, "ASN.1 Error: %s\n", __location__); \
+	} \
+} while (0)
+
+#define _ASN1_PULL_CHECK(call) do { \
+	BOOL _status; \
+	_status = call; \
+	if (_status != True) { \
+		return ndr_pull_error(ndr, NDR_ERR_SUBCONTEXT, "ASN.1 Error: %s\n", __location__); \
+	} \
+} while (0)
+
+NTSTATUS ndr_push_drsuapi_DsReplicaOID(struct ndr_push *ndr, int ndr_flags, const struct drsuapi_DsReplicaOID *r)
+{
+	if (ndr_flags & NDR_SCALARS) {
+		NDR_CHECK(ndr_push_align(ndr, 4));
+		NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, ndr_size_drsuapi_DsReplicaOID_oid(r->oid, 0)));
+		NDR_CHECK(ndr_push_unique_ptr(ndr, r->oid));
+	}
+	if (ndr_flags & NDR_BUFFERS) {
+		if (r->oid) {
+			struct asn1_data _asn1;
+
+			ZERO_STRUCT(_asn1);
+			_ASN1_PUSH_CHECK(asn1_write_OID_String(&_asn1, r->oid));
+			talloc_steal(ndr, _asn1.data);
+
+			NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, _asn1.ofs));
+			NDR_CHECK(ndr_push_array_uint8(ndr, NDR_SCALARS, _asn1.data, _asn1.ofs));
+			asn1_free(&_asn1);
+		}
+	}
+	return NT_STATUS_OK;
+}
+
+NTSTATUS ndr_pull_drsuapi_DsReplicaOID(struct ndr_pull *ndr, int ndr_flags, struct drsuapi_DsReplicaOID *r)
+{
+	uint32_t _ptr_oid;
+	TALLOC_CTX *_mem_save_oid_0;
+	if (ndr_flags & NDR_SCALARS) {
+		NDR_CHECK(ndr_pull_align(ndr, 4));
+		NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &r->__ndr_size));
+		if (r->__ndr_size < 0 || r->__ndr_size > 10000) {
+			return ndr_pull_error(ndr, NDR_ERR_RANGE, "value out of range");
+		}
+		NDR_CHECK(ndr_pull_generic_ptr(ndr, &_ptr_oid));
+		if (_ptr_oid) {
+			NDR_PULL_ALLOC(ndr, r->oid);
+		} else {
+			r->oid = NULL;
+		}
+	}
+	if (ndr_flags & NDR_BUFFERS) {
+		if (r->oid) {
+			DATA_BLOB _oid_array;
+			const char *_oid;
+			struct asn1_data _asn1;
+
+			ZERO_STRUCT(_asn1);
+
+			_mem_save_oid_0 = NDR_PULL_GET_MEM_CTX(ndr);
+			NDR_PULL_SET_MEM_CTX(ndr, ndr, 0);
+			NDR_CHECK(ndr_pull_array_size(ndr, &r->oid));
+			_oid_array.length = ndr_get_array_size(ndr, &r->oid);
+			NDR_PULL_ALLOC_N(ndr, _oid_array.data, _oid_array.length);
+			NDR_CHECK(ndr_pull_array_uint8(ndr, NDR_SCALARS, _oid_array.data, _oid_array.length));
+			NDR_PULL_SET_MEM_CTX(ndr, _mem_save_oid_0, 0);
+
+			_ASN1_PULL_CHECK(asn1_load(&_asn1, _oid_array));
+			talloc_steal(ndr, _asn1.data);
+			data_blob_free(&_oid_array);
+			_ASN1_PULL_CHECK(asn1_start_fake_tag(&_asn1));
+			_ASN1_PULL_CHECK(asn1_read_OID_String(&_asn1, &_oid));
+			talloc_steal(r->oid, _oid);
+			r->oid = _oid;
+			_ASN1_PULL_CHECK(asn1_end_tag(&_asn1));
+			asn1_free(&_asn1);
+		}
+		if (r->oid) {
+			NDR_CHECK(ndr_check_array_size(ndr, (void*)&r->oid, r->__ndr_size));
+		}
+	}
+	return NT_STATUS_OK;
+}
+
+size_t ndr_size_drsuapi_DsReplicaOID_oid(const char *oid, int flags)
+{
+	struct asn1_data _asn1;
+	size_t ret = 0;
+
+	if (!oid) return 0;
+
+	ZERO_STRUCT(_asn1);
+	if (asn1_write_OID_String(&_asn1, oid)) {
+		ret = _asn1.ofs;
+	}
+
+	asn1_free(&_asn1);
+	return ret;
+}