# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from autotest_lib.client.bin import test
from autotest_lib.client.common_lib import error

import os, time
import dbus, dbus.mainloop.glib, gobject
import random

from autotest_lib.client.cros import flimflam_test_path
import mm

class TestFailure(Exception):
    pass

class network_LockedSIM(test.test):
    version = 1

    def modem(self, mm):
        return self.bus.get_object(mm[0].service, mm[1])

    def EnableModem(self, mm):
        self.modem(mm).Enable(True, dbus_interface=self.imodem)

    def ChangePin(self, mm, old, new):
        self.modem(mm).ChangePin(old, new, dbus_interface=self.icard)

    def EnablePin(self, mm, pin):
        self.modem(mm).EnablePin(pin, True, dbus_interface=self.icard)

    def DisablePin(self, mm, pin):
        self.modem(mm).EnablePin(pin, False, dbus_interface=self.icard)

    def Reset(self, mm):
        self.modem(mm).Reset(dbus_interface=self.imodem)

    def Unlock(self, mm, pin):
        self.modem(mm).SendPin(pin, dbus_interface=self.icard)

    def retries(self, mm):
        return self.modem(mm).Get(self.imodem, 'UnlockRetries',
                                    dbus_interface=self.iprops)

    def run_once(self):
        global mm
        self.iprops = 'org.freedesktop.DBus.Properties'
        self.imm = 'org.freedesktop.ModemManager'
        self.imodem = 'org.freedesktop.ModemManager.Modem'
        self.icard = 'org.freedesktop.ModemManager.Modem.Gsm.Card'
        failed = []
        self.bus = dbus.SystemBus()

        self.devs = mm.EnumerateDevices()
        print 'devs: %d' % len(self.devs)
        for modem in self.devs:
            print 'device: %s' % modem[1]
            # Make sure we can change the pin - this guarantees that the pin is
            # properly set to start with.
            try:
                self.Unlock(modem, '1111')
            except dbus.exceptions.DBusException:
                # We get this back if the sim's already unlocked.
                pass
            self.EnableModem(modem)
            self.ChangePin(modem, '1111', '1112')
            self.ChangePin(modem, '1112', '1111')
            try:
                self.DisablePin(modem, '1111')
            except dbus.exceptions.DBusException:
                # We get this back if the pin's already disabled.
                pass
            self.EnablePin(modem, '1111')
            self.Reset(modem)

        # Give the modem a little while to come back...
        time.sleep(20)

        # Re-enumerate devices, since we're hoping they all disappeared and
        # reappeared.
        self.devs = mm.EnumerateDevices()
        print 'newdevs: %d' % len(self.devs)
        for modem in self.devs:
            print 'newdevice: %s' % modem[1]
            # Send a command to the modem, then wait a second. It seems to take
            # the Ericsson F3307 (at least) from initial access to usefulness,
            # so we make a dummy retries() call, wait a second, then get the
            # real retry count.
            self.retries(modem)
            time.sleep(1)
            retries = self.retries(modem)
            print 'real retries: %u' % retries
            if retries < 2:
                print 'retries too low (%d), bailing' % retries
                failed.append(modem)
                continue
            try:
                self.Unlock(modem, '1112')
            except dbus.exceptions.DBusException:
                pass
                # We expect a failure here, so swallow the DBus exception.
            nretries = self.retries(modem)
            self.Unlock(modem, '1111')
            self.EnableModem(modem)
            self.DisablePin(modem, '1111')
            print '%s retries %d nretries %d' % (modem, retries, nretries)
            if nretries != (retries - 1):
                # We can't just raise the exception here - if there are multiple
                # modems in the system, we might raise on the first one and
                # leave the others locked.
                failed.append(modem)

        if failed:
            raise error.TestFail("Failed for devices: %s" % ', '.join(
                map(lambda x: x[1], failed)))