1
0
mirror of synced 2024-11-24 14:30:11 +01:00
bemaniutils/bemani/client/popn/eclale.py
Jennifer Taylor 509cb4f0d9 Convert most of the format() string calls to f-strings using libcst.
Exact commands run were:

  python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format
  python3 setup.py build_ext --inplace
2020-01-07 21:29:07 +00:00

593 lines
24 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import random
import time
from typing import Any, Dict, Optional
from bemani.client.base import BaseClient
from bemani.protocol import Node
class PopnMusicEclaleClient(BaseClient):
NAME = ''
def verify_pcb23_boot(self, loc: str) -> None:
call = self.call_node()
# Construct node
pcb23 = Node.void('pcb23')
call.add_child(pcb23)
pcb23.set_attribute('method', 'boot')
pcb23.add_child(Node.string('loc_id', loc))
pcb23.add_child(Node.u8('loc_type', 0))
pcb23.add_child(Node.string('loc_name', ''))
pcb23.add_child(Node.string('country', 'US'))
pcb23.add_child(Node.string('region', '.'))
pcb23.add_child(Node.s16('pref', 51))
pcb23.add_child(Node.string('customer', ''))
pcb23.add_child(Node.string('company', ''))
pcb23.add_child(Node.ipv4('gip', '127.0.0.1'))
pcb23.add_child(Node.u16('gp', 10011))
pcb23.add_child(Node.string('rom_number', 'M39-JB-G01'))
pcb23.add_child(Node.u64('c_drive', 10028228608))
pcb23.add_child(Node.u64('d_drive', 47945170944))
pcb23.add_child(Node.u64('e_drive', 10394677248))
pcb23.add_child(Node.string('etc', ''))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/pcb23/@status")
def __verify_common(self, root: str, resp: Node) -> None:
self.assert_path(resp, f"response/{root}/phase/event_id")
self.assert_path(resp, f"response/{root}/phase/phase")
self.assert_path(resp, f"response/{root}/area/area_id")
self.assert_path(resp, f"response/{root}/area/end_date")
self.assert_path(resp, f"response/{root}/area/medal_id")
self.assert_path(resp, f"response/{root}/area/is_limit")
def verify_info23_common(self, loc: str) -> None:
call = self.call_node()
# Construct node
info23 = Node.void('info23')
call.add_child(info23)
info23.set_attribute('loc_id', loc)
info23.set_attribute('method', 'common')
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.__verify_common('info23', resp)
def verify_lobby22_getlist(self, loc: str) -> None:
call = self.call_node()
# Construct node
lobby22 = Node.void('lobby22')
call.add_child(lobby22)
lobby22.set_attribute('method', 'getList')
lobby22.add_child(Node.string('location_id', loc))
lobby22.add_child(Node.u8('net_version', 53))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/lobby22/@status")
def __verify_profile(self, resp: Node) -> None:
self.assert_path(resp, "response/player23/account/name")
self.assert_path(resp, "response/player23/account/g_pm_id")
self.assert_path(resp, "response/player23/account/tutorial")
self.assert_path(resp, "response/player23/account/area_id")
self.assert_path(resp, "response/player23/account/lumina")
self.assert_path(resp, "response/player23/account/read_news")
self.assert_path(resp, "response/player23/account/welcom_pack")
self.assert_path(resp, "response/player23/account/medal_set")
self.assert_path(resp, "response/player23/account/nice")
self.assert_path(resp, "response/player23/account/favorite_chara")
self.assert_path(resp, "response/player23/account/special_area")
self.assert_path(resp, "response/player23/account/chocolate_charalist")
self.assert_path(resp, "response/player23/account/teacher_setting")
self.assert_path(resp, "response/player23/account/staff")
self.assert_path(resp, "response/player23/account/item_type")
self.assert_path(resp, "response/player23/account/item_id")
self.assert_path(resp, "response/player23/account/is_conv")
self.assert_path(resp, "response/player23/account/meteor_flg")
self.assert_path(resp, "response/player23/account/license_data")
self.assert_path(resp, "response/player23/account/my_best")
self.assert_path(resp, "response/player23/account/latest_music")
self.assert_path(resp, "response/player23/account/active_fr_num")
self.assert_path(resp, "response/player23/account/total_play_cnt")
self.assert_path(resp, "response/player23/account/today_play_cnt")
self.assert_path(resp, "response/player23/account/consecutive_days")
self.assert_path(resp, "response/player23/account/total_days")
self.assert_path(resp, "response/player23/account/interval_day")
self.assert_path(resp, "response/player23/netvs")
self.assert_path(resp, "response/player23/config")
self.assert_path(resp, "response/player23/option")
self.assert_path(resp, "response/player23/info/ep")
self.assert_path(resp, "response/player23/custom_cate")
self.assert_path(resp, "response/player23/customize")
self.assert_path(resp, "response/player23/event/enemy_medal")
self.assert_path(resp, "response/player23/event/hp")
self.assert_path(resp, "response/player23/stamp/stamp_id")
self.assert_path(resp, "response/player23/stamp/cnt")
def verify_player23_read(self, ref_id: str, msg_type: str) -> Dict[str, Dict[int, Dict[str, int]]]:
call = self.call_node()
# Construct node
player23 = Node.void('player23')
call.add_child(player23)
player23.set_attribute('method', 'read')
player23.add_child(Node.string('ref_id', ref_id))
player23.add_child(Node.s8('pref', 51))
# Swap with server
resp = self.exchange('', call)
if msg_type == 'new':
# Verify that response is correct
self.assert_path(resp, "response/player23/result")
status = resp.child_value('player23/result')
if status != 2:
raise Exception(f'Reference ID \'{ref_id}\' returned invalid status \'{status}\'')
return {
'medals': {},
'items': {},
'characters': {},
'lumina': {},
}
elif msg_type == 'query':
# Verify that the response is correct
self.__verify_profile(resp)
self.assert_path(resp, "response/player23/result")
status = resp.child_value('player23/result')
if status != 0:
raise Exception(f'Reference ID \'{ref_id}\' returned invalid status \'{status}\'')
name = resp.child_value('player23/account/name')
if name != self.NAME:
raise Exception(f'Invalid name \'{name}\' returned for Ref ID \'{ref_id}\'')
# Medals and items
items: Dict[int, Dict[str, int]] = {}
medals: Dict[int, Dict[str, int]] = {}
charas: Dict[int, Dict[str, int]] = {}
for obj in resp.child('player23').children:
if obj.name == 'medal':
medals[obj.child_value('medal_id')] = {
'level': obj.child_value('level'),
'exp': obj.child_value('exp'),
}
elif obj.name == 'item':
items[obj.child_value('id')] = {
'type': obj.child_value('type'),
'param': obj.child_value('param'),
}
elif obj.name == 'chara_param':
charas[obj.child_value('chara_id')] = {
'friendship': obj.child_value('friendship'),
}
return {
'medals': medals,
'items': items,
'characters': charas,
'lumina': {0: {'lumina': resp.child_value('player23/account/lumina')}},
}
else:
raise Exception(f'Unrecognized message type \'{msg_type}\'')
def verify_player23_read_score(self, ref_id: str) -> Dict[str, Dict[int, Dict[int, int]]]:
call = self.call_node()
# Construct node
player23 = Node.void('player23')
call.add_child(player23)
player23.set_attribute('method', 'read_score')
player23.add_child(Node.string('ref_id', ref_id))
player23.add_child(Node.s8('pref', 51))
# Swap with server
resp = self.exchange('', call)
# Verify defaults
self.assert_path(resp, "response/player23/@status")
# Grab scores
scores: Dict[int, Dict[int, int]] = {}
medals: Dict[int, Dict[int, int]] = {}
for child in resp.child('player23').children:
if child.name != 'music':
continue
musicid = child.child_value('music_num')
chart = child.child_value('sheet_num')
score = child.child_value('score')
medal = child.child_value('clear_type')
if musicid not in scores:
scores[musicid] = {}
if musicid not in medals:
medals[musicid] = {}
scores[musicid][chart] = score
medals[musicid][chart] = medal
return {
'scores': scores,
'medals': medals,
}
def verify_player23_start(self, ref_id: str, loc: str) -> None:
call = self.call_node()
# Construct node
player23 = Node.void('player23')
call.add_child(player23)
player23.set_attribute('loc_id', loc)
player23.set_attribute('ref_id', ref_id)
player23.set_attribute('method', 'start')
player23.set_attribute('start_type', '0')
pcb_card = Node.void('pcb_card')
player23.add_child(pcb_card)
pcb_card.add_child(Node.s8('card_enable', 0))
pcb_card.add_child(Node.s8('card_soldout', 0))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.__verify_common('player23', resp)
def verify_player23_logout(self, ref_id: str) -> None:
call = self.call_node()
# Construct node
player23 = Node.void('player23')
call.add_child(player23)
player23.set_attribute('ref_id', ref_id)
player23.set_attribute('method', 'logout')
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/player23/@status")
def verify_player23_write(
self,
ref_id: str,
medal: Optional[Dict[str, int]]=None,
item: Optional[Dict[str, int]]=None,
character: Optional[Dict[str, int]]=None,
) -> None:
call = self.call_node()
# Construct node
player23 = Node.void('player23')
call.add_child(player23)
player23.set_attribute('method', 'write')
player23.add_child(Node.string('ref_id', ref_id))
# Add required children
config = Node.void('config')
player23.add_child(config)
config.add_child(Node.s16('chara', 1543))
if medal is not None:
medalnode = Node.void('medal')
player23.add_child(medalnode)
medalnode.add_child(Node.s16('medal_id', medal['id']))
medalnode.add_child(Node.s16('level', medal['level']))
medalnode.add_child(Node.s32('exp', medal['exp']))
medalnode.add_child(Node.s32('set_count', 0))
medalnode.add_child(Node.s32('get_count', 0))
if item is not None:
itemnode = Node.void('item')
player23.add_child(itemnode)
itemnode.add_child(Node.u8('type', item['type']))
itemnode.add_child(Node.u16('id', item['id']))
itemnode.add_child(Node.u16('param', item['param']))
itemnode.add_child(Node.bool('is_new', False))
if character is not None:
chara_param = Node.void('chara_param')
player23.add_child(chara_param)
chara_param.add_child(Node.u16('chara_id', character['id']))
chara_param.add_child(Node.u16('friendship', character['friendship']))
# Swap with server
resp = self.exchange('', call)
self.assert_path(resp, "response/player23/@status")
def verify_player23_buy(self, ref_id: str, item: Dict[str, int]) -> None:
call = self.call_node()
# Construct node
player23 = Node.void('player23')
call.add_child(player23)
player23.set_attribute('method', 'buy')
player23.add_child(Node.s32('play_id', 0))
player23.add_child(Node.string('ref_id', ref_id))
player23.add_child(Node.u16('id', item['id']))
player23.add_child(Node.u8('type', item['type']))
player23.add_child(Node.u16('param', item['param']))
player23.add_child(Node.s32('lumina', item['lumina']))
player23.add_child(Node.u16('price', item['price']))
# Swap with server
resp = self.exchange('', call)
self.assert_path(resp, "response/player23/@status")
def verify_player23_write_music(self, ref_id: str, score: Dict[str, Any]) -> None:
call = self.call_node()
# Construct node
player23 = Node.void('player23')
call.add_child(player23)
player23.set_attribute('method', 'write_music')
player23.add_child(Node.string('ref_id', ref_id))
player23.add_child(Node.string('data_id', ref_id))
player23.add_child(Node.string('name', self.NAME))
player23.add_child(Node.u8('stage', 0))
player23.add_child(Node.s16('music_num', score['id']))
player23.add_child(Node.u8('sheet_num', score['chart']))
player23.add_child(Node.u8('clearmedal', score['medal']))
player23.add_child(Node.s32('score', score['score']))
player23.add_child(Node.s16('combo', 0))
player23.add_child(Node.s16('cool', 0))
player23.add_child(Node.s16('great', 0))
player23.add_child(Node.s16('good', 0))
player23.add_child(Node.s16('bad', 0))
# Swap with server
resp = self.exchange('', call)
self.assert_path(resp, "response/player23/@status")
def verify_player23_new(self, ref_id: str) -> None:
call = self.call_node()
# Construct node
player23 = Node.void('player23')
call.add_child(player23)
player23.set_attribute('method', 'new')
player23.add_child(Node.string('ref_id', ref_id))
player23.add_child(Node.string('name', self.NAME))
player23.add_child(Node.s8('pref', 51))
# Swap with server
resp = self.exchange('', call)
# Verify nodes
self.__verify_profile(resp)
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_pcb23_boot(location)
self.verify_info23_common(location)
self.verify_lobby22_getlist(location)
# 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')
self.verify_player23_read(ref_id, msg_type='new')
self.verify_player23_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 proper handling of basic stuff
self.verify_player23_read(ref_id, msg_type='query')
self.verify_player23_start(ref_id, location)
self.verify_player23_write(ref_id)
self.verify_player23_logout(ref_id)
# Verify unlocks/story mode work
unlocks = self.verify_player23_read(ref_id, msg_type='query')
for item in unlocks['items']:
raise Exception('Got nonzero items count on a new card!')
for med in unlocks['medals']:
raise Exception('Got nonzero medals count on a new card!')
for char in unlocks['characters']:
raise Exception('Got nonzero characters count on a new card!')
if unlocks['lumina'][0]['lumina'] != 300:
raise Exception('Got wrong default value for lumina on a new card!')
self.verify_player23_write(ref_id, medal={'id': 1, 'level': 3, 'exp': 42})
unlocks = self.verify_player23_read(ref_id, msg_type='query')
if 1 not in unlocks['medals']:
raise Exception('Expecting to see medal ID 1 in medals!')
if unlocks['medals'][1]['level'] != 3:
raise Exception('Expecting to see medal ID 1 to have level 3 in medals!')
if unlocks['medals'][1]['exp'] != 42:
raise Exception('Expecting to see medal ID 1 to have exp 42 in medals!')
self.verify_player23_write(ref_id, item={'id': 4, 'type': 2, 'param': 69})
unlocks = self.verify_player23_read(ref_id, msg_type='query')
if 4 not in unlocks['items']:
raise Exception('Expecting to see item ID 4 in items!')
if unlocks['items'][4]['type'] != 2:
raise Exception('Expecting to see item ID 4 to have type 2 in items!')
if unlocks['items'][4]['param'] != 69:
raise Exception('Expecting to see item ID 4 to have param 69 in items!')
self.verify_player23_write(ref_id, character={'id': 5, 'friendship': 420})
unlocks = self.verify_player23_read(ref_id, msg_type='query')
if 5 not in unlocks['characters']:
raise Exception('Expecting to see chara ID 5 in characters!')
if unlocks['characters'][5]['friendship'] != 420:
raise Exception('Expecting to see chara ID 5 to have type 2 in characters!')
# Verify purchases work
self.verify_player23_buy(ref_id, item={'id': 6, 'type': 7, 'param': 8, 'lumina': 400, 'price': 250})
unlocks = self.verify_player23_read(ref_id, msg_type='query')
if 6 not in unlocks['items']:
raise Exception('Expecting to see item ID 6 in items!')
if unlocks['items'][6]['type'] != 7:
raise Exception('Expecting to see item ID 6 to have type 7 in items!')
if unlocks['items'][6]['param'] != 8:
raise Exception('Expecting to see item ID 6 to have param 8 in items!')
if unlocks['lumina'][0]['lumina'] != 150:
raise Exception(f'Got wrong value for lumina {unlocks["lumina"][0]["lumina"]} after purchase!')
if cardid is None:
# Verify score handling
scores = self.verify_player23_read_score(ref_id)
for medal in scores['medals']:
raise Exception('Got nonzero medals count on a new card!')
for score in scores['scores']:
raise Exception('Got nonzero scores count on a new card!')
for phase in [1, 2]:
if phase == 1:
dummyscores = [
# An okay score on a chart
{
'id': 987,
'chart': 2,
'medal': 5,
'score': 76543,
},
# A good score on an easier chart of the same song
{
'id': 987,
'chart': 0,
'medal': 6,
'score': 99999,
},
# A bad score on a hard chart
{
'id': 741,
'chart': 3,
'medal': 2,
'score': 45000,
},
# A terrible score on an easy chart
{
'id': 742,
'chart': 1,
'medal': 2,
'score': 1,
},
]
# Random score to add in
songid = random.randint(907, 950)
chartid = random.randint(0, 3)
score = random.randint(0, 100000)
medal = random.randint(1, 11)
dummyscores.append({
'id': songid,
'chart': chartid,
'medal': medal,
'score': score,
})
if phase == 2:
dummyscores = [
# A better score on the same chart
{
'id': 987,
'chart': 2,
'medal': 5,
'score': 98765,
},
# A worse score on another same chart
{
'id': 987,
'chart': 0,
'medal': 3,
'score': 12345,
'expected_score': 99999,
'expected_medal': 6,
},
]
for dummyscore in dummyscores:
self.verify_player23_write_music(ref_id, dummyscore)
scores = self.verify_player23_read_score(ref_id)
for expected in dummyscores:
newscore = scores['scores'][expected['id']][expected['chart']]
newmedal = scores['medals'][expected['id']][expected['chart']]
if 'expected_score' in expected:
expected_score = expected['expected_score']
else:
expected_score = expected['score']
if 'expected_medal' in expected:
expected_medal = expected['expected_medal']
else:
expected_medal = expected['medal']
if newscore != expected_score:
raise Exception(f'Expected a score of \'{expected_score}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got score \'{newscore}\'')
if newmedal != expected_medal:
raise Exception(f'Expected a medal of \'{expected_medal}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got medal \'{newmedal}\'')
# 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)