#!/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()