import const import cmd import threading import socket import time import calendar import sys import struct import cPickle import math #------------------------------------------------------------------------------ ## @package mlx.pyuipc_sim # # Simulator of the PyUIPC module. # # This is a simulation of the PyUIPC module emulating offsets that are needed # by the logger. # # This module can also be run as a program, in which case it connects to an # already running logger (that uses this module in place of the real pyuipc), # and various commands can be given to query or modify the values of the # offsets. There is a 'help' command and completion also works. # # This module is used instead of the real PyUIPC module, if the program is not # running on Windows or the FORCE_PYUIPC_SIM environment variable is present. #------------------------------------------------------------------------------ #------------------------------------------------------------------------------ ## Version constants SIM_ANY=0 SIM_FS98=1 SIM_FS2K=2 SIM_CFS2=3 SIM_CFS1=4 SIM_FLY=5 SIM_FS2K2=6 SIM_FS2K4=7 SIM_FSX=8 #------------------------------------------------------------------------------ ## Error constants ERR_OK=0 ERR_OPEN=1 ERR_NOFS=2 ERR_REGMSG=3 ERR_ATOM=4 ERR_MAP=5 ERR_VIEW=6 ERR_VERSION=7 ERR_WRONGFS=8 ERR_NOTOPEN=9 ERR_NODATA=10 ERR_TIMEOUT=11 ERR_SENDMSG=12 ERR_DATA=13 ERR_RUNNING=14 ERR_SIZE=15 #------------------------------------------------------------------------------ ## The version of FSUIPC fsuipc_version=0x0401 lib_version=0x0302 fs_version=SIM_FS2K4 #------------------------------------------------------------------------------ class FSUIPCException(Exception): """FSUIPC exception class. It contains a member variable named errorCode. The string is a text describing the error.""" errors=["OK", "Attempt to Open when already Open", "Cannot link to FSUIPC or WideClient", "Failed to Register common message with Windows", "Failed to create Atom for mapping filename", "Failed to create a file mapping object", "Failed to open a view to the file map", "Incorrect version of FSUIPC, or not FSUIPC", "Sim is not version requested", "Call cannot execute, link not Open", "Call cannot execute: no requests accumulated", "IPC timed out all retries", "IPC sendmessage failed all retries", "IPC request contains bad data", "Maybe running on WideClient, but FS not running on Server, or wrong FSUIPC", "Read or Write request cannot be added, memory for Process is full"] def __init__(self, errorCode): """ Construct the exception """ if errorCode>12) & 0x0f bcd *= 10 bcd += (value>>8) & 0x0f bcd *= 10 bcd += (value>>4) & 0x0f bcd *= 10 bcd += (value>>0) & 0x0f return bcd def __init__(self): """Construct the values with defaults.""" self._timeOffset = 0 self.airPath = "C:\\Program Files\\Microsoft Games\\" \ "FS9\\Aircraft\\Cessna\\cessna172.air" self.aircraftName = "Cessna 172SP" self.flapsNotches = [0, 1, 2, 5, 10, 15, 25, 30, 40] self.fuelCapacities = [10000.0, 5000.0, 5000.0, 5000.0, 5000.0, 5000.0, 5000.0, 5000.0, 5000.0, 5000.0] self.latitude = 47.5 self.longitude = 19.05 self.paused = False self.frozen = False self.replay = False self.slew = False self.overspeed = False self.stalled = False self.onTheGround = True self.zfw = 50000.0 self.fuelWeights = [0.0, 3000.0, 3000.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0] # Wikipedia: "Jet fuel", Jet A-1 density at 15*C .804kg/l -> 6.7 pounds/gallon self.fuelWeight = 6.70970518 self.heading = 220.0 self.pitch = 0.0 self.bank = 0.0 self.ias = 0.0 self.vs = 0.0 self.tdRate = 0.0 self.radioAltitude = None self.altitude = 513.0 self.gLoad = 1.0 self.flapsControl = 0.0 self.flaps = 0.0 self.navLightsOn = True self.antiCollisionLightsOn = False self.landingLightsOn = False self.strobeLightsOn = False self.pitot = False self.parking = True self.gearControl = 1.0 self.noseGear = 1.0 self.spoilersArmed = False self.spoilers = 0.0 self.altimeter = 1013.0 self.nav1 = 117.3 self.nav1_obs = 128 self.nav2 = 109.5 self.nav2_obs = 308 self.adf1 = 382.7 self.adf2 = 1540.6 self.squawk = 2200 self.windSpeed = 8.0 self.windDirection = 300.0 self.visibility = 10000 self.n1 = [0.0, 0.0, 0.0] self.throttles = [0.0, 0.0, 0.0] self.payloadCount = 1 self.payload = [] for i in range(0, 61): self.payload.append(0.0) self.textScrolling = False self.message = "" self.messageDuration = 0 self.hotkeyTable = [] for i in range(0, Values.HOTKEY_SIZE): self.hotkeyTable.append([0, 0, 0, 0]) self.cog = 0.27 self.pmdg_737ng_switches = 0 self.pmdg_737ngx_lts_positionsw = 0 self.xpdrC = False self.apMaster = False self.apHeadingHold = False self.apHeading = 124 self.apAltitudeHold = False self.apAltitude = 7000 def read(self, offset, type): """Read the value at the given offset.""" try: return self._read(offset, type) except Exception, e: print "failed to read offset %04x: %s" % (offset, str(e)) raise FSUIPCException(ERR_DATA) def _read(self, offset, type): """Read the value at the given offset.""" if offset==0x023a: # Second of time return self._readUTC().tm_sec elif offset==0x023b: # Hour of Zulu time return self._readUTC().tm_hour elif offset==0x023c: # Minute of Zulu time return self._readUTC().tm_min elif offset==0x023e: # Day number on year return self._readUTC().tm_yday elif offset==0x0240: # Year in FS return self._readUTC().tm_year elif offset==0x0264: # Paused return 1 if self.paused else 0 elif offset==0x029c: # Pitot return 1 if self.pitot else 0 elif offset==0x02b4: # Ground speed # FIXME: calculate TAS first, then from the heading and # wind the GS return int(self.ias * 65536.0 * 1852.0 / 3600.0) elif offset==0x02b8: # TAS return int(self._getTAS() * 128.0) elif offset==0x02bc: # IAS return int(self.ias * 128.0) elif offset==0x02c8: # VS return int(self.vs * const.FEETTOMETRES * 256.0 / 60.0) elif offset==0x02d4: # ADF2 main return Values._readADFFrequency(self.adf2)[0] elif offset==0x02d6: # ADF2 extended return Values._readADFFrequency(self.adf2)[1] elif offset==0x030c: # TD rate return int(self.tdRate * const.FEETTOMETRES * 256.0 / 60.0) elif offset==0x0330: # Altimeter return int(self.altimeter * 16.0) elif offset==0x034c: # ADF1 main return Values._readADFFrequency(self.adf1)[0] elif offset==0x0350: # NAV1 return Values._readFrequency(self.nav1) elif offset==0x0352: # NAV2 return Values._readFrequency(self.nav2) elif offset==0x0354: # Squawk return Values._readBCD(self.squawk) elif offset==0x0356: # ADF1 extended return Values._readADFFrequency(self.adf1)[1] elif offset==0x0366: # On the ground return 1 if self.onTheGround else 0 elif offset==0x036c: # Stalled return 1 if self.stalled else 0 elif offset==0x036d: # Overspeed return 1 if self.overspeed else 0 elif offset==0x0560: # Latitude return long(self.latitude * 10001750.0 * 65536.0 * 65536.0 / 90.0) elif offset==0x0568: # Longitude return long(self.longitude * 65536.0 * 65536.0 * 65536.0 * 65536.0 / 360.0) elif offset==0x0570: # Altitude return long(self.altitude * const.FEETTOMETRES * 65536.0 * 65536.0) elif offset==0x0578: # Pitch return int(self.pitch * 65536.0 * 65536.0 / 360.0) elif offset==0x057c: # Bank return int(self.bank * 65536.0 * 65536.0 / 360.0) elif offset==0x0580: # Heading return int(self.heading * 65536.0 * 65536.0 / 360.0) elif offset==0x05dc: # Slew return 1 if self.slew else 0 elif offset==0x0628: # Replay return 1 if self.replay else 0 elif offset==0x07bc: # AP Master switch return 1 if self.apMaster else 0 elif offset==0x07c8: # AP heading hold return 1 if self.apHeadingHold else 0 elif offset==0x07cc: # AP heading return int(self.apHeading * 65536.0 / 360.0) elif offset==0x07d0: # AP altitude hold return 1 if self.apAltitudeHold else 0 elif offset==0x07d4: # AP altitude return int(self.apAltitude * const.FEETTOMETRES * 65536.0) elif offset==0x088c: # Engine #1 throttle return self._getThrottle(self.ENGINE_1) elif offset==0x0924: # Engine #2 throttle return self._getThrottle(self.ENGINE_2) elif offset==0x09bc: # Engine #3 throttle return self._getThrottle(self.ENGINE_3) elif offset==0x0af4: # Fuel weight return int(self.fuelWeight * 256.0) elif offset==0x0b74: # Centre tank level return self._getFuelLevel(self.FUEL_CENTRE) elif offset==0x0b78: # Centre tank capacity return self._getFuelCapacity(self.FUEL_CENTRE) elif offset==0x0b7c: # Left tank level return self._getFuelLevel(self.FUEL_LEFT) elif offset==0x0b80: # Left tank capacity return self._getFuelCapacity(self.FUEL_LEFT) elif offset==0x0b84: # Left aux tank level return self._getFuelLevel(self.FUEL_LEFT_AUX) elif offset==0x0b88: # Left aux tank capacity return self._getFuelCapacity(self.FUEL_LEFT_AUX) elif offset==0x0b8c: # Left tip tank level return self._getFuelLevel(self.FUEL_LEFT_TIP) elif offset==0x0b90: # Left tip tank capacity return self._getFuelCapacity(self.FUEL_LEFT_TIP) elif offset==0x0b94: # Right aux tank level return self._getFuelLevel(self.FUEL_RIGHT) elif offset==0x0b98: # Right aux tank capacity return self._getFuelCapacity(self.FUEL_RIGHT) elif offset==0x0b9c: # Right tank level return self._getFuelLevel(self.FUEL_RIGHT_AUX) elif offset==0x0ba0: # Right tank capacity return self._getFuelCapacity(self.FUEL_RIGHT_AUX) elif offset==0x0ba4: # Right tip tank level return self._getFuelLevel(self.FUEL_RIGHT_TIP) elif offset==0x0ba8: # Right tip tank capacity return self._getFuelCapacity(self.FUEL_RIGHT_TIP) elif offset==0x0bc8: # Parking return 1 if self.parking else 0 elif offset==0x0bcc: # Spoilers armed return 1 if self.spoilersArmed else 0 elif offset==0x0bd0: # Spoilers return 0 if self.spoilers == 0 \ else int(self.spoilers * (16383 - 4800) + 4800) elif offset==0x0bdc: # Flaps control numNotchesM1 = len(self.flapsNotches) - 1 flapsIncrement = 16383.0 / numNotchesM1 index = 0 while indexself.flapsNotches[index]: index += 1 if index==numNotchesM1: return 16383 else: return int(index * flapsIncrement + (self.flapsControl-self.flapsNotches[index]) * flapsIncrement / (self.flapsNotches[index+1] - self.flapsNotches[index])) elif offset==0x0be0 or offset==0x0be4: # Flaps left and right return self.flaps * 16383.0 / self.flapsNotches[-1] elif offset==0x0be8: # Gear control return int(self.gearControl * 16383.0) elif offset==0x0bec: # Nose gear return int(self.noseGear * 16383.0) elif offset==0x0c4e: # NAV1 OBS return self.nav1_obs elif offset==0x0c5e: # NAV2 OBS return self.nav2_obs elif offset==0x0d0c: # Lights lights = 0 if self.navLightsOn: lights |= 0x01 if self.antiCollisionLightsOn: lights |= 0x02 if self.landingLightsOn: lights |= 0x04 if self.strobeLightsOn: lights |= 0x10 return lights elif offset==0x0e8a: # Visibility return int(self.visibility * 100.0 / 1609.344) elif offset==0x0e90: # Wind speed return int(self.windSpeed) elif offset==0x0e92: # Wind direction return int(self.windDirection * 65536.0 / 360.0) elif offset==0x11ba: # G-Load return int(self.gLoad * 625.0) elif offset==0x11c6: # Mach # FIXME: calculate from IAS, altitude and QNH return int(self.ias * 0.05 * 20480.) elif offset==0x1244: # Centre 2 tank level return self._getFuelLevel(self.FUEL_CENTRE_2) elif offset==0x1248: # Centre 2 tank capacity return self._getFuelCapacity(self.FUEL_CENTRE_2) elif offset==0x1254: # External 1 tank level return self._getFuelLevel(self.FUEL_EXTERNAL_1) elif offset==0x1258: # External 1 tank capacity return self._getFuelCapacity(self.FUEL_EXTERNAL_1) elif offset==0x125c: # External 2 tank level return self._getFuelLevel(self.FUEL_EXTERNAL_2) elif offset==0x1260: # External 2 tank capacity return self._getFuelCapacity(self.FUEL_EXTERNAL_2) elif offset==0x1274: # Text display mode return 1 if self.textScrolling else 0 elif offset==0x13fc: # The number of the payload stations return self.payloadCount elif offset>=0x1400 and offset<=0x1f40 and \ ((offset-0x1400)%48)==0: # Payload return self.payload[ (offset - 0x1400) / 48 ] elif offset==0x2000: # Engine #1 N1 return self.n1[self.ENGINE_1] elif offset==0x2100: # Engine #2 N1 return self.n1[self.ENGINE_2] elif offset==0x2200: # Engine #3 N1 return self.n1[self.ENGINE_3] elif offset==0x2ef8: # Centre of Gravity return self.cog elif offset==0x30c0: # Gross weight return (self.zfw + sum(self.fuelWeights)) * const.KGSTOLB elif offset==0x31e4: # Radio altitude # FIXME: if self.radioAltitude is None, calculate from the # altitude with some, perhaps random, ground altitude # value radioAltitude = (self.altitude - 517) \ if self.radioAltitude is None else self.radioAltitude return (radioAltitude * const.FEETTOMETRES * 65536.0) elif offset==0x320c: return Values.HOTKEY_SIZE elif offset>=0x3210 and offset<0x3210+Values.HOTKEY_SIZE*4: tableOffset = offset - 0x3210 hotkeyIndex = tableOffset / 4 index = tableOffset % 4 if type=="b" or type=="c": return self.hotkeyTable[hotkeyIndex][index] elif type=="d" or type=="u": if index==0: hotkey = self.hotkeyTable[hotkeyIndex] value = hotkey[3] value <<= 8 value |= hotkey[2] value <<= 8 value |= hotkey[1] value <<= 8 value |= hotkey[0] return value else: print "Unhandled offset: %04x" % (offset,) raise FSUIPCException(ERR_DATA) else: print "Unhandled offset: %04x" % (offset,) raise FSUIPCException(ERR_DATA) elif offset==0x32fa: # Message duration return self.messageDuration elif offset==0x3380: # Message return self.message elif offset==0x3364: # Frozen return 1 if self.frozen else 0 elif offset==0x3bfc: # ZFW return int(self.zfw * 256.0 * const.KGSTOLB) elif offset==0x3c00: # Path of the current AIR file return self.airPath elif offset==0x3d00: # Name of the current aircraft return self.aircraftName elif offset==0x6202: # PMDG 737NG switches return self.pmdg_737ng_switches elif offset==0x6500: # PMDG 737NGX lights position SW return self.pmdg_737ngx_lts_positionsw elif offset==0x7b91: # Transponder standby return 0 if self.xpdrC else 1 else: print "Unhandled offset: %04x" % (offset,) raise FSUIPCException(ERR_DATA) def write(self, offset, value, type): """Write the value at the given offset.""" try: return self._write(offset, value, type) except Exception, e: print "failed to write offset %04x: %s" % (offset, str(e)) raise FSUIPCException(ERR_DATA) def _write(self, offset, value, type): """Write the given value at the given offset.""" if offset==0x023a: # Second of time self._updateTimeOffset(5, value) elif offset==0x023b: # Hour of Zulu time self._updateTimeOffset(3, value) elif offset==0x023c: # Minute of Zulu time self._updateTimeOffset(4, value) elif offset==0x023e: # Day number in the year self._updateTimeOffset(7, value) elif offset==0x0240: # Year in FS self._updateTimeOffset(0, value) elif offset==0x0264: # Paused self.paused = value!=0 elif offset==0x029c: # Pitot self.pitot = value!=0 elif offset==0x02b4: # Ground speed # FIXME: calculate TAS using the heading and the wind, and # then IAS based on the altitude self.ias = value * 3600.0 / 65536.0 / 1852.0 elif offset==0x02bc: # IAS self.ias = value / 128.0 elif offset==0x02c8: # VS self.vs = value * 60.0 / const.FEETTOMETRES / 256.0 if not self.onTheGround: self.tdRate = self.vs elif offset==0x02d4: # ADF2 main self.adf2 = self._writeADFFrequency(self.adf2, value, True) elif offset==0x02d6: # ADF2 main self.adf2 = self._writeADFFrequency(self.adf2, value, False) elif offset==0x0330: # Altimeter self.altimeter = value / 16.0 elif offset==0x034c: # ADF1 main self.adf1 = self._writeADFFrequency(self.adf1, value, True) elif offset==0x0350: # NAV1 self.nav1 = Values._writeFrequency(value) elif offset==0x0352: # NAV2 self.nav2 = Values._writeFrequency(value) elif offset==0x0354: # Squawk self.squawk = Values._writeBCD(value) elif offset==0x0356: # ADF1 ext self.adf1 = self._writeADFFrequency(self.adf1, value, False) elif offset==0x0366: # On the groud self.onTheGround = value!=0 if not self.onTheGround: self.tdRate = self.vs elif offset==0x036c: # Stalled self.stalled = value!=0 elif offset==0x036d: # Overspeed self.overspeed = value!=0 elif offset==0x0560: # Latitude self.latitude = value * 90.0 / 10001750.0 / 65536.0 / 65536.0 elif offset==0x0568: # Longitude self.longitude = value * 360.0 / 65536.0 / 65536.0 / 65536.0 / 65536.0 elif offset==0x0570: # Altitude self.altitude = value / const.FEETTOMETRES / 65536.0 / 65536.0 self.radioAltitude = self.altitude - 517 elif offset==0x0578: # Pitch self.pitch = value * 360.0 / 65536.0 / 65536.0 elif offset==0x057c: # Bank self.bank = value * 360.0 / 65536.0 / 65536.0 elif offset==0x0580: # Heading self.heading = value * 360.0 / 65536.0 / 65536.0 elif offset==0x05dc: # Slew self.slew = value!=0 elif offset==0x0628: # Replay self.replay = value!=0 elif offset==0x07bc: # AP Master switch self.apMaster = value!=0 elif offset==0x07c8: # AP heading hold self.apHeadingHold = value!=0 elif offset==0x07cc: # AP heading self.apHeading = value * 360.0 / 65536.0 elif offset==0x07d0: # AP altitude hold self.apAltitudeHold = value!=0 elif offset==0x07d4: # AP altitude self.apAltitude = value / const.FEETTOMETRES / 65536.0 elif offset==0x088c: # Engine #1 throttle self._setThrottle(self.ENGINE_1, value) elif offset==0x0924: # Engine #2 throttle self._setThrottle(self.ENGINE_2, value) elif offset==0x09bc: # Engine #3 throttle self._setThrottle(self.ENGINE_3, value) elif offset==0x0af4: # Fuel weight self.fuelWeight = value / 256.0 elif offset==0x0b74: # Centre tank level self._setFuelLevel(self.FUEL_CENTRE, value) elif offset==0x0b78: # Centre tank capacity self._setFuelCapacity(self.FUEL_CENTRE, value) elif offset==0x0b7c: # Left tank level self._setFuelLevel(self.FUEL_LEFT, value) elif offset==0x0b80: # Left tank capacity self._setFuelCapacity(self.FUEL_LEFT, value) elif offset==0x0b84: # Left aux tank level self._setFuelLevel(self.FUEL_LEFT_AUX, value) elif offset==0x0b88: # Left aux tank capacity self._setFuelCapacity(self.FUEL_LEFT_AUX, value) elif offset==0x0b8c: # Left tip tank level self._setFuelLevel(self.FUEL_LEFT_TIP, value) elif offset==0x0b90: # Left tip tank capacity self._setFuelCapacity(self.FUEL_LEFT_TIP, value) elif offset==0x0b94: # Right aux tank level self._setFuelLevel(self.FUEL_RIGHT, value) elif offset==0x0b98: # Right aux tank capacity self._setFuelCapacity(self.FUEL_RIGHT, value) elif offset==0x0b9c: # Right tank level self._setFuelLevel(self.FUEL_RIGHT_AUX, value) elif offset==0x0ba0: # Right tank capacity self._setFuelCapacity(self.FUEL_RIGHT_AUX, value) elif offset==0x0ba4: # Right tip tank level self._setFuelLevel(self.FUEL_RIGHT_TIP, value) elif offset==0x0ba8: # Right tip tank capacity self._setFuelCapacity(self.FUEL_RIGHT_TIP, value) elif offset==0x0bc8: # Parking self.parking = value!=0 elif offset==0x0bcc: # Spoilers armed self.spoilersArmed = value!=0 elif offset==0x0bd0: # Spoilers self.spoilters = 0 if value==0 \ else (value - 4800) / (16383 - 4800) elif offset==0x0bdc: # Flaps control numNotchesM1 = len(self.flapsNotches) - 1 flapsIncrement = 16383.0 / numNotchesM1 index = int(value / flapsIncrement) if index>=numNotchesM1: self.flapsControl = self.flapsNotches[-1] else: self.flapsControl = self.flapsNotches[index] self.flapsControl += (value - index * flapsIncrement) * \ (self.flapsNotches[index+1] - self.flapsNotches[index]) / \ flapsIncrement elif offset==0x0be0 or offset==0x0be4: # Flaps left and right self.flaps = value * self.flapsNotches[-1] / 16383.0 elif offset==0x0be8: # Gear control self.gearControl = value / 16383.0 elif offset==0x0bec: # Nose gear self.noseGear = value / 16383.0 elif offset==0x0c4e: # NAV1 OBS self.nav1_obs = value elif offset==0x0c5e: # NAV2 OBS self.nav2_obs = value elif offset==0x0d0c: # Lights self.navLightsOn = (value&0x01)!=0 self.antiCollisionLightsOn = (value&0x02)!=0 self.landingLightsOn = (value&0x04)!=0 self.strobeLightsOn = (value&0x10)!=0 elif offset==0x0e8a: # Visibility self.visibility = value * 1609.344 / 100.0 elif offset==0x0e90: # Wind speed self.windSpeed = value elif offset==0x0e92: # Wind direction self.windDirection = value * 360.0 / 65536.0 elif offset==0x11ba: # G-Load self.gLoad = value / 625.0 elif offset==0x11c6: # Mach # FIXME: calculate IAS using the altitude and QNH self.ias = value / 0.05 / 20480 elif offset==0x1244: # Centre 2 tank level self._setFuelLevel(self.FUEL_CENTRE_2, value) elif offset==0x1248: # Centre 2 tank capacity self._setFuelCapacity(self.FUEL_CENTRE_2, value) elif offset==0x1254: # External 1 tank level self._setFuelLevel(self.FUEL_EXTERNAL_1, value) elif offset==0x1258: # External 1 tank capacity self._setFuelCapacity(self.FUEL_EXTERNAL_1, value) elif offset==0x125c: # External 2 tank level self._setFuelLevel(self.FUEL_EXTERNAL_2, value) elif offset==0x1260: # External 2 tank capacity self._setFuelCapacity(self.FUEL_EXTERNAL_2, value) elif offset==0x1274: # Text display mode textScrolling = value!=0 elif offset==0x13fc: # The number of the payload stations self.payloadCount = int(value) elif offset>=0x1400 and offset<=0x1f40 and \ ((offset-0x1400)%48)==0: # Payload self.payload[ (offset - 0x1400) / 48 ] = value elif offset==0x2000: # Engine #1 N1 self.n1[self.ENGINE_1] = value elif offset==0x2100: # Engine #2 N1 self.n1[self.ENGINE_2] = value elif offset==0x2200: # Engine #3 N1 self.n1[self.ENGINE_3] = value elif offset==0x2ef8: # Centre of Gravity self.cog = value elif offset==0x30c0: # Gross weight raise FSUIPCException(ERR_DATA) elif offset==0x31e4: # Radio altitude self.radioAltitude = value / const.FEETTOMETRES / 65536.0 self.altitude = self.radioAltitude + 517 elif offset==0x31e4: # Radio altitude raise FSUIPCException(ERR_DATA) elif offset==0x320c: return Values.HOTKEY_SIZE elif offset>=0x3210 and offset<0x3210+Values.HOTKEY_SIZE*4: tableOffset = offset - 0x3210 hotkeyIndex = tableOffset / 4 index = tableOffset % 4 if type=="b" or type=="c": self.hotkeyTable[hotkeyIndex][index] = value elif type=="d" or type=="u": if index==0: hotkey = self.hotkeyTable[hotkeyIndex] hotkey[0] = value & 0xff hotkey[1] = (value>>8) & 0xff hotkey[2] = (value>>16) & 0xff hotkey[3] = (value>>24) & 0xff else: print "Unhandled offset: %04x for type '%s'" % (offset, type) raise FSUIPCException(ERR_DATA) else: print "Unhandled offset: %04x for type '%s'" % (offset, type) raise FSUIPCException(ERR_DATA) elif offset==0x32fa: # Message duration self.messageDuration = value elif offset==0x3380: # Message self.message = value elif offset==0x3364: # Frozen self.frozen = value!=0 elif offset==0x3bfc: # ZFW self.zfw = value * const.LBSTOKG / 256.0 elif offset==0x3c00: # Path of the current AIR file self.airPath = value elif offset==0x3d00: # Name of the current aircraft self.aircraftName = value elif offset==0x6202: # PMDG 737NG switches self.pmdg_737ng_switches = value elif offset==0x6500: # PMDG 737NGX lights position SW self.pmdg_737ngx_lts_positionsw = value elif offset==0x7b91: # Transponder standby self.xpdrC = value==0 else: print "Unhandled offset: %04x" % (offset,) raise FSUIPCException(ERR_DATA) def _readUTC(self): """Read the UTC time. The current offset is added to it.""" return time.gmtime(time.time() + self._timeOffset) def _getFuelLevel(self, index): """Get the fuel level for the fuel tank with the given index.""" return 0 if self.fuelCapacities[index]==0.0 else \ int(self.fuelWeights[index] * 65536.0 * 128.0 / self.fuelCapacities[index]) def _getFuelCapacity(self, index): """Get the capacity of the fuel tank with the given index.""" return int(self.fuelCapacities[index] * const.KGSTOLB / self.fuelWeight) def _getThrottle(self, index): """Get the throttle value for the given index.""" return int(self.throttles[index] * 16383.0) def _updateTimeOffset(self, index, value): """Update the time offset if the value in the tm structure is replaced by the given value.""" tm = self._readUTC() tm1 = tm[:index] + (value,) + tm[(index+1):] self._timeOffset += calendar.timegm(tm1) - calendar.timegm(tm) def _setThrottle(self, index, value): """Set the throttle value for the given index.""" self.throttles[index] = value / 16383.0 def _setFuelLevel(self, index, value): """Set the fuel level for the fuel tank with the given index.""" self.fuelWeights[index] = self.fuelCapacities[index] * float(value) / \ 65536.0 / 128.0 def _setFuelCapacity(self, index, value): """Set the capacity of the fuel tank with the given index.""" self.fuelCapacities[index] = value * self.fuelWeight * const.LBSTOKG def _getTAS(self): """Calculate the true airspeed.""" pressure = 101325 * math.pow(1 - 2.25577e-5 * self.altitude * const.FEETTOMETRES, 5.25588) temperature = 15 - self.altitude * 6.5 * const.FEETTOMETRES / 1000.0 temperature += 273.15 # Celsius -> Kelvin airDensity = pressure / (temperature * 287.05) #print "pressure:", pressure, "temperature:", temperature, "airDensity:", airDensity return self.ias * math.sqrt(1.225 / airDensity) #------------------------------------------------------------------------------ values = Values() #------------------------------------------------------------------------------ failOpen = False opened = False #------------------------------------------------------------------------------ def open(request): """Open the connection.""" global opened if failOpen: raise FSUIPCException(ERR_NOFS) elif opened: raise FSUIPCException(ERR_OPEN) else: time.sleep(0.5) opened = True return True #------------------------------------------------------------------------------ def prepare_data(pattern, forRead = True, checkOpened = True): """Prepare the given pattern for reading and/or writing.""" if not checkOpened or opened: return pattern else: raise FSUIPCException(ERR_NOTOPEN) #------------------------------------------------------------------------------ def read(data, checkOpened = True): """Read the given data.""" if not checkOpened or opened: return [values.read(offset, type) for (offset, type) in data] else: raise FSUIPCException(ERR_NOTOPEN) #------------------------------------------------------------------------------ def write(data, checkOpened = True): """Write the given data.""" if not checkOpened or opened: for (offset, type, value) in data: values.write(offset, value, type) else: raise FSUIPCException(ERR_NOTOPEN) #------------------------------------------------------------------------------ def close(): """Close the connection.""" global opened opened = False #------------------------------------------------------------------------------ PORT=15015 CALL_READ=1 CALL_WRITE=2 CALL_SETVERSION=3 CALL_CLOSE=4 CALL_FAILOPEN=5 CALL_QUIT = 99 RESULT_RETURNED=1 RESULT_EXCEPTION=2 #------------------------------------------------------------------------------ class Server(threading.Thread): """The server thread.""" def __init__(self): """Construct the thread.""" super(Server, self).__init__() self.daemon = True def run(self): """Perform the server's operation.""" serverSocket = socket.socket() serverSocket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) serverSocket.bind(("", PORT)) serverSocket.listen(5) while True: (clientSocket, clientAddress) = serverSocket.accept() thread = threading.Thread(target = self._process, args=(clientSocket,)) thread.start() def _process(self, clientSocket): """Process the commands arriving on the given socket.""" socketFile = clientSocket.makefile() try: while True: (length,) = struct.unpack("I", clientSocket.recv(4)) data = clientSocket.recv(length) (call, args) = cPickle.loads(data) exception = None try: if call==CALL_READ: result = read(args[0], checkOpened = False) elif call==CALL_WRITE: result = write(args[0], checkOpened = False) elif call==CALL_SETVERSION: global fs_version fs_version = args[0] result = None elif call==CALL_CLOSE: global opened opened = False result = None elif call==CALL_FAILOPEN: global failOpen failOpen = args[0] result = None else: break except Exception, e: exception = e if exception is None: data = cPickle.dumps((RESULT_RETURNED, result)) else: data = cPickle.dumps((RESULT_EXCEPTION, str(exception))) clientSocket.send(struct.pack("I", len(data)) + data) except Exception, e: print >> sys.stderr, "pyuipc_sim.Server._process: failed with exception:", str(e) finally: try: socketFile.close() except: pass clientSocket.close() #------------------------------------------------------------------------------ class Client(object): """Client to the server.""" def __init__(self, serverHost): """Construct the client and connect to the given server host.""" self._socket = socket.socket() self._socket.connect((serverHost, PORT)) self._socketFile = self._socket.makefile() def read(self, data): """Read the given data.""" return self._call(CALL_READ, data) def write(self, data): """Write the given data.""" return self._call(CALL_WRITE, data) def setVersion(self, version): """Set the FS version to emulate.""" return self._call(CALL_SETVERSION, int(version)) def close(self): """Close the connection currently opened in the simulator.""" return self._call(CALL_CLOSE, None) def failOpen(self, really): """Enable/disable open failure in the simulator.""" return self._call(CALL_FAILOPEN, really) def quit(self): """Quit from the simulator.""" data = cPickle.dumps((CALL_QUIT, None)) self._socket.send(struct.pack("I", len(data)) + data) def _call(self, command, data): """Perform a call with the given command and data.""" data = cPickle.dumps((command, [data])) self._socket.send(struct.pack("I", len(data)) + data) (length,) = struct.unpack("I", self._socket.recv(4)) data = self._socket.recv(length) (resultCode, result) = cPickle.loads(data) if resultCode==RESULT_RETURNED: return result else: raise Exception(result) #------------------------------------------------------------------------------ #------------------------------------------------------------------------------ # FIXME: implement proper completion and history class CLI(cmd.Cmd): """The command-line interpreter.""" @staticmethod def str2bool(s): """Convert the given string to a PyUIPC boolean value (i.e. 0 or 1).""" return 1 if s in ["yes", "true", "on"] else 0 @staticmethod def bool2str(value): """Convert the PyUIPC boolean value (i.e. 0 or 1) into a string.""" return "no" if value==0 else "yes" @staticmethod def degree2pyuipc(degree): """Convert the given degree (as a string) into a PyUIPC value.""" return int(float(degree) * 65536.0 * 65536.0 / 360.0) @staticmethod def pyuipc2degree(value): """Convert the given PyUIPC value into a degree.""" return value * 360.0 / 65536.0 / 65536.0 @staticmethod def fuelLevel2pyuipc(level): """Convert the given percentage value (as a string) into a PyUIPC value.""" return int(float(level) * 128.0 * 65536.0 / 100.0) @staticmethod def pyuipc2fuelLevel(value): """Convert the PyUIPC value into a percentage value.""" return value * 100.0 / 128.0 / 65536.0 @staticmethod def fuelCapacity2pyuipc(capacity): """Convert the given capacity value (as a string) into a PyUIPC value.""" return int(capacity) @staticmethod def pyuipc2fuelCapacity(value): """Convert the given capacity value into a PyUIPC value.""" return value @staticmethod def throttle2pyuipc(throttle): """Convert the given throttle value (as a string) into a PyUIPC value.""" return int(float(throttle) * 16384.0 / 100.0) @staticmethod def pyuipc2throttle(value): """Convert the given PyUIPC value into a throttle value.""" return value * 100.0 / 16384.0 @staticmethod def heading2pyuipc(heading): """Convert the given heading (as a string) into a PyUIPC value.""" return int(float(heading) * 65536.0 / 360.0) @staticmethod def pyuipc2heading(value): """Convert the given PyUIPC value into a heading.""" return value * 360.0 / 65536.0 @staticmethod def altitude2pyuipc(altitude): """Convert the given altitude (as a string) into a PyUIPC value.""" return int(float(altitude) * const.FEETTOMETRES * 65536.0) @staticmethod def pyuipc2altitude(value): """Convert the given PyUIPC value into an altitude.""" return value / const.FEETTOMETRES / 65536.0 def __init__(self): """Construct the CLI.""" cmd.Cmd.__init__(self) self.use_rawinput = True self.intro = "\nPyUIPC simulator command prompt\n" self.prompt = "PyUIPC> " self.daemon = True host = sys.argv[1] if len(sys.argv)>1 else "localhost" self._client = Client(host) self._valueHandlers = {} self._valueHandlers["year"] = ([(0x0240, "H")], lambda value: value, lambda word: int(word)) self._valueHandlers["yday"] = ([(0x023e, "H")], lambda value: value, lambda word: int(word)) self._valueHandlers["hour"] = ([(0x023b, "b")], lambda value: value, lambda word: int(word)) self._valueHandlers["min"] = ([(0x023c, "b")], lambda value: value, lambda word: int(word)) self._valueHandlers["sec"] = ([(0x023a, "b")], lambda value: value, lambda word: int(word)) self._valueHandlers["acftName"] = ([(0x3d00, -256)], lambda value: value, lambda word: word) self._valueHandlers["airPath"] = ([(0x3c00, -256)], lambda value: value, lambda word: word) self._valueHandlers["latitude"] = ([(0x0560, "l")], lambda value: value * 90.0 / 10001750.0 / 65536.0 / 65536.0, lambda word: long(float(word) * 10001750.0 * 65536.0 * 65536.0 / 90.0)) self._valueHandlers["longitude"] = ([(0x0568, "l")], lambda value: value * 360.0 / 65536.0 / 65536.0 / 65536.0 / 65536.0, lambda word: long(float(word) * 65536.0 * 65536.0 * 65536.0 * 65536.0 / 360.0)) self._valueHandlers["paused"] = ([(0x0264, "H")], CLI.bool2str, CLI.str2bool) self._valueHandlers["frozen"] = ([(0x3364, "H")], CLI.bool2str, CLI.str2bool) self._valueHandlers["replay"] = ([(0x0628, "d")], CLI.bool2str, CLI.str2bool) self._valueHandlers["slew"] = ([(0x05dc, "H")], CLI.bool2str, CLI.str2bool) self._valueHandlers["overspeed"] = ([(0x036d, "b")], CLI.bool2str, CLI.str2bool) self._valueHandlers["stalled"] = ([(0x036c, "b")], CLI.bool2str, CLI.str2bool) self._valueHandlers["onTheGround"] = ([(0x0366, "H")], CLI.bool2str, CLI.str2bool) self._valueHandlers["zfw"] = ([(0x3bfc, "d")], lambda value: value * const.LBSTOKG / 256.0, lambda word: int(float(word) * 256.0 * const.KGSTOLB)) self._valueHandlers["grossWeight"] = ([(0x30c0, "f")], lambda value: value * const.LBSTOKG, lambda word: None) self._valueHandlers["heading"] = ([(0x0580, "d")], CLI.pyuipc2degree, CLI.degree2pyuipc) self._valueHandlers["pitch"] = ([(0x0578, "d")], CLI.pyuipc2degree, CLI.degree2pyuipc) self._valueHandlers["bank"] = ([(0x057c, "d")], CLI.pyuipc2degree, CLI.degree2pyuipc) self._valueHandlers["ias"] = ([(0x02bc, "d")], lambda value: value / 128.0, lambda word: int(float(word) * 128.0)) self._valueHandlers["mach"] = ([(0x11c6, "H")], lambda value: value / 20480.0, lambda word: int(float(word) * 20480.0)) self._valueHandlers["gs"] = ([(0x02b4, "d")], lambda value: value * 3600.0 / 65536.0 / 1852.0, lambda word: int(float(word) * 65536.0 * 1852.0 / 3600)) self._valueHandlers["tas"] = ([(0x02b8, "d")], lambda value: value / 128.0, lambda word: None) self._valueHandlers["vs"] = ([(0x02c8, "d")], lambda value: value * 60 / const.FEETTOMETRES / 256.0, lambda word: int(float(word) * const.FEETTOMETRES * 256.0 / 60.0)) self._valueHandlers["tdRate"] = ([(0x030c, "d")], lambda value: value * 60 / const.FEETTOMETRES / 256.0, lambda word: int(float(word) * const.FEETTOMETRES * 256.0 / 60.0)) self._valueHandlers["radioAltitude"] = ([(0x31e4, "d")], lambda value: value / const.FEETTOMETRES / 65536.0, lambda word: int(float(word) * const.FEETTOMETRES * 65536.0)) self._valueHandlers["altitude"] = ([(0x0570, "l")], lambda value: value / const.FEETTOMETRES / 65536.0 / 65536.0, lambda word: long(float(word) * const.FEETTOMETRES * 65536.0 * 65536.0)) self._valueHandlers["gLoad"] = ([(0x11ba, "H")], lambda value: value / 625.0, lambda word: int(float(word) * 625.0)) self._valueHandlers["flapsControl"] = ([(0x0bdc, "d")], lambda value: value * 100.0 / 16383.0, lambda word: int(float(word) * 16383.0 / 100.0)) self._valueHandlers["flaps"] = ([(0x0be0, "d")], lambda value: value * 100.0 / 16383.0, lambda word: int(float(word) * 16383.0 / 100.0)) self._valueHandlers["lights"] = ([(0x0d0c, "H")], lambda value: value, lambda word: int(word)) self._valueHandlers["pitot"] = ([(0x029c, "b")], CLI.bool2str, CLI.str2bool) self._valueHandlers["parking"] = ([(0x0bc8, "H")], CLI.bool2str, CLI.str2bool) self._valueHandlers["gearControl"] = ([(0x0be8, "d")], lambda value: value * 100.0 / 16383.0, lambda word: int(float(word) * 16383.0 / 100.0)) self._valueHandlers["noseGear"] = ([(0x0bec, "d")], lambda value: value * 100.0 / 16383.0, lambda word: int(float(word) * 16383.0 / 100.0)) self._valueHandlers["spoilersArmed"] = ([(0x0bcc, "d")], CLI.bool2str, CLI.str2bool) self._valueHandlers["spoilers"] = ([(0x0bd0, "d")], lambda value: value, lambda word: int(word)) self._valueHandlers["qnh"] = ([(0x0330, "H")], lambda value: value / 16.0, lambda word: int(float(word)*16.0)) self._valueHandlers["nav1"] = ([(0x0350, "H")], Values._writeFrequency, lambda word: Values._readFrequency(float(word))) self._valueHandlers["nav1_obs"] = ([(0x0c4e, "H")], lambda value: value, lambda word: int(word)) self._valueHandlers["nav2"] = ([(0x0352, "H")], Values._writeFrequency, lambda word: Values._readFrequency(float(word))) self._valueHandlers["nav2_obs"] = ([(0x0c5e, "H")], lambda value: value, lambda word: int(word)) self._valueHandlers["adf1"] = ([(0x034c, "H"), (0x0356, "H")], lambda values: Values._toADFFrequency(values[0], values[1]), lambda word: Values._readADFFrequency(float(word))) self._valueHandlers["adf2"] = ([(0x02d4, "H"), (0x02d6, "H")], lambda values: Values._toADFFrequency(values[0], values[1]), lambda word: Values._readADFFrequency(float(word))) self._valueHandlers["squawk"] = ([(0x0354, "H")], Values._writeBCD, lambda word: Values._readBCD(int(word))) self._valueHandlers["windSpeed"] = ([(0x0e90, "H")], lambda value: value, lambda word: int(word)) self._valueHandlers["windDirection"] = ([(0x0e92, "H")], lambda value: value * 360.0 / 65536.0, lambda word: int(int(word) * 65536.0 / 360.0)) self._valueHandlers["fuelWeight"] = ([(0x0af4, "H")], lambda value: value / 256.0, lambda word: int(float(word)*256.0)) self._valueHandlers["centreLevel"] = ([(0x0b74, "d")], CLI.pyuipc2fuelLevel, CLI.fuelLevel2pyuipc) self._valueHandlers["centreCapacity"] = ([(0x0b78, "d")], CLI.pyuipc2fuelCapacity, CLI.fuelCapacity2pyuipc) self._valueHandlers["leftMainLevel"] = ([(0x0b7c, "d")], CLI.pyuipc2fuelLevel, CLI.fuelLevel2pyuipc) self._valueHandlers["leftMainCapacity"] = ([(0x0b80, "d")], CLI.pyuipc2fuelCapacity, CLI.fuelCapacity2pyuipc) self._valueHandlers["leftAuxLevel"] = ([(0x0b84, "d")], CLI.pyuipc2fuelLevel, CLI.fuelLevel2pyuipc) self._valueHandlers["leftAuxCapacity"] = ([(0x0b88, "d")], CLI.pyuipc2fuelCapacity, CLI.fuelCapacity2pyuipc) self._valueHandlers["leftTipLevel"] = ([(0x0b8c, "d")], CLI.pyuipc2fuelLevel, CLI.fuelLevel2pyuipc) self._valueHandlers["leftTipCapacity"] = ([(0x0b90, "d")], CLI.pyuipc2fuelCapacity, CLI.fuelCapacity2pyuipc) self._valueHandlers["rightMainLevel"] = ([(0x0b94, "d")], CLI.pyuipc2fuelLevel, CLI.fuelLevel2pyuipc) self._valueHandlers["rightMainCapacity"] = ([(0x0b98, "d")], CLI.pyuipc2fuelCapacity, CLI.fuelCapacity2pyuipc) self._valueHandlers["rightAuxLevel"] = ([(0x0b9c, "d")], CLI.pyuipc2fuelLevel, CLI.fuelLevel2pyuipc) self._valueHandlers["rightAuxCapacity"] = ([(0x0ba0, "d")], CLI.pyuipc2fuelCapacity, CLI.fuelCapacity2pyuipc) self._valueHandlers["rightTipLevel"] = ([(0x0ba4, "d")], CLI.pyuipc2fuelLevel, CLI.fuelLevel2pyuipc) self._valueHandlers["rightTipCapacity"] = ([(0x0ba8, "d")], CLI.pyuipc2fuelCapacity, CLI.fuelCapacity2pyuipc) self._valueHandlers["centre2Level"] = ([(0x1244, "d")], CLI.pyuipc2fuelLevel, CLI.fuelLevel2pyuipc) self._valueHandlers["centre2Capacity"] = ([(0x1248, "d")], CLI.pyuipc2fuelCapacity, CLI.fuelCapacity2pyuipc) self._valueHandlers["external1Level"] = ([(0x1254, "d")], CLI.pyuipc2fuelLevel, CLI.fuelLevel2pyuipc) self._valueHandlers["external1Capacity"] = ([(0x1258, "d")], CLI.pyuipc2fuelCapacity, CLI.fuelCapacity2pyuipc) self._valueHandlers["external2Level"] = ([(0x125c, "d")], CLI.pyuipc2fuelLevel, CLI.fuelLevel2pyuipc) self._valueHandlers["external2Capacity"] = ([(0x1260, "d")], CLI.pyuipc2fuelCapacity, CLI.fuelCapacity2pyuipc) self._valueHandlers["n1_1"] = ([(0x2000, "f")], lambda value: value, lambda word: float(word)) self._valueHandlers["n1_2"] = ([(0x2100, "f")], lambda value: value, lambda word: float(word)) self._valueHandlers["n1_3"] = ([(0x2200, "f")], lambda value: value, lambda word: float(word)) self._valueHandlers["throttle_1"] = ([(0x088c, "H")], CLI.pyuipc2throttle, CLI.throttle2pyuipc) self._valueHandlers["throttle_2"] = ([(0x0924, "H")], CLI.pyuipc2throttle, CLI.throttle2pyuipc) self._valueHandlers["throttle_3"] = ([(0x09bc, "H")], CLI.pyuipc2throttle, CLI.throttle2pyuipc) self._valueHandlers["visibility"] = ([(0x0e8a, "H")], lambda value: value*1609.344/100.0, lambda word: int(float(word)* 100.0/1609.344)) self._valueHandlers["payloadCount"] = ([(0x13fc, "d")], lambda value: value, lambda word: int(word)) for i in range(0, 61): self._valueHandlers["payload%d" % (i,)] = ([(0x1400 + i * 48, "f")], lambda value: value * const.LBSTOKG, lambda word: float(word)*const.KGSTOLB) self._valueHandlers["textScrolling"] = ([(0x1274, "h")], CLI.bool2str, CLI.str2bool) self._valueHandlers["messageDuration"] = ([(0x32fa, "h")], lambda value: value, lambda word: int(word)) self._valueHandlers["message"] = ([(0x3380, -128)], lambda value: value, lambda word: word) for i in range(0, Values.HOTKEY_SIZE): self._valueHandlers["hotkey%d" % (i,)] = ([(0x3210 + i*4, "u")], lambda value: "0x%08x" % (value,), lambda word: long(word, 16)) self._valueHandlers["cog"] = ([(0x2ef8, "f")], lambda value: value, lambda word: float(word)) self._valueHandlers["pmdg_737ng_switches"] = ([(0x6202, "b")], lambda value: value, lambda word: int(word)) self._valueHandlers["pmdg_737ngx_lts_positionsw"] = ([(0x6500, "b")], lambda value: value, lambda word: int(word)) self._valueHandlers["xpdrC"] = ([(0x7b91, "b")], lambda value: value, lambda word: int(word)) self._valueHandlers["apMaster"] = ([(0x07bc, "u")], CLI.bool2str, CLI.str2bool) self._valueHandlers["apHeadingHold"] = ([(0x07c8, "u")], CLI.bool2str, CLI.str2bool) self._valueHandlers["apHeading"] = ([(0x07cc, "H")], CLI.pyuipc2heading, CLI.heading2pyuipc) self._valueHandlers["apAltitudeHold"] = ([(0x07d0, "u")], CLI.bool2str, CLI.str2bool) self._valueHandlers["apAltitude"] = ([(0x07d4, "H")], CLI.pyuipc2altitude, CLI.altitude2pyuipc) def default(self, line): """Handle unhandle commands.""" if line=="EOF": print return self.do_quit("") else: return super(CLI, self).default(line) def do_get(self, args): """Handle the get command.""" names = args.split() data = [] for name in names: if name not in self._valueHandlers: print >> sys.stderr, "Unknown variable: " + name return False valueHandler = self._valueHandlers[name] data += valueHandler[0] try: results = self._client.read(data) index = 0 i = 0 while index [...]" def complete_get(self, text, line, begidx, endidx): """Try to complete the get command.""" return [key for key in self._valueHandlers if key.startswith(text)] def do_set(self, args): """Handle the set command.""" arguments = [] inWord = False inQuote = False word = "" for c in args: if c.isspace() and not inQuote: if inWord: arguments.append(word) word = "" inWord = False elif c=='"': inQuote = not inQuote else: inWord = True word += c if inWord: arguments.append(word) names = [] data = [] for argument in arguments: words = argument.split("=") if len(words)!=2: print >> sys.stderr, "Invalid argument: " + argument return False (name, value) = words if name not in self._valueHandlers: print >> sys.stderr, "Unknown variable: " + name return False valueHandler = self._valueHandlers[name] try: values = valueHandler[2](value) if len(valueHandler[0])==1: values = [values] index = 0 for (offset, type) in valueHandler[0]: data.append((offset, type, values[index])) index += 1 except Exception, e: print >> sys.stderr, "Invalid value '%s' for variable %s: %s" % \ (value, name, str(e)) return False try: self._client.write(data) print "Data written" except Exception, e: print >> sys.stderr, "Failed to write data: " + str(e) return False def help_set(self): """Print help for the set command.""" print "set = [=...]" def complete_set(self, text, line, begidx, endidx): """Try to complete the set command.""" if not text and begidx>0 and line[begidx-1]=="=": return [] else: return [key + "=" for key in self._valueHandlers if key.startswith(text)] def do_setversion(self, args): """Set the version number to simulate""" try: value = int(args) self._client.setVersion(value) print "Emulating version %d" % (value,) except Exception, e: print >> sys.stderr, "Failed to set the version: " + str(e) def help_setversion(self, usage = False): """Help for the setversion command""" if usage: print "Usage:", print "setversion " def do_close(self, args): """Close an existing connection so that FS will fail.""" try: self._client.close() print "Connection closed" except Exception, e: print >> sys.stderr, "Failed to close the connection: " + str(e) def do_failopen(self, args): """Enable/disable the failing of opens.""" try: value = self.str2bool(args) self._client.failOpen(value) print "Opening will%s fail" % ("" if value else " not",) except Exception, e: print >> sys.stderr, "Failed to set open failure: " + str(e) def help_failopen(self, usage = False): """Help for the failopen command""" if usage: print "Usage:", print "failopen yes|no" def complete_failopen(self, text, line, begidx, endidx): if text: if "yes".startswith(text): return ["yes"] elif "no".startswith(text): return ["no"] else: return [] else: return ["yes", "no"] def do_quit(self, args): """Handle the quit command.""" self._client.quit() return True #------------------------------------------------------------------------------ if __name__ == "__main__": CLI().cmdloop() else: server = Server() server.start() #------------------------------------------------------------------------------