 1 #! /usr/bin/env python
 2 import commands
 3 import socket
 4 import time
 5 import os
 6 import multiprocessing
 7 import uuid
 8 from BaseHTTPServer import BaseHTTPRequestHandler,HTTPServer
 9 
10 #HTTPServer的监听端口
11 PORT=12345
12 
13 class HttpHandler(BaseHTTPRequestHandler):
14     tmpfile=''
15     #处理POST消息
16     def do_POST(self):
17         print self.path
18         content_len = int(self.headers.getheader('content-length',0))
19         #获取timeout
20         timeout = int(self.headers.getheader('timeout',0))
21         if timeout==0:
22             timeout=5
23         #解析消息并存储为文件
24         script=self.rfile.read(content_len)
25         x=uuid.uuid4()
26         self.tmpfile="."+str(x.int)
27         fd=open(self.tmpfile,'w')
28         fd.write(script)
29         fd.close()
30         os.system("chmod +x "+self.tmpfile)
31         script="./"+self.tmpfile
32         #执行脚本
33         self.ExecuteScript(script,timeout)
34         
35     def ExecuteScript(self,script,timeout=5):
36         #启动另一个进程执行脚本
37         p=multiprocessing.Process(target=self.ScriptWorker,args=(script,))
38         p.start()
39         i=0
40         while i<timeout:
41             if(not p.is_alive()):
42                 return "successful"
43             else:
44                 time.sleep(1)
45             i=i+1
46         #超时的话终止进程并杀掉执行任务的进程
47         p.terminate()
48         os.system("kill -9 "+str(p.pid))
49         self.send_error(400,"time out")
50         self.request.shutdown(socket.SHUT_RDWR)
51         self.request.close()
52         #删除临时文件
53         if self.tmpfile != '':
54             os.system("rm "+self.tmpfile)
55         self.tmpfile=''
56         
57     def ScriptWorker(self,script):
58         #执行脚本，返回状态码和输出
59         (status,result)=commands.getstatusoutput(script)
60         print script
61         print result
62         #如果成功返回200，如果失败返回400
63         if status == 0:
64             self.send_response(200)
65         else:
66             self.send_response(400)
67         self.send_header('Content-type','text/html')
68         self.end_headers()
69         self.wfile.write(result)
70         #删除临时文件
71         if self.tmpfile != '':
72             os.system("rm "+self.tmpfile)
73         self.tmpfile=''
74         
75 if __name__=='__main__':
76     os.system('rm .*')
77     server_address=('0.0.0.0',PORT)
78     http_server=HTTPServer(server_address,HttpHandler)
79     http_server.serve_forever()
80