From e7d72b585a09f6b43df0f38ff20c74050069fb60 Mon Sep 17 00:00:00 2001
From: =?utf8?q?G=C3=BCnther=20Deschner?= <gd@samba.org>
Date: Tue, 30 Jun 2009 13:11:16 +0200
Subject: [PATCH] s4-smbtorture: Test for newly added form with enum call in
 RPC-SPOOLSS.

Guenther
---
 source4/torture/rpc/spoolss.c | 45 +++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/source4/torture/rpc/spoolss.c b/source4/torture/rpc/spoolss.c
index ad86df6baa36..1c4b8ed0d0f9 100644
--- a/source4/torture/rpc/spoolss.c
+++ b/source4/torture/rpc/spoolss.c
@@ -5,6 +5,7 @@
    Copyright (C) Tim Potter 2003
    Copyright (C) Stefan Metzmacher 2005
    Copyright (C) Jelmer Vernooij 2007
+   Copyright (C) Guenther Deschner 2009
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
@@ -1003,6 +1004,50 @@ static bool test_AddForm(struct torture_context *tctx,
 
 	if (!print_server) ret &= test_GetForm(tctx, p, handle, form_name, 1);
 
+	{
+		struct spoolss_EnumForms e;
+		union spoolss_FormInfo *info;
+		uint32_t needed;
+		uint32_t count;
+		bool found = false;
+
+		e.in.handle = handle;
+		e.in.level = 1;
+		e.in.buffer = NULL;
+		e.in.offered = 0;
+		e.out.needed = &needed;
+		e.out.count = &count;
+		e.out.info = &info;
+
+		torture_comment(tctx, "Testing EnumForms level 1\n");
+
+		status = dcerpc_spoolss_EnumForms(p, tctx, &e);
+		torture_assert_ntstatus_ok(tctx, status, "EnumForms failed");
+
+		if (print_server && W_ERROR_EQUAL(e.out.result, WERR_BADFID))
+			torture_fail(tctx, "EnumForms on the PrintServer isn't supported by test server (NT4)");
+
+		if (W_ERROR_EQUAL(e.out.result, WERR_INSUFFICIENT_BUFFER)) {
+			int j;
+			DATA_BLOB blob = data_blob_talloc(tctx, NULL, needed);
+			data_blob_clear(&blob);
+			e.in.buffer = &blob;
+			e.in.offered = needed;
+
+			status = dcerpc_spoolss_EnumForms(p, tctx, &e);
+
+			torture_assert(tctx, info, "No forms returned");
+
+			for (j = 0; j < count; j++) {
+				if (strequal(form_name, info[j].info1.form_name)) {
+					found = true;
+					break;
+				}
+			}
+		}
+		torture_assert(tctx, found, "Newly added form not found in enum call");
+	}
+
 	if (!test_DeleteForm(tctx, p, handle, form_name)) {
 		ret = false;
 	}
-- 
2.34.1