lib/CTRL: Improve and add more helper templates and functions

Change-Id: Icc6ac860ebd6a719f9e0cb5c5345fb4d39a864ce
diff --git a/library/Osmocom_CTRL_Functions.ttcn b/library/Osmocom_CTRL_Functions.ttcn
index 94dcb8a..22ed405 100644
--- a/library/Osmocom_CTRL_Functions.ttcn
+++ b/library/Osmocom_CTRL_Functions.ttcn
@@ -71,10 +71,16 @@
 		}
 	}
 
+	/* send a TRAP */
+	function f_ctrl_trap(IPA_CTRL_PT pt, CtrlVariable variable, CtrlValue val) {
+		pt.send(ts_CtrlMsgTrap(variable, val));
+	}
+
 	/* Expect a matching TRAP */
 	function f_ctrl_exp_trap(IPA_CTRL_PT pt, template CtrlVariable variable,
-				 template CtrlValue val := ?) return CtrlValue {
-		timer T := 2.0;
+				 template CtrlValue val := ?, float timeout_val := 2.0)
+	return CtrlValue {
+		timer T := timeout_val;
 		var CtrlMessage rx;
 		T.start;
 		alt {
@@ -88,6 +94,29 @@
 		return rx.trap.val;
 	}
 
+	/* Expect a matching SET, optionally answer */
+	function f_ctrl_exp_set(IPA_CTRL_PT pt, template CtrlVariable variable,
+				 template CtrlValue val := ?,
+				 template (omit) CtrlValue rsp := omit,
+				 float timeout_val := 2.0)
+	return CtrlValue {
+		timer T := timeout_val;
+		var CtrlMessage rx;
+		T.start;
+		alt {
+		[] pt.receive(tr_CtrlMsgSet(?, variable, val)) -> value rx {
+			if (ispresent(rsp)) {
+				pt.send(ts_CtrlMsgSetRepl(rx.cmd.id, valueof(variable), valueof(rsp)));
+			}
+			}
+		[] T.timeout {
+			setverdict(fail, "Timeout waiting for SET ", variable);
+			mtc.stop;
+			}
+		}
+		return rx.cmd.val;
+	}
+
 	/* Expect a matching GET result */
 	function f_ctrl_get_exp(IPA_CTRL_PT pt, CtrlVariable variable, template CtrlValue exp) {
 		var charstring ctrl_resp;