bemaniutils/bemani/client/mga/mga.py

import base64
import time
from typing import Optional

from bemani.client.base import BaseClient
from bemani.protocol import Node


class MetalGearArcadeClient(BaseClient):
    NAME = "TEST"

    def verify_eventlog_write(self, location: str) -> None:
        call = self.call_node()

        # Construct node
        eventlog = Node.void("eventlog")
        call.add_child(eventlog)
        eventlog.set_attribute("method", "write")
        eventlog.add_child(Node.u32("retrycnt", 0))
        data = Node.void("data")
        eventlog.add_child(data)
        data.add_child(Node.string("eventid", "S_PWRON"))
        data.add_child(Node.s32("eventorder", 0))
        data.add_child(Node.u64("pcbtime", int(time.time() * 1000)))
        data.add_child(Node.s64("gamesession", -1))
        data.add_child(Node.string("strdata1", "1.9.1"))
        data.add_child(Node.string("strdata2", ""))
        data.add_child(Node.s64("numdata1", 1))
        data.add_child(Node.s64("numdata2", 0))
        data.add_child(Node.string("locationid", location))

        # Swap with server
        resp = self.exchange("", call)

        # Verify that response is correct
        self.assert_path(resp, "response/eventlog/gamesession")
        self.assert_path(resp, "response/eventlog/logsendflg")
        self.assert_path(resp, "response/eventlog/logerrlevel")
        self.assert_path(resp, "response/eventlog/evtidnosendflg")

    def verify_system_getmaster(self) -> None:
        call = self.call_node()

        # Construct node
        system = Node.void("system")
        call.add_child(system)
        system.set_attribute("method", "getmaster")
        data = Node.void("data")
        system.add_child(data)
        data.add_child(Node.string("gamekind", "I36"))
        data.add_child(Node.string("datatype", "S_SRVMSG"))
        data.add_child(Node.string("datakey", "INFO"))

        # Swap with server
        resp = self.exchange("", call)

        # Verify that response is correct
        self.assert_path(resp, "response/system/result")

    def verify_usergamedata_send(self, ref_id: str, msg_type: str) -> None:
        call = self.call_node()

        # Set up profile write
        profiledata = [
            b"ffffffff",
            b"PLAYDATA",
            b"8",
            b"1",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"ffffffffffffa928",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"PLAYER",
            b"JP-13",
            b"ea",
            b"",
            b"JP-13",
            b"",
            b"",
            b"",
        ]

        outfitdata = [
            b"ffffffff",
            b"OUTFIT",
            b"8",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"202000020400800",
            b"1000100",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAABAAAAAAAAABwACxEWEUgRWBEuADkRZgBmAAAcAAsRFhFIEVgRLgA5EWYAZgAAHAALERYRSBFYES4AORFmAGYA",
            b"AAAAAA==",
            b"",
        ]

        weapondata = [
            b"ffffffff",
            b"WEAPON",
            b"8",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"201000000003",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
        ]

        mainruledata = [
            b"ffffffff",
            b"MAINRULE",
            b"8",
            b"6",
            b"800000",
            b"1",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"0",
            b"1.000000",
            b"0.000000",
            b"10.000000",
            b"4.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"0.000000",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
            b"AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAEAAAAAAAA=",
            b"",
            b"",
            b"",
            b"",
        ]

        if msg_type == "new":
            # New profile gets blank name, because we save over it at the end of the round.
            profiledata[27] = b""
        elif msg_type == "existing":
            # Exiting profile gets our hardcoded name saved.
            profiledata[27] = self.NAME.encode("shift-jis")

        # Construct node
        playerdata = Node.void("playerdata")
        call.add_child(playerdata)
        playerdata.set_attribute("method", "usergamedata_send")
        playerdata.add_child(Node.u32("retrycnt", 0))

        data = Node.void("data")
        playerdata.add_child(data)
        data.add_child(Node.string("eaid", ref_id))
        data.add_child(Node.string("gamekind", "I36"))
        data.add_child(Node.u32("datanum", 4))
        record = Node.void("record")
        data.add_child(record)
        d = Node.string("d", base64.b64encode(b",".join(profiledata)).decode("ascii"))
        record.add_child(d)
        d.add_child(Node.string("bin1", ""))
        d = Node.string("d", base64.b64encode(b",".join(outfitdata)).decode("ascii"))
        record.add_child(d)
        d.add_child(Node.string("bin1", ""))
        d = Node.string("d", base64.b64encode(b",".join(weapondata)).decode("ascii"))
        record.add_child(d)
        d.add_child(Node.string("bin1", ""))
        d = Node.string("d", base64.b64encode(b",".join(mainruledata)).decode("ascii"))
        record.add_child(d)
        d.add_child(Node.string("bin1", ""))

        # Swap with server
        resp = self.exchange("", call)
        self.assert_path(resp, "response/playerdata/result")

    def verify_usergamedata_recv(self, ref_id: str) -> str:
        call = self.call_node()

        # Construct node
        playerdata = Node.void("playerdata")
        call.add_child(playerdata)
        playerdata.set_attribute("method", "usergamedata_recv")
        data = Node.void("data")
        playerdata.add_child(data)
        data.add_child(Node.string("eaid", ref_id))
        data.add_child(Node.string("gamekind", "I36"))
        data.add_child(Node.u32("recv_num", 4))
        data.add_child(
            Node.string(
                "recv_csv",
                "PLAYDATA,3fffffffff,OUTFIT,3fffffffff,WEAPON,3fffffffff,MAINRULE,3fffffffff",
            )
        )

        # Swap with server
        resp = self.exchange("", call)
        self.assert_path(resp, "response/playerdata/result")
        self.assert_path(resp, "response/playerdata/player/record/d/bin1")
        self.assert_path(resp, "response/playerdata/player/record_num")

        # Grab binary data, parse out name
        bindata = resp.child_value("playerdata/player/record/d")
        profiledata = base64.b64decode(bindata).split(b",")

        # We lob off the first two values in returning profile, so the name is offset by two
        return profiledata[25].decode("shift-jis")

    def verify(self, cardid: Optional[str]) -> None:
        # Verify boot sequence is okay
        self.verify_services_get(
            expected_services=[
                "pcbtracker",
                "pcbevent",
                "local",
                "message",
                "facility",
                "cardmng",
                "package",
                "posevent",
                "pkglist",
                "dlstatus",
                "eacoin",
                "lobby",
                "ntp",
                "keepalive",
            ]
        )
        paseli_enabled = self.verify_pcbtracker_alive()
        self.verify_message_get()
        self.verify_package_list()
        location = self.verify_facility_get()
        self.verify_pcbevent_put()
        self.verify_eventlog_write(location)
        self.verify_system_getmaster()

        # Verify card registration and profile lookup
        if cardid is not None:
            card = cardid
        else:
            card = self.random_card()
            print(f"Generated random card ID {card} for use.")

        if cardid is None:
            self.verify_cardmng_inquire(
                card, msg_type="unregistered", paseli_enabled=paseli_enabled
            )
            ref_id = self.verify_cardmng_getrefid(card)
            if len(ref_id) != 16:
                raise Exception(
                    f"Invalid refid '{ref_id}' returned when registering card"
                )
            if ref_id != self.verify_cardmng_inquire(
                card, msg_type="new", paseli_enabled=paseli_enabled
            ):
                raise Exception(f"Invalid refid '{ref_id}' returned when querying card")
            # MGA doesn't read a new profile, it just writes out CSV for a blank one
            self.verify_usergamedata_send(ref_id, msg_type="new")
        else:
            print("Skipping new card checks for existing card")
            ref_id = self.verify_cardmng_inquire(
                card, msg_type="query", paseli_enabled=paseli_enabled
            )

        # Verify pin handling and return card handling
        self.verify_cardmng_authpass(ref_id, correct=True)
        self.verify_cardmng_authpass(ref_id, correct=False)
        if ref_id != self.verify_cardmng_inquire(
            card, msg_type="query", paseli_enabled=paseli_enabled
        ):
            raise Exception(f"Invalid refid '{ref_id}' returned when querying card")

        if cardid is None:
            # Verify profile saving
            name = self.verify_usergamedata_recv(ref_id)
            if name != "":
                raise Exception("New profile has a name associated with it!")

            self.verify_usergamedata_send(ref_id, msg_type="existing")
            name = self.verify_usergamedata_recv(ref_id)
            if name != self.NAME:
                raise Exception("Existing profile has no name associated with it!")
        else:
            print("Skipping score checks for existing card")