| #!/usr/bin/env python |
| |
| # |
| # Copyright 2007, The Android Open Source Project |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # |
| |
| """ |
| axl.py: HTTP Client torture tester |
| |
| """ |
| |
| import sys, time |
| |
| from twisted.internet import protocol, reactor, defer |
| from twisted.internet.protocol import ServerFactory, Protocol |
| |
| import singletonmixin, log |
| |
| class BaseProtocol(Protocol): |
| def __init__(self): |
| self.log = log.Log.getInstance() |
| |
| def write(self, data): |
| self.log("BaseProtocol.write()", len(data), data) |
| return self.transport.write(data) |
| |
| def dataReceived(self, data): |
| self.log("BaseProtocol.dataReceived()", len(data), data) |
| |
| def connectionMade(self): |
| self.log("BaseProtocol.connectionMade()") |
| self.transport.setTcpNoDelay(1) # send immediately |
| |
| def connectionLost(self, reason): |
| self.log("BaseProtocol.connectionLost():", reason) |
| |
| def sendResponse(self, response): |
| self.write("HTTP/1.1 200 OK\r\n") |
| self.write("Content-Length: %d\r\n\r\n" % len(response)) |
| if len(response) > 0: |
| self.write(response) |
| |
| |
| # Tests |
| # 8000: test driven by resource request |
| |
| class Drop(BaseProtocol): |
| """Drops connection immediately after connect""" |
| PORT = 8001 |
| def connectionMade(self): |
| BaseProtocol.connectionMade(self) |
| self.transport.loseConnection() |
| |
| class ReadAndDrop(BaseProtocol): |
| """Read 1st line of request, then drop connection""" |
| PORT = 8002 |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| self.transport.loseConnection() |
| |
| class GarbageStatus(BaseProtocol): |
| """Send garbage statusline""" |
| PORT = 8003 |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| self.write("welcome to the jungle baby\r\n") |
| |
| class BadHeader(BaseProtocol): |
| """Drop connection after a header is half-sent""" |
| PORT = 8004 |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| self.write("HTTP/1.1 200 OK\r\n") |
| self.write("Cache-Contr") |
| time.sleep(1) |
| self.transport.loseConnection() |
| |
| class PauseHeader(BaseProtocol): |
| """Pause for a second in middle of a header""" |
| PORT = 8005 |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| self.write("HTTP/1.1 200 OK\r\n") |
| self.write("Cache-Contr") |
| time.sleep(1) |
| self.write("ol: private\r\n\r\nwe've got fun and games") |
| time.sleep(1) |
| self.transport.loseConnection() |
| |
| class Redirect(BaseProtocol): |
| PORT = 8006 |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| self.write("HTTP/1.1 302 Moved Temporarily\r\n") |
| self.write("Content-Length: 0\r\n") |
| self.write("Location: http://shopping.yahoo.com/p:Canon PowerShot SD630 Digital Camera:1993588104;_ylc=X3oDMTFhZXNmcjFjBF9TAzI3MTYxNDkEc2VjA2ZwLXB1bHNlBHNsawNyc3NfcHVsc2U0LmluYw--\r\n\r\n") |
| self.transport.loseConnection() |
| |
| class DataDrop(BaseProtocol): |
| """Drop connection in body""" |
| PORT = 8007 |
| def dataReceived(self, data): |
| if data.find("favico") >= 0: |
| self.write("HTTP/1.1 404 Not Found\r\n\r\n") |
| self.transport.loseConnection() |
| return |
| |
| BaseProtocol.dataReceived(self, data) |
| self.write("HTTP/1.1 200 OK\r\n") |
| # self.write("Content-Length: 100\r\n\r\n") |
| self.write("\r\n") |
| # self.write("Data cuts off < 100 here!") |
| # time.sleep(4) |
| self.transport.loseConnection() |
| |
| class DropOnce(BaseProtocol): |
| """Drop every other connection""" |
| PORT = 8008 |
| COUNT = 0 |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| self.write("HTTP/1.1 200 OK\r\n") |
| self.write("Content-Length: 5\r\n\r\n") |
| |
| if (not(DropOnce.COUNT & 1)): |
| self.write("HE") |
| else: |
| self.write("HELLO") |
| self.transport.loseConnection() |
| |
| DropOnce.COUNT += 1 |
| |
| class NoCR(BaseProtocol): |
| """Send headers without carriage returns""" |
| PORT = 8009 |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| self.write("HTTP/1.1 200 OK\n") |
| self.write("Content-Length: 5\n\n") |
| |
| self.write("HELLO") |
| self.transport.loseConnection() |
| |
| class PipeDrop(BaseProtocol): |
| PORT = 8010 |
| COUNT = 0 |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| if not PipeDrop.COUNT % 3: |
| self.write("HTTP/1.1 200 OK\n") |
| self.write("Content-Length: 943\n\n") |
| |
| self.write(open("./stfu.jpg").read()) |
| PipeDrop.COUNT += 1 |
| |
| else: |
| self.transport.loseConnection() |
| PipeDrop.COUNT += 1 |
| |
| class RedirectLoop(BaseProtocol): |
| """Redirect back to same resource""" |
| PORT = 8011 |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| self.write("HTTP/1.1 302 Moved Temporarily\r\n") |
| self.write("Content-Length: 0\r\n") |
| self.write("Location: http://localhost:8011/\r\n") |
| self.write("\r\n") |
| self.transport.loseConnection() |
| |
| class ReadAll(BaseProtocol): |
| """Read entire request""" |
| PORT = 8012 |
| |
| def connectionMade(self): |
| self.count = 0 |
| |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| self.count += len(data) |
| if self.count == 190890: |
| self.transport.loseConnection() |
| |
| class Timeout(BaseProtocol): |
| """Timout sending body""" |
| PORT = 8013 |
| |
| def connectionMade(self): |
| self.count = 0 |
| |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| if self.count == 0: self.write("HTTP/1.1 200 OK\r\n\r\n") |
| self.count += 1 |
| |
| class SlowResponse(BaseProtocol): |
| """Ensure client does not time out on slow writes""" |
| PORT = 8014 |
| |
| def connectionMade(self): |
| self.count = 0 |
| |
| def dataReceived(self, data): |
| BaseProtocol.dataReceived(self, data) |
| if self.count == 0: self.write("HTTP/1.1 200 OK\r\n\r\n") |
| self.sendPack(0) |
| |
| def sendPack(self, count): |
| if count > 10: |
| self.transport.loseConnection() |
| |
| self.write("all work and no play makes jack a dull boy %s\n" % count) |
| d = defer.Deferred() |
| d.addCallback(self.sendPack) |
| reactor.callLater(15, d.callback, count + 1) |
| |
| |
| # HTTP/1.1 200 OK |
| # Cache-Control: private |
| # Content-Type: text/html |
| # Set-Cookie: PREF=ID=10644de62c423aa5:TM=1155044293:LM=1155044293:S=0lHtymefQRs2j7nD; expires=Sun, 17-Jan-2038 19:14:07 GMT; path=/; domain=.google.com |
| # Server: GWS/2.1 |
| # Transfer-Encoding: chunked |
| # Date: Tue, 08 Aug 2006 13:38:13 GMT |
| |
| def main(): |
| # Initialize log |
| log.Log.getInstance(sys.stdout) |
| |
| for protocol in Drop, ReadAndDrop, GarbageStatus, BadHeader, PauseHeader, \ |
| Redirect, DataDrop, DropOnce, NoCR, PipeDrop, RedirectLoop, ReadAll, \ |
| Timeout, SlowResponse: |
| factory = ServerFactory() |
| factory.protocol = protocol |
| reactor.listenTCP(protocol.PORT, factory) |
| |
| |
| reactor.run() |
| |
| if __name__ == '__main__': |
| main() |
