|
| 1 | +#!/usr/bin/env python3 |
| 2 | +# |
| 3 | +# Copyright (C) 2023 Intel Corporation. All rights reserved. |
| 4 | +# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 5 | +# |
| 6 | + |
| 7 | +import argparse |
| 8 | +import shlex |
| 9 | +import subprocess |
| 10 | +import sys |
| 11 | +import time |
| 12 | +import traceback |
| 13 | +import glob |
| 14 | + |
| 15 | +WAMRC_CMD = "../../wamr-compiler/build/wamrc" |
| 16 | + |
| 17 | +def compile_wasm_files_to_aot(wasm_apps_dir): |
| 18 | + wasm_files = glob.glob(wasm_apps_dir + "/*.wasm") |
| 19 | + print("Compile wasm app into aot files") |
| 20 | + for wasm_file in wasm_files: |
| 21 | + aot_file = wasm_file[0 : len(wasm_file) - 5] + ".aot"; |
| 22 | + cmd = [ WAMRC_CMD, "-o", aot_file, wasm_file ] |
| 23 | + subprocess.check_call(cmd) |
| 24 | + |
| 25 | +def start_server(cmd, cwd): |
| 26 | + app_server = subprocess.Popen(shlex.split(cmd), cwd=cwd) |
| 27 | + return app_server |
| 28 | + |
| 29 | +def run_cmd(cmd, cwd): |
| 30 | + qry_prc = subprocess.run( |
| 31 | + shlex.split(cmd), cwd=cwd, check=False, capture_output=True |
| 32 | + ) |
| 33 | + if (qry_prc.returncode != 0): |
| 34 | + print("Run {} failed, return {}".format(cmd), qry_prc.returncode) |
| 35 | + return |
| 36 | + print("return code: {}, output:\n{}".format(qry_prc.returncode, |
| 37 | + qry_prc.stdout.decode())) |
| 38 | + |
| 39 | +def main(): |
| 40 | + """ |
| 41 | + GO!GO!!GO!!! |
| 42 | + """ |
| 43 | + parser = argparse.ArgumentParser(description="run the sample and examine outputs") |
| 44 | + parser.add_argument("working_directory", type=str) |
| 45 | + parser.add_argument("--aot", action='store_true', help="Test with AOT") |
| 46 | + args = parser.parse_args() |
| 47 | + |
| 48 | + test_aot = False |
| 49 | + suffix = ".wasm" |
| 50 | + if not args.aot: |
| 51 | + print("Test with interpreter mode") |
| 52 | + else: |
| 53 | + print("Test with AOT mode") |
| 54 | + test_aot = True |
| 55 | + suffix = ".aot" |
| 56 | + wasm_apps_dir = args.working_directory |
| 57 | + compile_wasm_files_to_aot(wasm_apps_dir) |
| 58 | + |
| 59 | + ret = 1 |
| 60 | + app_server = None |
| 61 | + try: |
| 62 | + print("\n================================") |
| 63 | + print("Test TCP server and client") |
| 64 | + cmd = "./iwasm --addr-pool=0.0.0.0/15 tcp_server" + suffix |
| 65 | + app_server = start_server(cmd, args.working_directory) |
| 66 | + # wait for a second |
| 67 | + time.sleep(1) |
| 68 | + cmd = "./iwasm --addr-pool=127.0.0.1/15 tcp_client" + suffix |
| 69 | + for i in range(5): |
| 70 | + run_cmd(cmd, args.working_directory) |
| 71 | + |
| 72 | + print("\n================================") |
| 73 | + print("Test UDP server and client") |
| 74 | + cmd = "./iwasm --addr-pool=0.0.0.0/15 udp_server" + suffix |
| 75 | + app_server = start_server(cmd, args.working_directory) |
| 76 | + # wait for a second |
| 77 | + time.sleep(1) |
| 78 | + cmd = "./iwasm --addr-pool=127.0.0.1/15 udp_client" + suffix |
| 79 | + for i in range(5): |
| 80 | + run_cmd(cmd, args.working_directory) |
| 81 | + |
| 82 | + print("\n=====================================================") |
| 83 | + print("Sleep 80 seconds to wait TCP server port actually close") |
| 84 | + time.sleep(80) |
| 85 | + |
| 86 | + print("\n================================") |
| 87 | + print("Test send and receive") |
| 88 | + cmd = "./iwasm --addr-pool=127.0.0.1/0 ./send_recv" + suffix |
| 89 | + run_cmd(cmd, args.working_directory) |
| 90 | + |
| 91 | + print("\n================================") |
| 92 | + print("Test socket options") |
| 93 | + cmd = "./iwasm socket_opts" + suffix |
| 94 | + run_cmd(cmd, args.working_directory) |
| 95 | + |
| 96 | + print("\n================================") |
| 97 | + print("Test timeout server and client") |
| 98 | + cmd = "./iwasm --addr-pool=0.0.0.0/15 timeout_server" + suffix |
| 99 | + app_server = start_server(cmd, args.working_directory) |
| 100 | + # wait for a second |
| 101 | + time.sleep(1) |
| 102 | + cmd = "./iwasm --addr-pool=127.0.0.1/15 timeout_client" + suffix |
| 103 | + run_cmd(cmd, args.working_directory) |
| 104 | + |
| 105 | + print("\n==========================================") |
| 106 | + print("Test multicast_client and multicast_server") |
| 107 | + cmd = "./iwasm --addr-pool=0.0.0.0/0,::/0 multicast_client.wasm 224.0.0.1" |
| 108 | + app_server = start_server(cmd, args.working_directory) |
| 109 | + # wait for a second |
| 110 | + time.sleep(1) |
| 111 | + cmd = "./multicast_server 224.0.0.1" |
| 112 | + run_cmd(cmd, args.working_directory) |
| 113 | + |
| 114 | + cmd = "./iwasm --addr-pool=0.0.0.0/0,::/0 multicast_client.wasm FF02:113D:6FDD:2C17:A643:FFE2:1BD1:3CD2" |
| 115 | + app_server = start_server(cmd, args.working_directory) |
| 116 | + # wait for a second |
| 117 | + time.sleep(1) |
| 118 | + cmd = "./multicast_server FF02:113D:6FDD:2C17:A643:FFE2:1BD1:3CD2" |
| 119 | + run_cmd(cmd, args.working_directory) |
| 120 | + |
| 121 | + print("\n================================") |
| 122 | + print("Test address resolving") |
| 123 | + cmd = "./iwasm --allow-resolve=*.com addr_resolve.wasm github.com" |
| 124 | + cmd = "./multicast_server FF02:113D:6FDD:2C17:A643:FFE2:1BD1:3CD2" |
| 125 | + run_cmd(cmd, args.working_directory) |
| 126 | + |
| 127 | + # wait for a second |
| 128 | + time.sleep(1) |
| 129 | + |
| 130 | + print("--> All pass") |
| 131 | + ret = 0 |
| 132 | + except AssertionError: |
| 133 | + traceback.print_exc() |
| 134 | + finally: |
| 135 | + app_server.kill() |
| 136 | + |
| 137 | + return ret |
| 138 | + |
| 139 | + |
| 140 | +if __name__ == "__main__": |
| 141 | + sys.exit(main()) |
0 commit comments