api

  1. # coding:utf-8
  2. """
  3. 雷池API功能大全脚本，分多个模块。
  4. """
  5. import os
  7. print(os.getcwd())
  8. print('WAFapi')

syslog server

写接口自动化测试的时候需要自己搭一个syslog服务器，找到了一个udp协议的还挺好用
https://gist.github.com/marcelom/4218010
因为雷池还需要测tcp协议的，所以参考下面的文档完善了一下tcp的部分。
https://docs.python.org/3.4/library/socketserver.html

  1. #!/usr/bin/env python3
  3. LOG_FILE = 'syslog.log'
  4. HOST, PORT = "0.0.0.0", 514
  5. MODE = 'udp'
  7. import sys
  8. import getopt
  9. import logging
  10. import socketserver
  13. class SyslogUDPHandler(socketserver.BaseRequestHandler):
  15.     def handle(self):
  16.         self.data = bytes.decode(self.request[0].strip())
  17.         socket = self.request[1]
  18.         print( "%s : " % self.client_address[0], str(self.data))
  19.         logging.info(str(self.data))
  21. class SyslogTCPHandler(socketserver.BaseRequestHandler):
  23.     def handle(self):
  24.         # self.request is the TCP socket connected to the client
  25.         self.data = self.request.recv(1024).strip()
  26.         print("{} wrote:".format(self.client_address[0]))
  27.         print(self.data)
  28.         # just send back the same data, but upper-cased
  29.         self.request.sendall(self.data.upper())
  30.         logging.info(str(self.data))
  32. def show_help():
  33.     """
  34.     Help message to display if bad arguments or help requested.
  35.     """
  36.     print('-------------------------------------------------------------------------------')
  38.     print('')
  39.     print(' -l <logfile>      -- Specify the log file to write to. Default is pysyslog.log')
  40.     print(' -p <port>         -- Specify the port to listen on. Default is 514')
  41.     print(' -i <host>      -- Specify the IP address to listen on. Default is 0.0.0.0')
  42.     print(' -m <mode>      -- Specify the mode tcp or udp. Default is udp')
  43.     print('')
  44.     print('Example:')
  45.     print(' python3 server_server.py -l syslog.log -p 514 -i 192.168.12.1 -m udp ')
  46.     print('-------------------------------------------------------------------------------')
  47.     sys.exit(2)
  49. def main(argv):
  50.     logfile = LOG_FILE
  51.     host = HOST
  52.     port = PORT
  53.     mode = MODE
  55.     try:
  56.         opts, args = getopt.getopt(argv, "hl:p:i:m:", ["logfile=", "port=", "host=","mode="])
  57.     except getopt.GetoptError:
  58.         show_help()
  59.     for opt, arg in opts:
  60.         if opt == '-h':
  61.             show_help()
  62.             sys.exit()
  63.         elif opt in ("-l", "--logfile"):
  64.             logfile = arg
  65.         elif opt in ("-p", "--port"):
  66.             port = int(arg)
  67.         elif opt in ("-i", "--host"):
  68.             host = arg
  69.         elif opt in ("-m", "--mode"):
  70.             if arg!='tcp':
  71.                 mode = 'udp'
  72.             else:
  73.                 mode = arg
  75.     print('       Writing to: {}'.format(logfile))
  76.     print('     Listening on: {}'.format(port))
  77.     print('         Bound to: {}'.format(host))
  78.     print('         Bound to: {}'.format(mode))
  79.     print('')
  80.     print('Starting Server, press Ctrl+C to shutdown.')
  81.     print('')
  83.     logging.basicConfig(level=logging.INFO, format='%(message)s', datefmt='', filename=logfile, filemode='a')
  85.     if mode=='tcp':
  86.         try:
  87.             #server = socketserver.UDPServer((HOST,PORT), SyslogUDPHandler)
  88.             server = socketserver.TCPServer((host, port), SyslogTCPHandler)
  89.             server.serve_forever(poll_interval=0.5)
  90.         except (IOError, SystemExit):
  91.             raise
  92.         except KeyboardInterrupt:
  93.             print ("Crtl+C Pressed. Shutting down.")
  94.     else:
  95.         try:
  96.             server = socketserver.UDPServer((host,port), SyslogUDPHandler)
  97.             #server = socketserver.TCPServer((HOST, PORT), SyslogTCPHandler)
  98.             server.serve_forever(poll_interval=0.5)
  99.         except (IOError, SystemExit):
  100.             raise
  101.         except KeyboardInterrupt:
  102.             print ("Crtl+C Pressed. Shutting down.")
  104. if __name__ == "__main__":
  105.     main(sys.argv[1:])

rtcp-master

把虚拟机的443转发到本机1443以便让局域网他人访问
./rtcp.py l:1443 c:10.211.55.19:443
https://github.com/knownsec/rtcp