233 lines
8.0 KiB
Python
233 lines
8.0 KiB
Python
import base64
|
||
import time
|
||
from typing import Optional
|
||
|
||
from bemani.client.base import BaseClient
|
||
from bemani.protocol import Node
|
||
|
||
|
||
class TheStarBishiBashiClient(BaseClient):
|
||
NAME = "ＴＥＳＴ"
|
||
|
||
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.7.6"))
|
||
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", "IBB"))
|
||
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"IBBDAT00",
|
||
b"1",
|
||
b"0",
|
||
b"0",
|
||
b"0",
|
||
b"0",
|
||
b"0",
|
||
b"e474c1b",
|
||
b"0",
|
||
b"0",
|
||
b"ff",
|
||
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"",
|
||
b"",
|
||
b"",
|
||
b"\x96\xa2\x90\xdd\x92\xe8",
|
||
b"\x8d\x81\x8d`",
|
||
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", "IBB"))
|
||
data.add_child(Node.u32("datanum", 1))
|
||
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", ""))
|
||
|
||
# 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", "IBB"))
|
||
data.add_child(Node.u32("recv_num", 1))
|
||
data.add_child(Node.string("recv_csv", "IBBDAT00,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")
|
||
# Bishi 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")
|