1#!/usr/bin/python 2# 3# Server that will communicate with Vim through the netbeans interface. 4# Used by test_netbeans.vim. 5# 6# This requires Python 2.6 or later. 7 8from __future__ import print_function 9import socket 10import sys 11import time 12import threading 13import re 14 15try: 16 # Python 3 17 import socketserver 18except ImportError: 19 # Python 2 20 import SocketServer as socketserver 21 22class ThreadedTCPRequestHandler(socketserver.BaseRequestHandler): 23 24 def process_msgs(self, msgbuf): 25 # Process all the received netbeans commands/responses/events from Vim. 26 # Each one is separated by a newline character. If a partial command 27 # is received, process it later after the rest of it is received. 28 while True: 29 (line, sep, rest) = msgbuf.partition('\n') 30 if sep == '': 31 # received partial line 32 return line 33 msgbuf = rest 34 35 # Process a command only after receiving a newline. 36 response = '' 37 if line.find('Xcmdbuf') > 0: 38 name = line.split('"')[1] 39 response = '1:putBufferNumber!15 "' + name + '"\n' 40 response += '1:startDocumentListen!16\n' 41 elif re.match('1:insert=.* "\\\\n"', line): 42 # extract the command from the previous line 43 cmd = re.search('.*"(.*)"', self.prev_line).group(1) 44 45 # map of test names and the netbeans commands/functions 46 testmap = { 47 'getCursor_Test' : '0:getCursor/30\n', 48 'E627_Test' : '0 setReadOnly!31\n', 49 'E628_Test' : '0:setReadOnly 32\n', 50 'E632_Test' : '0:getLength/33\n', 51 'E633_Test' : '0:getText/34\n', 52 'E634_Test' : '0:remove/35 1 1\n', 53 'E635_Test' : '0:insert/36 0 "line1\\n"\n', 54 'E636_Test' : '0:create!37\n', 55 'E637_Test' : '0:startDocumentListen!38\n', 56 'E638_Test' : '0:stopDocumentListen!39\n', 57 'E639_Test' : '0:setTitle!40 "Title"\n', 58 'E640_Test' : '0:initDone!41\n', 59 'E641_Test' : '0:putBufferNumber!42 "XSomeBuf"\n', 60 'E642_Test' : '9:putBufferNumber!43 "XInvalidBuf"\n', 61 'E643_Test' : '0:setFullName!44 "XSomeBuf"\n', 62 'E644_Test' : '0:editFile!45 "Xfile3"\n', 63 'E645_Test' : '0:setVisible!46 T\n', 64 'E646_Test' : '0:setModified!47 T\n', 65 'E647_Test' : '0:setDot!48 1/1\n', 66 'E648_Test' : '0:close!49\n', 67 'E650_Test' : '0:defineAnnoType!50 1 "abc" "a" "a" 1 1\n', 68 'E651_Test' : '0:addAnno!51 1 1 1 1\n', 69 'E652_Test' : '0:getAnno/52 8\n', 70 'editFile_Test' : '2:editFile!53 "Xfile3"\n', 71 'getLength_Test' : '2:getLength/54\n', 72 'getModified_Test' : '2:getModified/55\n', 73 'getText_Test' : '2:getText/56\n', 74 'setDot_Test' : '2:setDot!57 3/6\n', 75 'setDot2_Test' : '2:setDot!57 9\n', 76 'startDocumentListen_Test' : '2:startDocumentListen!58\n', 77 'stopDocumentListen_Test' : '2:stopDocumentListen!59\n', 78 'define_anno_Test' : '2:defineAnnoType!60 1 "s1" "x" "=>" blue none\n', 79 'E532_Test' : '2:defineAnnoType!61 1 "s1" "x" "=>" aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa none\n', 80 'add_anno_Test' : '2:addAnno!62 1 1 2/1 0\n', 81 'get_anno_Test' : '2:getAnno/63 1\n', 82 'remove_anno_Test' : '2:removeAnno!64 1\n', 83 'getModifiedAll_Test' : '0:getModified/65\n', 84 'create_Test' : '3:create!66\n', 85 'setTitle_Test' : '3:setTitle!67 "Xfile4"\n', 86 'setFullName_Test' : '3:setFullName!68 "Xfile4"\n', 87 'initDone_Test' : '3:initDone!69\n', 88 'setVisible_Test' : '3:setVisible!70 T\n', 89 'setModtime_Test' : '3:setModtime!71 6\n', 90 'insert_Test' : '3:insert/72 0 "line1\\nline2\\n"\n', 91 'remove_Test' : '3:remove/73 3 4\n', 92 'remove_invalid_offset_Test' : '3:remove/74 900 4\n', 93 'remove_invalid_count_Test' : '3:remove/75 1 800\n', 94 'guard_Test' : '3:guard!76 8 7\n', 95 'setModified_Test' : '3:setModified!77 T\n', 96 'setModifiedClear_Test' : '3:setModified!77 F\n', 97 'insertDone_Test' : '3:insertDone!78 T F\n', 98 'saveDone_Test' : '3:saveDone!79\n', 99 'invalidcmd_Test' : '3:invalidcmd!80\n', 100 'invalidfunc_Test' : '3:invalidfunc/81\n', 101 'removeAnno_fail_Test' : '0:removeAnno/82 1\n', 102 'guard_fail_Test' : '0:guard/83 1 1\n', 103 'save_fail_Test' : '0:save/84\n', 104 'netbeansBuffer_fail_Test' : '0:netbeansBuffer/85 T\n', 105 'setExitDelay_Test' : '0:setExitDelay!86 2\n', 106 'setReadOnly_Test' : '3:setReadOnly!87 T\n', 107 'setReadOnlyClear_Test' : '3:setReadOnly!88 F\n', 108 'save_Test' : '3:save!89\n', 109 'close_Test' : '3:close!90\n', 110 'specialKeys_Test' : '0:specialKeys!91 "F12 F13 C-F13"\n', 111 'nbbufwrite_Test' : '4:editFile!92 "XnbBuffer"\n4:netbeansBuffer!93 T\n', 112 'startAtomic_Test' : '0:startAtomic!94\n', 113 'endAtomic_Test' : '0:endAtomic!95\n', 114 'AnnoScale_Test' : "".join(['2:defineAnnoType!60 ' + str(i) + ' "s' + str(i) + '" "x" "=>" blue none\n' for i in range(2, 26)]), 115 'detach_Test' : '2:close!96\n1:close!97\nDETACH\n' 116 } 117 # execute the specified test 118 if cmd not in testmap: 119 print("=== invalid command %s ===" % (cmd)) 120 else: 121 response = testmap[cmd] 122 elif line.find('disconnect') > 0: 123 # we're done 124 self.server.shutdown() 125 return 126 127 # save the current line, this is used as the test to run after 128 # receiving a newline only line. 129 self.prev_line = line 130 131 if len(response) > 0: 132 self.request.sendall(response.encode('utf-8')) 133 # Write the response into the file, so that the test can knows 134 # the command was sent. 135 with open("Xnetbeans", "a") as myfile: 136 myfile.write('send: ' + response) 137 if self.debug: 138 with open("save_Xnetbeans", "a") as myfile: 139 myfile.write('send: ' + response) 140 141 def handle(self): 142 print("=== socket opened ===") 143 # To preserve the Xnetbeans file as save_Xnetbeans, set debug to 1 144 self.debug = 0 145 self.prev_line = '' 146 msgbuf = '' 147 while True: 148 try: 149 received = self.request.recv(4096).decode('utf-8') 150 except socket.error: 151 print("=== socket error ===") 152 break 153 except IOError: 154 print("=== socket closed ===") 155 break 156 if received == '': 157 print("=== socket closed ===") 158 break 159 print("received: {0}".format(received)) 160 161 # Write the received line into the file, so that the test can check 162 # what happened. 163 with open("Xnetbeans", "a") as myfile: 164 myfile.write(received) 165 if self.debug: 166 with open("save_Xnetbeans", "a") as myfile: 167 myfile.write(received) 168 169 # Can receive more than one line in a response or a partial line. 170 # Accumulate all the received characters and process one line at 171 # a time. 172 msgbuf += received 173 msgbuf = self.process_msgs(msgbuf) 174 175class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer): 176 pass 177 178def writePortInFile(port): 179 # Write the port number in Xportnr, so that the test knows it. 180 f = open("Xportnr", "w") 181 f.write("{0}".format(port)) 182 f.close() 183 184if __name__ == "__main__": 185 HOST, PORT = "localhost", 0 186 187 server = ThreadedTCPServer((HOST, PORT), ThreadedTCPRequestHandler) 188 ip, port = server.server_address 189 190 # Start a thread with the server. That thread will then start a new thread 191 # for each connection. 192 server_thread = threading.Thread(target=server.serve_forever) 193 server_thread.start() 194 195 writePortInFile(port) 196 197 print("Listening on port {0}".format(port)) 198 199 # Main thread terminates, but the server continues running 200 # until server.shutdown() is called. 201 try: 202 while server_thread.isAlive(): 203 server_thread.join(1) 204 except (KeyboardInterrupt, SystemExit): 205 server.shutdown() 206