コード例 #1
0
def main():
    #Getting here implies all pre-requisites have been met (NOT FINISHED!)
    accountName = options.a
    ipAddress = options.i
    port = int(options.p)

    request = {}
    response = {}
    request["accountName"] = accountName
    request["pin"] = getCardPin()
    response["account"] = accountName
    if options.n:
        request["action"] = "new"
        request["amount"] = float(options.n)
        response["initial_balance"] = float(options.n)
    elif options.d:
        request["action"] = "deposit"
        request["amount"] = float(options.d)
        response["deposit"] = float(options.d)
    elif options.w:
        request["action"] = "withdraw"
        request["amount"] = float(options.w)
        response["withdraw"] = float(options.w)
    elif options.g:
        request["action"] = "get"

    request = {"request" : request}
    # print >> sys.stderr, request
    sys.stdout.flush()
    message = NetMsg(request)
    encodedMessage = message.encryptedJson(getAuthKey(), message.getJson())

    # Create socket and send data to bank
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        sock.connect((ipAddress, port))
        sock.sendall(encodedMessage)

        sock.settimeout(10.0)
        received = sock.recv(1024)
        sock.settimeout(None)
        received = json.loads(NetMsg.decryptJson(getAuthKey(), received))
        if received["msg"] == "transaction_completed":
            if options.g:
                response["balance"] = received["balance"]
            print json.dumps(response)
            sys.stdout.flush()
        elif received["msg"] == "transaction_failed":
            sys.exit(255)
        elif received["msg"] == "protocol_error":
            sys.exit(63)
    except socket.error, e:
        print >> sys.stderr, e
        sys.stdout.flush()
        sys.exit(63)
コード例 #2
0
def generate_auth_file(auth_file):
    global auth_key
    if not os.path.isfile(auth_file):
        f = open(auth_file, "wb")
        json_out = {}
        json_out["SecretKey"] = NetMsg.generateKey()
        auth_key = json_out["SecretKey"]
        f.write(json.dumps(json_out))
        f.close()
        # print "created"
    else:
        exit("Authenication file already exists\n")
コード例 #3
0
try:
    if indexData[options.a] == options.c and options.n: #Trying to make a new account when it already exists
        sys.exit(255)
    elif indexData[options.a] != options.c: #The card file you passed does not match the file we expect
        sys.exit(255)
except KeyError:
    for entry in indexData:
        if indexData[entry] == options.c: #If card file is associated with another account
            sys.exit(255)

cardFilePath = os.path.join("./CardFiles", options.c)
try:
    if not os.path.exists(cardFilePath) and options.n:
        cardFile = open(cardFilePath, "w")
        cardPin = NetMsg.generateKey()
        cardFile.write(cardPin)
        cardFile.close()
        
        indexData[options.a] = options.c
        indexFile = open("./CardFiles/index", "w")
        indexFile.write(json.dumps(indexData))
        indexFile.close()
except IOError:
    sys.exit(255)


def getAuthKey():
    key = {}
    numLines = 0
    for line in authFile:
コード例 #4
0
 def send_ack(self, ack):
     ack = NetMsg.encryptedJson(auth_key, ack)
     self.request.sendall(ack)
コード例 #5
0
    def parse_request(self, data):
        try:
            # decrypt message
            net_msg = json.loads(NetMsg.decryptJson(auth_key, data))
            msg = net_msg["msg"]
            response = {}
            response["msgID"] = net_msg["msgID"] + 1
            response["code"] = 0

            # perform atm request
            request = msg["request"]
            if request["action"] == "new":
                if request["amount"] < 10:
                    self.print_to_stderr("Amount was less than 10\n")
                    response["msg"] = "transaction_failed"
                    self.send_ack(json.dumps(response))
                    # self.send_ack("transaction_failed")
                    return

                if self.get_acct_w_acct_name_n_card_key(request["accountName"],
                                                        request["pin"]):
                    self.print_to_stderr("Account already exists\n")
                    response["msg"] = "transaction_failed"
                    # self.send_ack("transaction_failed")
                    return

                new_account = Account(request["pin"],
                                      request["accountName"], request["amount"])
                self.accounts.append(new_account)
                response["msg"] = "transaction_completed"
                self.send_ack(json.dumps(response))
                # self.send_ack("transaction_completed")
            elif request["action"] == "deposit":
                if request["amount"] <= 0:
                    self.print_to_stderr("Amount was less than or equal to 0\n")
                    response["msg"] = "transaction_failed"
                    self.send_ack(json.dumps(response))
                    # self.send_ack("transaction_failed")
                    return

                account = self.get_acct_w_acct_name_n_card_key(request[
                                                               "accountName"],
                                                               request["pin"])

                if account is None:
                    self.print_to_stderr("Could not verify user account\n")
                    response["msg"] = "transaction_failed"
                    self.send_ack(json.dumps(response))
                    # self.send_ack("transaction_failed")
                    return
                else:
                    account.deposit(request["amount"])
                    response["msg"] = "transaction_completed"
                    response["balance"] = account.get_balance()
                    self.send_ack(json.dumps(response))
                    # self.send_ack("transaction_completed")
            elif request["action"] == "withdraw":
                # account = self.get_acct_w_acct_name_n_card_key(request["pin"])
                account = self.get_acct_w_acct_name_n_card_key(request[
                                                               "accountName"],
                                                               request["pin"])
                if account is None:
                    self.print_to_stderr("Could not verify user account\n")
                    response["msg"] = "transaction_failed"
                    self.send_ack(json.dumps(response))
                    # self.send_ack("transaction_failed")
                    return
                else:
                    if not account.withdraw(request["amount"]):
                        response["msg"] = "transaction_failed"
                        response["code"] = 419
                        self.send_ack(json.dumps(response))
                        # self.request.sendall("419")
                    else:
                        # self.send_ack("transaction_completed")
                        response["msg"] = "transaction_completed"
                        response["balance"] = account.get_balance()
                        self.send_ack(json.dumps(response))
            elif request["action"] == "get":
                account = self.get_acct_w_acct_name_n_card_key(request[
                                                               "accountName"],
                                                               request["pin"])

                if account is None:
                    self.print_to_stderr("Could not verify user account\n")
                    return
                else:
                    account.current_balance()
                    response["msg"] = "transaction_completed"
                    response["balance"] = account.get_balance()
                    self.send_ack(json.dumps(response))
                    # self.send_ack(str(account.get_balance()))
        except ValueError:
            # self.request.sendall("protocol_error")
            response["msg"] = "protocol_error"
            self.send_ack(json.dumps(response))
            print "protocol_error"