1import os
2import time
3
4def main():
5    if (os.path.exists("/tmp/RT-Thread-STM32MP1_CM4.elf")):
6        os.system("mv /tmp/RT-Thread-STM32MP1_CM4.elf /lib/firmware/")
7    os.system("echo stop > /sys/class/remoteproc/remoteproc0/state")
8    os.system("echo RT-Thread-STM32MP1_CM4.elf > /sys/class/remoteproc/remoteproc0/firmware")
9    os.system("echo start > /sys/class/remoteproc/remoteproc0/state")
10    os.system("stty -onlcr -echo -F /dev/ttyRPMSG0")
11    os.system("cat /dev/ttyRPMSG0 &")
12    time.sleep(1)
13    print("\nYou can input \"exit\" to quit rt-thread shell!\n")
14    os.system("echo version >/dev/ttyRPMSG0")
15
16    while True:
17        a = input()
18        if a == "exit":
19           os.system("echo stop > /sys/class/remoteproc/remoteproc0/state")
20           return
21
22        os.system("echo {0} >/dev/ttyRPMSG0".format(a))
23
24
25if __name__ == "__main__":
26    main()
27