-# ==================================================================================
-# Copyright (c) 2019 Nokia
-# Copyright (c) 2018-2019 AT&T Intellectual Property.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ==================================================================================
-import time
-import random
-import string
-import os
-import signal
-import sys
-from rmr import rmr
-
-
-# Demonstrate NNG cleanup
-def signal_handler(sig, frame):
- print('SIGINT received! Cleaning up rmr')
- rmr.rmr_close(mrc)
- print("Byeee")
- sys.exit(0)
-
-
-# Init rmr
-mrc = rmr.rmr_init(b"4562", rmr.RMR_MAX_RCV_BYTES, 0x00)
-while rmr.rmr_ready(mrc) == 0:
- time.sleep(1)
- print("not yet ready")
-rmr.rmr_set_stimeout(mrc, 2)
-sbuf = rmr.rmr_alloc_msg(mrc, 256)
-
-# capture ctrl-c
-signal.signal(signal.SIGINT, signal_handler)
-
-while True:
- # generate a random value between 1 and 256 bytes, then gen some random bytes with several nulls thrown in
- for val in [''.join([random.choice(string.ascii_letters + string.digits) for n in range(random.randint(1,256))]).encode("utf8"),
- b"\x00" + os.urandom(4) + b"\x00" + os.urandom(4) + b"\x00"]:
- rmr.set_payload_and_length(val, sbuf)
- rmr.generate_and_set_transaction_id(sbuf)
- sbuf.contents.state = 0
- sbuf.contents.mtype = 0
- print("Pre send summary: {}".format(rmr.message_summary(sbuf)))
- sbuf = rmr.rmr_send_msg(mrc, sbuf)
- print("Post send summary: {}".format(rmr.message_summary(sbuf)))
- print("Waiting for return, will timeout after 2000ms")
- sbuf = rmr.rmr_torcv_msg(mrc, sbuf, 2000)
- summary = rmr.message_summary(sbuf)
- if summary['message state'] == 12:
- print("Nothing received yet")
- else:
- print("Ack Message received!: {}".format(summary))
-
- time.sleep(1)