1*4882a593Smuzhiyun# 2*4882a593Smuzhiyun# SPDX-License-Identifier: MIT 3*4882a593Smuzhiyun# 4*4882a593Smuzhiyun 5*4882a593Smuzhiyunimport http.server 6*4882a593Smuzhiyunimport multiprocessing 7*4882a593Smuzhiyunimport os 8*4882a593Smuzhiyunimport traceback 9*4882a593Smuzhiyunimport signal 10*4882a593Smuzhiyunimport logging 11*4882a593Smuzhiyunfrom socketserver import ThreadingMixIn 12*4882a593Smuzhiyun 13*4882a593Smuzhiyunclass HTTPServer(ThreadingMixIn, http.server.HTTPServer): 14*4882a593Smuzhiyun 15*4882a593Smuzhiyun def server_start(self, root_dir, logger): 16*4882a593Smuzhiyun os.chdir(root_dir) 17*4882a593Smuzhiyun self.serve_forever() 18*4882a593Smuzhiyun 19*4882a593Smuzhiyunclass HTTPRequestHandler(http.server.SimpleHTTPRequestHandler): 20*4882a593Smuzhiyun 21*4882a593Smuzhiyun def log_message(self, format_str, *args): 22*4882a593Smuzhiyun pass 23*4882a593Smuzhiyun 24*4882a593Smuzhiyunclass HTTPService(object): 25*4882a593Smuzhiyun 26*4882a593Smuzhiyun def __init__(self, root_dir, host='', port=0, logger=None): 27*4882a593Smuzhiyun self.root_dir = root_dir 28*4882a593Smuzhiyun self.host = host 29*4882a593Smuzhiyun self.port = port 30*4882a593Smuzhiyun if not logger: 31*4882a593Smuzhiyun logger = logging.getLogger() 32*4882a593Smuzhiyun self.logger = logger 33*4882a593Smuzhiyun 34*4882a593Smuzhiyun def start(self): 35*4882a593Smuzhiyun print(self.root_dir) 36*4882a593Smuzhiyun if not os.path.exists(self.root_dir): 37*4882a593Smuzhiyun self.logger.info("Not starting HTTPService for directory %s which doesn't exist" % (self.root_dir)) 38*4882a593Smuzhiyun return 39*4882a593Smuzhiyun 40*4882a593Smuzhiyun self.server = HTTPServer((self.host, self.port), HTTPRequestHandler) 41*4882a593Smuzhiyun if self.port == 0: 42*4882a593Smuzhiyun self.port = self.server.server_port 43*4882a593Smuzhiyun self.process = multiprocessing.Process(target=self.server.server_start, args=[self.root_dir, self.logger]) 44*4882a593Smuzhiyun 45*4882a593Smuzhiyun # The signal handler from testimage.bbclass can cause deadlocks here 46*4882a593Smuzhiyun # if the HTTPServer is terminated before it can restore the standard 47*4882a593Smuzhiyun #signal behaviour 48*4882a593Smuzhiyun orig = signal.getsignal(signal.SIGTERM) 49*4882a593Smuzhiyun signal.signal(signal.SIGTERM, signal.SIG_DFL) 50*4882a593Smuzhiyun self.process.start() 51*4882a593Smuzhiyun signal.signal(signal.SIGTERM, orig) 52*4882a593Smuzhiyun 53*4882a593Smuzhiyun if self.logger: 54*4882a593Smuzhiyun self.logger.info("Started HTTPService on %s:%s" % (self.host, self.port)) 55*4882a593Smuzhiyun 56*4882a593Smuzhiyun 57*4882a593Smuzhiyun def stop(self): 58*4882a593Smuzhiyun if hasattr(self, "server"): 59*4882a593Smuzhiyun self.server.server_close() 60*4882a593Smuzhiyun if hasattr(self, "process"): 61*4882a593Smuzhiyun self.process.terminate() 62*4882a593Smuzhiyun self.process.join() 63*4882a593Smuzhiyun if self.logger: 64*4882a593Smuzhiyun self.logger.info("Stopped HTTPService on %s:%s" % (self.host, self.port)) 65*4882a593Smuzhiyun 66