aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorSylvain Munaut <tnt@246tNt.com>2010-12-26 01:01:21 +0100
committerSylvain Munaut <tnt@246tNt.com>2010-12-26 01:01:21 +0100
commitbe6f44dc14a02ccc040474a7678ea3076e9957dd (patch)
tree2f909a3cf4774367ad2c4c129521f37b370561b5
parent4876712f90dec21af1df79ada0c375332fc5ca66 (diff)
Import of the programming tool
Signed-off-by: Sylvain Munaut <tnt@246tNt.com>
-rwxr-xr-xccc-prog.py213
1 files changed, 213 insertions, 0 deletions
diff --git a/ccc-prog.py b/ccc-prog.py
new file mode 100755
index 0000000..25254cf
--- /dev/null
+++ b/ccc-prog.py
@@ -0,0 +1,213 @@
+#!/usr/bin/env python
+
+#
+# Utility to write the cards
+#
+#
+# Copyright (C) 2009 Sylvain Munaut <tnt@246tNt.com>
+# Copyright (C) 2010 Harald Welte <laforge@gnumonks.org>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+#
+
+from optparse import OptionParser
+
+from pySim.commands import SimCardCommands
+from pySim.cards import _cards_classes
+
+from ccc import StateManager, CardParameters
+
+
+
+def csv_load_cards(filename):
+ import csv
+ fh = open(filename, 'r')
+ cr = csv.reader(fh)
+ cards = dict([(int(x[0]), CardParameters(int(x[0]), x[1], x[2], x[3])) for x in cr])
+ fh.close()
+ return cards
+
+
+def card_detect(opts, scc):
+
+ # Detect type if needed
+ card = None
+ ctypes = dict([(kls.name, kls) for kls in _cards_classes])
+
+ if opts.type in ("auto", "auto_once"):
+ for kls in _cards_classes:
+ card = kls.autodetect(scc)
+ if card:
+ print "Autodetected card type %s" % card.name
+ card.reset()
+ break
+
+ if card is None:
+ print "Autodetection failed"
+ return
+
+ if opts.type == "auto_once":
+ opts.type = card.name
+
+ elif opts.type in ctypes:
+ card = ctypes[opts.type](scc)
+
+ else:
+ raise ValueError("Unknown card type %s" % opts.type)
+
+ return card
+
+
+def print_parameters(params):
+
+ print """Generated card parameters :
+ > Name : %(name)s
+ > SMSP : %(smsp)s
+ > ICCID : %(iccid)s
+ > MCC/MNC : %(mcc)d/%(mnc)d
+ > IMSI : %(imsi)s
+ > Ki : %(ki)s
+""" % params
+
+
+#
+# Main
+#
+
+def parse_options():
+
+ parser = OptionParser(usage="usage: %prog [options]")
+
+ # Card interface
+ parser.add_option("-d", "--device", dest="device", metavar="DEV",
+ help="Serial Device for SIM access [default: %default]",
+ default="/dev/ttyUSB0",
+ )
+ parser.add_option("-b", "--baud", dest="baudrate", type="int", metavar="BAUD",
+ help="Baudrate used for SIM access [default: %default]",
+ default=9600,
+ )
+ parser.add_option("-p", "--pcsc-device", dest="pcsc_dev", type='int', metavar="PCSC",
+ help="Which PC/SC reader number for SIM access",
+ default=None,
+ )
+ parser.add_option("-t", "--type", dest="type",
+ help="Card type (user -t list to view) [default: %default]",
+ default="auto",
+ )
+ parser.add_option("-e", "--erase", dest="erase", action='store_true',
+ help="Erase beforehand [default: %default]",
+ default=False,
+ )
+
+ # Data source
+ parser.add_option("--state", dest="state_file", metavar="FILE",
+ help="Use this state file",
+ )
+ parser.add_option("--read-csv", dest="read_csv", metavar="FILE",
+ help="Read parameters from CSV file",
+ )
+
+ (options, args) = parser.parse_args()
+
+ if options.type == 'list':
+ for kls in _cards_classes:
+ print kls.name
+ sys.exit(0)
+
+ if args:
+ parser.error("Extraneous arguments")
+
+ return options
+
+
+def main():
+
+ # Parse options
+ opts = parse_options()
+
+ # Connect to the card
+ if opts.pcsc_dev is None:
+ from pySim.transport.serial import SerialSimLink
+ sl = SerialSimLink(device=opts.device, baudrate=opts.baudrate)
+ else:
+ from pySim.transport.pcsc import PcscSimLink
+ sl = PcscSimLink(opts.pcsc_dev)
+
+ # Create command layer
+ scc = SimCardCommands(transport=sl)
+
+ # Load state
+ sm = StateManager(opts.state_file)
+ sm.load()
+
+ np = sm.network
+
+ # Load cards
+ cards = csv_load_cards(opts.read_csv)
+
+ # Iterate
+ done = False
+ first = True
+ card = None
+
+ while not done:
+ # Connect transport
+ print "Insert card now (or CTRL-C to cancel)"
+ sl.wait_for_card(newcardonly=not first)
+
+ # Not the first anymore !
+ first = False
+
+ # Get card
+ card = card_detect(opts, scc)
+ if card is None:
+ if opts.batch_mode:
+ first = False
+ continue
+ else:
+ sys.exit(-1)
+
+ # Erase if requested
+ if opts.erase:
+ print "Formatting ..."
+ card.erase()
+ card.reset()
+
+ # Get parameters
+ cp = cards[sm.next_write_num()]
+ cpp = {
+ 'name': np.name,
+ 'smsp': np.smsp,
+ 'iccid': cp.iccid,
+ 'mcc': np.mcc,
+ 'mnc': np.mnc,
+ 'imsi': cp.imsi,
+ 'ki': cp.ki,
+ }
+ print_parameters(cpp)
+
+ # Program the card
+ print "Programming ..."
+ card.program(cpp)
+
+ # Update state
+ sm.save()
+
+ # Done for this card and maybe for everything ?
+ print "Card written !\n"
+
+
+if __name__ == '__main__':
+ main()