import random import time from typing import Any, Dict, List, Optional from bemani.client.base import BaseClient from bemani.protocol import Node class DDR2014Client(BaseClient): NAME = "TEST" def verify_game_shop(self, loc: str) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "shop") game.set_attribute("area", "51") game.set_attribute("boot", "34") game.set_attribute("close", "0") game.set_attribute("close_t", "0") game.set_attribute("coin", "02.01.--.--.01.G") game.set_attribute("diff", "3") game.set_attribute("during", "1") game.set_attribute("edit_cnt", "0") game.set_attribute("edit_used", "1") game.set_attribute("first", "1") game.set_attribute("ip", "1.5.7.3") game.set_attribute("is_freefirstplay", "1") game.set_attribute("is_paseli", "1") game.set_attribute("loc", loc) game.set_attribute("mac", "00:11:22:33:44:55") game.set_attribute("machine", "2") game.set_attribute("name", "ＴＥＳＴ") game.set_attribute("pay", "0") game.set_attribute("region", ".") game.set_attribute("soft", self.config["model"]) game.set_attribute("softid", self.pcbid) game.set_attribute("stage", "1") game.set_attribute("time", "60") game.set_attribute("type", "0") game.set_attribute("ver", "2014102700") # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game/@stop") def verify_game_common(self) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "common") game.set_attribute("ver", "2014102700") # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game/flag/@id") self.assert_path(resp, "response/game/flag/@s1") self.assert_path(resp, "response/game/flag/@s2") self.assert_path(resp, "response/game/flag/@t") self.assert_path(resp, "response/game/cnt_music_monthly") self.assert_path(resp, "response/game/cnt_music_weekly") self.assert_path(resp, "response/game/cnt_music_daily") def verify_game_hiscore(self) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "hiscore") game.set_attribute("ver", "2014102700") # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game") for child in resp.child("game").children: self.assert_path(child, "music/@reclink_num") self.assert_path(child, "music/type/@diff") self.assert_path(child, "music/type/name") self.assert_path(child, "music/type/score") self.assert_path(child, "music/type/area") self.assert_path(child, "music/type/rank") self.assert_path(child, "music/type/combo_type") self.assert_path(child, "music/type/code") def verify_game_area_hiscore(self) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "area_hiscore") game.set_attribute("shop_area", "51") game.set_attribute("ver", "2014102700") # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game") for child in resp.child("game").children: self.assert_path(child, "music/@reclink_num") self.assert_path(child, "music/type/@diff") self.assert_path(child, "music/type/name") self.assert_path(child, "music/type/score") self.assert_path(child, "music/type/area") self.assert_path(child, "music/type/rank") self.assert_path(child, "music/type/combo_type") self.assert_path(child, "music/type/code") def verify_game_message(self) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "message") game.set_attribute("ver", "2014102700") # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game") def verify_game_ranking(self) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "ranking") game.set_attribute("max", "10") game.set_attribute("ver", "2014102700") # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game") def verify_game_log(self) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "log") game.set_attribute("type", "0") game.set_attribute("soft", self.config["model"]) game.set_attribute("softid", self.pcbid) game.set_attribute("ver", "2014102700") game.set_attribute("boot", "34") game.set_attribute("mac", "00:11:22:33:44:55") clear = Node.void("clear") game.add_child(clear) clear.set_attribute("book", "0") clear.set_attribute("edit", "0") clear.set_attribute("rank", "0") clear.set_attribute("set", "0") auto = Node.void("auto") game.add_child(auto) auto.set_attribute("book", "1") auto.set_attribute("edit", "1") auto.set_attribute("rank", "1") auto.set_attribute("set", "1") # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game") def verify_game_tax_info(self) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "tax_info") game.set_attribute("ver", "2014102700") # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game/tax_info/@tax_phase") def verify_game_recorder(self) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "recorder") game.set_attribute("assert_cnt", "0") game.set_attribute("assert_info", "") game.set_attribute("assert_path", "") game.set_attribute("assert_time", "0") game.set_attribute("boot_time", "1706151228") game.set_attribute("cnt_demo", "1") game.set_attribute("cnt_music", "1") game.set_attribute("cnt_play", "0") game.set_attribute("last_mid", "481") game.set_attribute("last_seq", "36") game.set_attribute("last_step", "0") game.set_attribute("last_time", "1706151235") game.set_attribute("softcode", self.config["model"]) game.set_attribute("temp_seq", "15") game.set_attribute("temp_step", "8") game.set_attribute("temp_time", "1706151234") game.set_attribute("wd_restart", "0") # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game") def verify_game_lock(self, ref_id: str, play: int) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("refid", ref_id) game.set_attribute("method", "lock") game.set_attribute("ver", "2014102700") game.set_attribute("play", str(play)) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game/@now_login") def verify_game_new(self, ref_id: str) -> None: # Pad the name to 8 characters name = self.NAME[:8] while len(name) < 8: name = name + " " call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "new") game.set_attribute("ver", "2014102700") game.set_attribute("name", name) game.set_attribute("area", "51") game.set_attribute("old", "0") game.set_attribute("refid", ref_id) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game") def verify_game_load_daily(self, ref_id: str) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "load_daily") game.set_attribute("ver", "2014102700") game.set_attribute("refid", ref_id) # Swap with server resp = self.exchange("", call) self.assert_path(resp, "response/game/daycount/@playcount") self.assert_path(resp, "response/game/dailycombo/@daily_combo") self.assert_path(resp, "response/game/dailycombo/@daily_combo_lv") def verify_game_load(self, ref_id: str, msg_type: str) -> Dict[str, Any]: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "load") game.set_attribute("ver", "2014102700") game.set_attribute("refid", ref_id) # Swap with server resp = self.exchange("", call) if msg_type == "new": # Verify that response is correct self.assert_path(resp, "response/game/@none") return {} if msg_type == "existing": # Verify existing profile and return info self.assert_path(resp, "response/game/seq") self.assert_path(resp, "response/game/code") self.assert_path(resp, "response/game/name") self.assert_path(resp, "response/game/area") self.assert_path(resp, "response/game/cnt_s") self.assert_path(resp, "response/game/cnt_d") self.assert_path(resp, "response/game/cnt_b") self.assert_path(resp, "response/game/cnt_m0") self.assert_path(resp, "response/game/cnt_m1") self.assert_path(resp, "response/game/cnt_m2") self.assert_path(resp, "response/game/cnt_m3") self.assert_path(resp, "response/game/cnt_m4") self.assert_path(resp, "response/game/cnt_m5") self.assert_path(resp, "response/game/exp") self.assert_path(resp, "response/game/exp_o") self.assert_path(resp, "response/game/star") self.assert_path(resp, "response/game/star_c") self.assert_path(resp, "response/game/combo") self.assert_path(resp, "response/game/timing_diff") self.assert_path(resp, "response/game/chara") self.assert_path(resp, "response/game/chara_opt") self.assert_path(resp, "response/game/daycount/@playcount") self.assert_path(resp, "response/game/dailycombo/@daily_combo") self.assert_path(resp, "response/game/dailycombo/@daily_combo_lv") self.assert_path(resp, "response/game/last/@cate") self.assert_path(resp, "response/game/last/@cid") self.assert_path(resp, "response/game/last/@ctype") self.assert_path(resp, "response/game/last/@fri") self.assert_path(resp, "response/game/last/@mid") self.assert_path(resp, "response/game/last/@mode") self.assert_path(resp, "response/game/last/@mtype") self.assert_path(resp, "response/game/last/@rival1") self.assert_path(resp, "response/game/last/@rival2") self.assert_path(resp, "response/game/last/@rival3") self.assert_path(resp, "response/game/last/@sid") self.assert_path(resp, "response/game/last/@sort") self.assert_path(resp, "response/game/last/@style") self.assert_path(resp, "response/game/result_star/@slot1") self.assert_path(resp, "response/game/result_star/@slot2") self.assert_path(resp, "response/game/result_star/@slot3") self.assert_path(resp, "response/game/result_star/@slot4") self.assert_path(resp, "response/game/result_star/@slot5") self.assert_path(resp, "response/game/result_star/@slot6") self.assert_path(resp, "response/game/result_star/@slot7") self.assert_path(resp, "response/game/result_star/@slot8") self.assert_path(resp, "response/game/result_star/@slot9") self.assert_path(resp, "response/game/gr_s/@gr1") self.assert_path(resp, "response/game/gr_s/@gr2") self.assert_path(resp, "response/game/gr_s/@gr3") self.assert_path(resp, "response/game/gr_s/@gr4") self.assert_path(resp, "response/game/gr_s/@gr5") self.assert_path(resp, "response/game/gr_d/@gr1") self.assert_path(resp, "response/game/gr_d/@gr2") self.assert_path(resp, "response/game/gr_d/@gr3") self.assert_path(resp, "response/game/gr_d/@gr4") self.assert_path(resp, "response/game/gr_d/@gr5") self.assert_path(resp, "response/game/opt") self.assert_path(resp, "response/game/opt_ex") self.assert_path(resp, "response/game/option_ver/@ver") self.assert_path(resp, "response/game/flag") self.assert_path(resp, "response/game/flag_ex") self.assert_path(resp, "response/game/rank") self.assert_path(resp, "response/game/target/@flag") self.assert_path(resp, "response/game/target/@setnum") for i in range(55): self.assert_path(resp, f"response/game/play_area/@play_cnt{i}") gr_s = resp.child("game/gr_s") gr_d = resp.child("game/gr_d") return { "name": resp.child_value("game/name"), "ext_id": resp.child_value("game/code"), "single_plays": resp.child_value("game/cnt_s"), "double_plays": resp.child_value("game/cnt_d"), "groove_single": [ int(gr_s.attribute("gr1")), int(gr_s.attribute("gr2")), int(gr_s.attribute("gr3")), int(gr_s.attribute("gr4")), int(gr_s.attribute("gr5")), ], "groove_double": [ int(gr_d.attribute("gr1")), int(gr_d.attribute("gr2")), int(gr_d.attribute("gr3")), int(gr_d.attribute("gr4")), int(gr_d.attribute("gr5")), ], } raise Exception("Unknown load type!") def verify_game_load_m(self, ref_id: str) -> Dict[int, Dict[int, Dict[str, Any]]]: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("ver", "2014102700") game.set_attribute("all", "1") game.set_attribute("refid", ref_id) game.set_attribute("method", "load_m") # Swap with server resp = self.exchange("", call) # Verify that response is correct scores: Dict[int, Dict[int, Dict[str, Any]]] = {} self.assert_path(resp, "response/game") for child in resp.child("game").children: self.assert_path(child, "music/@reclink") reclink = int(child.attribute("reclink")) for typenode in child.children: self.assert_path(typenode, "type/@diff") self.assert_path(typenode, "type/score") self.assert_path(typenode, "type/count") self.assert_path(typenode, "type/rank") self.assert_path(typenode, "type/combo_type") chart = int(typenode.attribute("diff")) vals = { "score": typenode.child_value("score"), "count": typenode.child_value("count"), "rank": typenode.child_value("rank"), "halo": typenode.child_value("combo_type"), } if reclink not in scores: scores[reclink] = {} scores[reclink][chart] = vals return scores def verify_game_load_edit(self, ref_id: str) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("ver", "2014102700") game.set_attribute("pid", "0") game.set_attribute("refid", ref_id) game.set_attribute("method", "load_edit") # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game") def verify_game_save( self, ref_id: str, style: int, gauge: Optional[List[int]] = None ) -> None: gauge = gauge or [0, 0, 0, 0, 0] call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "save") game.set_attribute("refid", ref_id) game.set_attribute("ver", "2014102700") game.set_attribute("shop_area", "51") last = Node.void("last") game.add_child(last) last.set_attribute("mode", "1") last.set_attribute("style", str(style)) gr = Node.void("gr") game.add_child(gr) gr.set_attribute("gr1", str(gauge[0])) gr.set_attribute("gr2", str(gauge[1])) gr.set_attribute("gr3", str(gauge[2])) gr.set_attribute("gr4", str(gauge[3])) gr.set_attribute("gr5", str(gauge[4])) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game") def verify_game_save_m( self, ref_id: str, ext_id: str, score: Dict[str, Any] ) -> None: call = self.call_node() game = Node.void("game") call.add_child(game) game.set_attribute("method", "save_m") game.set_attribute("diff", "12345") game.set_attribute("mtype", str(score["chart"])) game.set_attribute("mid", str(score["id"])) game.set_attribute("refid", ref_id) game.set_attribute("ver", "2014102700") data = Node.void("data") game.add_child(data) data.set_attribute("score", str(score["score"])) data.set_attribute("rank", str(score["rank"])) data.set_attribute("shop_area", "0") data.set_attribute("playmode", "1") data.set_attribute("combo", str(score["combo"])) data.set_attribute("phase", "1") data.set_attribute("style", "0") data.set_attribute("full", "1" if score["halo"] >= 1 else "0") data.set_attribute("great_fc", "1" if score["halo"] == 1 else "0") data.set_attribute("good_fc", "1" if score["halo"] == 4 else "0") data.set_attribute("perf_fc", "1" if score["halo"] == 2 else "0") gauge = Node.void("gauge") game.add_child(gauge) gauge.set_attribute("life8", "0") gauge.set_attribute("assist", "0") gauge.set_attribute("risky", "0") gauge.set_attribute("life4", "0") gauge.set_attribute("hard", "0") player = Node.void("player") game.add_child(player) player.set_attribute("playcnt", "123") player.set_attribute("code", ext_id) option = Node.void("option_02") game.add_child(option) option.set_attribute("opt02_0", "6") option.set_attribute("opt02_6", "1") option.set_attribute("opt02_13", "2") game.add_child(Node.u8_array("trace", [0] * 512)) game.add_child(Node.u32("size", 512)) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/game") 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("EUC_JP") self.verify_pcbevent_put() self.verify_game_recorder() self.verify_game_tax_info() self.verify_game_shop(location) self.verify_game_common() self.verify_game_hiscore() self.verify_game_area_hiscore() self.verify_game_message() self.verify_game_ranking() self.verify_game_log() # 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_game_load(ref_id, msg_type="new") self.verify_game_new(ref_id) 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") # Verify locking and unlocking profile ability self.verify_game_lock(ref_id, 1) self.verify_game_lock(ref_id, 0) if cardid is None: # Verify empty profile profile = self.verify_game_load(ref_id, msg_type="existing") ext_id = str(profile["ext_id"]) if profile["name"] != self.NAME: raise Exception("Profile has invalid name associated with it!") if profile["single_plays"] != 0: raise Exception("Profile has plays on single already!") if profile["double_plays"] != 0: raise Exception("Profile has plays on double already!") if any([g != 0 for g in profile["groove_single"]]): raise Exception("Profile has single groove gauge values already!") if any([g != 0 for g in profile["groove_double"]]): raise Exception("Profile has double groove gauge values already!") # Verify empty scores scores = self.verify_game_load_m(ref_id) if len(scores) > 0: raise Exception("Scores exist on new profile!") self.verify_game_load_edit(ref_id) self.verify_game_load_daily(ref_id) # Verify profile saving self.verify_game_save(ref_id, 0, [1, 2, 3, 4, 5]) profile = self.verify_game_load(ref_id, msg_type="existing") if profile["name"] != self.NAME: raise Exception("Profile has invalid name associated with it!") if profile["single_plays"] != 1: raise Exception("Profile has invalid plays on single!") if profile["double_plays"] != 0: raise Exception("Profile has invalid plays on double!") if profile["groove_single"] != [1, 2, 3, 4, 5]: raise Exception("Profile has invalid single groove gauge values!") if any([g != 0 for g in profile["groove_double"]]): raise Exception("Profile has invalid double groove gauge values!") self.verify_game_save(ref_id, 1, [5, 4, 3, 2, 1]) profile = self.verify_game_load(ref_id, msg_type="existing") if profile["name"] != self.NAME: raise Exception("Profile has invalid name associated with it!") if profile["single_plays"] != 1: raise Exception("Profile has invalid plays on single!") if profile["double_plays"] != 1: raise Exception("Profile has invalid plays on double!") if profile["groove_single"] != [1, 2, 3, 4, 5]: raise Exception("Profile has invalid single groove gauge values!") if profile["groove_double"] != [5, 4, 3, 2, 1]: raise Exception("Profile has invalid double groove gauge values!") # Now, write some scores and verify saving for phase in [1, 2]: if phase == 1: dummyscores = [ # An okay score on a chart { "id": 593, "chart": 3, "score": 800000, "combo": 123, "rank": 4, "halo": 1, }, # A good score on an easier chart same song { "id": 593, "chart": 2, "score": 990000, "combo": 321, "rank": 2, "halo": 2, }, # A perfect score { "id": 483, "chart": 3, "score": 1000000, "combo": 400, "rank": 1, "halo": 3, }, # A bad score { "id": 483, "chart": 2, "score": 100000, "combo": 5, "rank": 7, "halo": 0, }, { "id": 483, "chart": 1, "score": 60000, "combo": 5, "rank": 6, "halo": 4, }, ] if phase == 2: dummyscores = [ # A better score on a chart { "id": 593, "chart": 3, "score": 850000, "combo": 234, "rank": 3, "halo": 2, }, # A worse score on another chart { "id": 593, "chart": 2, "score": 980000, "combo": 300, "rank": 3, "halo": 0, "expected_score": 990000, "expected_rank": 2, "expected_halo": 2, }, ] for score in dummyscores: self.verify_game_save_m(ref_id, ext_id, score) scores = self.verify_game_load_m(ref_id) for score in dummyscores: data = scores.get(score["id"], {}).get(score["chart"], None) if data is None: raise Exception( f'Expected to get score back for song {score["id"]} chart {score["chart"]}!' ) # Verify the attributes of the score expected_score = score.get("expected_score", score["score"]) expected_rank = score.get("expected_rank", score["rank"]) expected_halo = score.get("expected_halo", score["halo"]) if data["score"] != expected_score: raise Exception( f'Expected a score of \'{expected_score}\' for song \'{score["id"]}\' chart \'{score["chart"]}\' but got score \'{data["score"]}\'' ) if data["rank"] != expected_rank: raise Exception( f'Expected a rank of \'{expected_rank}\' for song \'{score["id"]}\' chart \'{score["chart"]}\' but got rank \'{data["rank"]}\'' ) if data["halo"] != expected_halo: raise Exception( f'Expected a halo of \'{expected_halo}\' for song \'{score["id"]}\' chart \'{score["chart"]}\' but got halo \'{data["halo"]}\'' ) # Sleep so we don't end up putting in score history on the same second time.sleep(1) else: print("Skipping score checks for existing card") # Verify paseli handling if paseli_enabled: print("PASELI enabled for this PCBID, executing PASELI checks") else: print("PASELI disabled for this PCBID, skipping PASELI checks") return sessid, balance = self.verify_eacoin_checkin(card) if balance == 0: print("Skipping PASELI consume check because card has 0 balance") else: self.verify_eacoin_consume(sessid, balance, random.randint(0, balance)) self.verify_eacoin_checkout(sessid)