406 lines
15 KiB
Python
406 lines
15 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
工作手机 SDK - 完整验证脚本 v9-fixed
|
||
使用exports_sync，正确调用所有接口
|
||
"""
|
||
|
||
import frida
|
||
import json
|
||
import time
|
||
import subprocess
|
||
import os
|
||
import sys
|
||
|
||
PHONE_IP = '192.168.110.80'
|
||
JS_FILE = '/Users/karuo/Documents/开发/2、私域银行/工作手机/sdk/agent/hook/wechat_api_v8_fix.js'
|
||
SCREENSHOT_DIR = '/Users/karuo/Documents/开发/2、私域银行/工作手机/开发文档/6、测试'
|
||
|
||
results = {}
|
||
errors = []
|
||
passed_count = 0
|
||
total_count = 0
|
||
|
||
def screenshot(name):
|
||
path = f'{SCREENSHOT_DIR}/v9fix_{name}.png'
|
||
try:
|
||
subprocess.run(['adb', '-s', f'{PHONE_IP}:5555', 'shell',
|
||
f'screencap -p /sdcard/v9fix_{name}.png'], timeout=5, capture_output=True)
|
||
subprocess.run(['adb', '-s', f'{PHONE_IP}:5555', 'pull',
|
||
f'/sdcard/v9fix_{name}.png', path], timeout=10, capture_output=True)
|
||
print(f" 📸 截图已保存: v9fix_{name}.png")
|
||
except Exception as e:
|
||
print(f" ⚠️ 截图失败: {e}")
|
||
|
||
def wake_phone():
|
||
subprocess.run(['adb', '-s', f'{PHONE_IP}:5555', 'shell', 'input keyevent 26'], timeout=5, capture_output=True)
|
||
time.sleep(0.5)
|
||
subprocess.run(['adb', '-s', f'{PHONE_IP}:5555', 'shell', 'input swipe 540 1800 540 900'], timeout=5, capture_output=True)
|
||
time.sleep(0.5)
|
||
|
||
def get_wechat_pid():
|
||
result = subprocess.run(
|
||
['adb', '-s', f'{PHONE_IP}:5555', 'shell', 'ps -A'],
|
||
capture_output=True, text=True, timeout=5
|
||
)
|
||
for line in result.stdout.split('\n'):
|
||
if 'com.tencent.mm' in line and ':push' not in line and ':sandbox' not in line and ':tools' not in line and ':appbrand' not in line:
|
||
parts = line.split()
|
||
if len(parts) >= 2:
|
||
return int(parts[1])
|
||
return None
|
||
|
||
def test(name, func):
|
||
global passed_count, total_count
|
||
total_count += 1
|
||
try:
|
||
r = func()
|
||
passed_count += 1
|
||
return r
|
||
except Exception as e:
|
||
errors.append(f"{name}: {e}")
|
||
print(f" ❌ {name}: {e}")
|
||
return None
|
||
|
||
def main():
|
||
global passed_count, total_count
|
||
|
||
print("=" * 70)
|
||
print("🔥 工作手机 SDK 完整验证 v9-fixed")
|
||
print(f"时间: {time.strftime('%Y-%m-%d %H:%M:%S')}")
|
||
print("=" * 70)
|
||
|
||
# 唤醒手机
|
||
wake_phone()
|
||
|
||
# 确保ADB端口转发
|
||
subprocess.run(['adb', '-s', f'{PHONE_IP}:5555', 'forward', 'tcp:27042', 'tcp:27042'],
|
||
timeout=5, capture_output=True)
|
||
|
||
# 获取微信PID
|
||
pid = get_wechat_pid()
|
||
if not pid:
|
||
print("❌ 微信未运行，启动微信...")
|
||
subprocess.run(['adb', '-s', f'{PHONE_IP}:5555', 'shell',
|
||
'monkey -p com.tencent.mm -c android.intent.category.LAUNCHER 1'],
|
||
timeout=10, capture_output=True)
|
||
time.sleep(5)
|
||
pid = get_wechat_pid()
|
||
|
||
if not pid:
|
||
print("❌ 无法启动微信")
|
||
return 0, 0
|
||
|
||
print(f"✅ 微信PID: {pid}")
|
||
|
||
# 连接Frida
|
||
print("\n📡 连接Frida...")
|
||
dm = frida.get_device_manager()
|
||
device = dm.add_remote_device('localhost:27042')
|
||
session = device.attach(pid)
|
||
|
||
with open(JS_FILE, 'r', encoding='utf-8') as f:
|
||
js_code = f.read()
|
||
|
||
script = session.create_script(js_code)
|
||
script.load()
|
||
time.sleep(2)
|
||
|
||
# 使用exports_sync
|
||
ex = script.exports_sync
|
||
|
||
# 初始化DB
|
||
try:
|
||
r = ex.initDbs()
|
||
print(f"✅ DB初始化: {json.dumps(r, ensure_ascii=False)[:200]}")
|
||
except Exception as e:
|
||
print(f"⚠️ DB初始化: {e}")
|
||
|
||
time.sleep(1)
|
||
|
||
print("\n" + "=" * 70)
|
||
print("📊 开始全量接口验证（26项）")
|
||
print("=" * 70)
|
||
|
||
# ===== 1. ping =====
|
||
print("\n【1】系统ping")
|
||
r = test("ping", lambda: ex.ping())
|
||
if r: print(f" ✅ {r}")
|
||
results['ping'] = r
|
||
|
||
# ===== 2. 账号信息 =====
|
||
print("\n【2】账号信息")
|
||
r = test("account", lambda: ex.getAccountInfo())
|
||
if r:
|
||
print(f" ✅ 账号: {r.get('nickname', '?')} | wxid: {r.get('wxid', '?')}")
|
||
print(f" 手机: {r.get('phone', '?')} | 地区: {r.get('region', '?')}")
|
||
results['account'] = r
|
||
|
||
# ===== 3. 全量统计 =====
|
||
print("\n【3】全量数据统计")
|
||
r = test("stats", lambda: ex.getAllStats())
|
||
if r:
|
||
print(f" ✅ 好友: {r.get('friends', 0)} | 群组: {r.get('groups', 0)} | 消息: {r.get('messages', 0)}")
|
||
print(f" 图片: {r.get('images', 0)} | 视频: {r.get('videos', 0)} | 文件: {r.get('files', 0)}")
|
||
results['stats'] = r
|
||
|
||
# ===== 4. 联系人列表 =====
|
||
print("\n【4】联系人列表（前20个）")
|
||
r = test("contacts", lambda: ex.getContacts(20, 0))
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 联系人: {len(rows)}个")
|
||
for c in rows[:3]:
|
||
print(f" - {c.get('nickName', '?')} ({c.get('username', '?')})")
|
||
results['contacts'] = {'count': len(rows), 'sample': rows[:3]}
|
||
|
||
# ===== 5. 联系人数量 =====
|
||
print("\n【5】联系人数量")
|
||
r = test("contact_count", lambda: ex.getContactCount())
|
||
if r:
|
||
print(f" ✅ 好友: {r.get('friends', 0)} | 总计: {r.get('total', 0)}")
|
||
results['contact_count'] = r
|
||
|
||
# ===== 6. 群组列表 =====
|
||
print("\n【6】群组列表")
|
||
r = test("groups", lambda: ex.getGroups(50))
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 群组: {len(rows)}个")
|
||
for g in rows[:3]:
|
||
print(f" - {g.get('nickName', g.get('NickName', '?'))} ({g.get('username', '?')})")
|
||
results['groups'] = {'count': len(rows), 'sample': rows[:3]}
|
||
|
||
# ===== 7. 标签 =====
|
||
print("\n【7】标签列表")
|
||
r = test("labels", lambda: ex.getLabels())
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 标签: {len(rows)}个")
|
||
for l in rows[:5]:
|
||
print(f" - {l.get('labelName', l.get('name', '?'))} (id:{l.get('labelId', l.get('id', '?'))})")
|
||
results['labels'] = {'count': len(rows)}
|
||
|
||
# ===== 8. 最近消息 =====
|
||
print("\n【8】最近消息（24小时）")
|
||
r = test("recent_messages", lambda: ex.getRecentMessages(24))
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 消息: {len(rows)}条")
|
||
results['recent_messages'] = {'count': len(rows)}
|
||
|
||
# ===== 9. 发送消息 =====
|
||
print("\n【9】发送消息到文件传输助手")
|
||
msg_content = f'[工作手机SDK v9] 自动测试消息 {time.strftime("%H:%M:%S")}'
|
||
r = test("send_message", lambda: ex.sendMessage('filehelper', msg_content, 1))
|
||
if r:
|
||
print(f" ✅ 发送结果: {json.dumps(r, ensure_ascii=False)[:200]}")
|
||
results['send_message'] = r
|
||
time.sleep(1)
|
||
screenshot('after_send_msg')
|
||
|
||
# ===== 10. 群发消息 =====
|
||
print("\n【10】群发消息")
|
||
try:
|
||
contacts_r = ex.getContacts(5, 0)
|
||
rows = contacts_r.get('rows', [])
|
||
targets = [c.get('username', '') for c in rows if c.get('username', '') and not c.get('username', '').startswith('gh_')][:3]
|
||
if targets:
|
||
r = test("mass_message", lambda: ex.massMessage(targets, f'[群发测试] {time.strftime("%H:%M:%S")}'))
|
||
if r:
|
||
print(f" ✅ 群发: 目标{len(targets)}人, 结果: {json.dumps(r, ensure_ascii=False)[:150]}")
|
||
results['mass_message'] = r
|
||
else:
|
||
print(f" ⚠️ 没有有效目标")
|
||
except Exception as e:
|
||
print(f" ❌ 群发: {e}")
|
||
errors.append(f"mass_message: {e}")
|
||
|
||
# ===== 11. 群成员 =====
|
||
print("\n【11】群成员（第一个群）")
|
||
try:
|
||
groups_r = ex.getGroups(1)
|
||
rows = groups_r.get('rows', [])
|
||
if rows:
|
||
chatroom_id = rows[0].get('username', '')
|
||
r = test("group_members", lambda: ex.getGroupMembers(chatroom_id))
|
||
if r:
|
||
members = r.get('rows', [])
|
||
print(f" ✅ 群: {chatroom_id} | 成员: {len(members)}人")
|
||
results['group_members'] = {'count': len(members), 'chatroom': chatroom_id}
|
||
except Exception as e:
|
||
print(f" ❌ 群成员: {e}")
|
||
errors.append(f"group_members: {e}")
|
||
|
||
# ===== 12. 朋友圈DB信息 =====
|
||
print("\n【12】朋友圈DB信息")
|
||
r = test("moments_db", lambda: ex.getMomentsDbInfo())
|
||
if r:
|
||
print(f" ✅ 朋友圈DB: {json.dumps(r, ensure_ascii=False)[:300]}")
|
||
results['moments_db'] = r
|
||
|
||
# ===== 13. 朋友圈列表 =====
|
||
print("\n【13】朋友圈列表")
|
||
r = test("moments", lambda: ex.getMoments(10))
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 朋友圈: {len(rows)}条")
|
||
for m in rows[:2]:
|
||
print(f" - {str(m)[:100]}")
|
||
results['moments'] = {'count': len(rows)}
|
||
|
||
# ===== 14. 发布朋友圈 =====
|
||
print("\n【14】发布朋友圈（写入本地DB）")
|
||
moment_content = f'[工作手机SDK测试] {time.strftime("%Y-%m-%d %H:%M:%S")} 自动发布'
|
||
r = test("post_moment", lambda: ex.postMoment(moment_content))
|
||
if r:
|
||
print(f" ✅ 发布结果: {json.dumps(r, ensure_ascii=False)[:200]}")
|
||
results['post_moment'] = r
|
||
time.sleep(1)
|
||
screenshot('after_post_moment')
|
||
|
||
# ===== 15. 视频号信息 =====
|
||
print("\n【15】视频号信息")
|
||
r = test("finder_info", lambda: ex.getFinderInfo())
|
||
if r:
|
||
print(f" ✅ 视频号: {json.dumps(r, ensure_ascii=False)[:300]}")
|
||
results['finder_info'] = r
|
||
|
||
# ===== 16. 视频号联系人 =====
|
||
print("\n【16】视频号联系人")
|
||
r = test("finder_contacts", lambda: ex.getFinderContacts(10))
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 视频号联系人: {len(rows)}个")
|
||
results['finder_contacts'] = {'count': len(rows)}
|
||
|
||
# ===== 17. 视频号账号 =====
|
||
print("\n【17】视频号账号")
|
||
r = test("finder_accounts", lambda: ex.getFinderAccounts())
|
||
if r:
|
||
print(f" ✅ 视频号账号: {json.dumps(r, ensure_ascii=False)[:200]}")
|
||
results['finder_accounts'] = r
|
||
|
||
# ===== 18. 收藏 =====
|
||
print("\n【18】收藏列表")
|
||
r = test("favorites", lambda: ex.getFavorites(10))
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 收藏: {len(rows)}条")
|
||
results['favorites'] = {'count': len(rows)}
|
||
|
||
# ===== 19. 钱包信息 =====
|
||
print("\n【19】钱包信息")
|
||
r = test("wallet_info", lambda: ex.getWalletInfo())
|
||
if r:
|
||
print(f" ✅ 钱包: {json.dumps(r, ensure_ascii=False)[:300]}")
|
||
results['wallet_info'] = r
|
||
|
||
# ===== 20. 钱包流水 =====
|
||
print("\n【20】钱包流水")
|
||
r = test("wallet_ledger", lambda: ex.getWalletLedger(10))
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 钱包流水: {len(rows)}条")
|
||
for l in rows[:2]:
|
||
print(f" - {str(l)[:100]}")
|
||
results['wallet_ledger'] = {'count': len(rows)}
|
||
|
||
# ===== 21. 红包记录 =====
|
||
print("\n【21】红包记录")
|
||
r = test("lucky_money", lambda: ex.getLuckyMoney(10))
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 红包: {len(rows)}条")
|
||
for l in rows[:2]:
|
||
print(f" - {str(l)[:100]}")
|
||
results['lucky_money'] = {'count': len(rows)}
|
||
|
||
# ===== 22. 图片统计 =====
|
||
print("\n【22】图片统计")
|
||
r = test("image_count", lambda: ex.getImageCount())
|
||
if r:
|
||
print(f" ✅ 图片: {json.dumps(r, ensure_ascii=False)}")
|
||
results['image_count'] = r
|
||
|
||
# ===== 23. 视频列表 =====
|
||
print("\n【23】视频列表")
|
||
r = test("videos", lambda: ex.getVideos(5))
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 视频: {len(rows)}条")
|
||
results['videos'] = {'count': len(rows)}
|
||
|
||
# ===== 24. 文件列表 =====
|
||
print("\n【24】文件列表")
|
||
r = test("files", lambda: ex.getFiles(5))
|
||
if r:
|
||
rows = r.get('rows', [])
|
||
print(f" ✅ 文件: {len(rows)}条")
|
||
results['files'] = {'count': len(rows)}
|
||
|
||
# ===== 25. 小程序 =====
|
||
print("\n【25】小程序信息")
|
||
r = test("appbrand", lambda: ex.getAppBrandInfo())
|
||
if r:
|
||
print(f" ✅ 小程序: {json.dumps(r, ensure_ascii=False)[:200]}")
|
||
results['appbrand'] = r
|
||
|
||
# ===== 26. 原始SQL =====
|
||
print("\n【26】原始SQL查询")
|
||
r = test("raw_sql", lambda: ex.rawSql('main', 'SELECT type, COUNT(*) as cnt FROM rcontact GROUP BY type ORDER BY cnt DESC LIMIT 10'))
|
||
if r:
|
||
print(f" ✅ SQL结果: {json.dumps(r, ensure_ascii=False)[:300]}")
|
||
results['raw_sql'] = r
|
||
|
||
# ===== 最终截图 =====
|
||
print("\n📸 最终截图...")
|
||
screenshot('final_state')
|
||
|
||
# ===== 汇总 =====
|
||
print("\n" + "=" * 70)
|
||
print(f"📊 验证结果: {passed_count}/{total_count} 通过 ({passed_count/total_count*100:.0f}%)")
|
||
print("=" * 70)
|
||
|
||
if errors:
|
||
print("\n❌ 失败项:")
|
||
for e in errors:
|
||
print(f" - {e}")
|
||
|
||
print("\n✅ 真实数据汇总:")
|
||
acc = results.get('account', {})
|
||
stats = results.get('stats', {})
|
||
cc = results.get('contact_count', {})
|
||
print(f" 📱 账号: {acc.get('nickname', '?')} | wxid: {acc.get('wxid', '?')}")
|
||
print(f" 👥 好友: {cc.get('friends', stats.get('friends', '?'))} | 总联系人: {cc.get('total', '?')}")
|
||
print(f" 💬 群组: {results.get('groups', {}).get('count', stats.get('groups', '?'))}")
|
||
print(f" 🏷️ 标签: {results.get('labels', {}).get('count', '?')}")
|
||
print(f" 📨 24h消息: {results.get('recent_messages', {}).get('count', '?')}")
|
||
print(f" 🖼️ 图片: {stats.get('images', '?')}")
|
||
print(f" 🎬 视频: {stats.get('videos', '?')}")
|
||
print(f" 📁 文件: {stats.get('files', '?')}")
|
||
print(f" ⭐ 收藏: {results.get('favorites', {}).get('count', '?')}")
|
||
print(f" 💰 钱包流水: {results.get('wallet_ledger', {}).get('count', '?')}")
|
||
print(f" 🧧 红包: {results.get('lucky_money', {}).get('count', '?')}")
|
||
print(f" 📹 视频号联系人: {results.get('finder_contacts', {}).get('count', '?')}")
|
||
|
||
# 保存结果
|
||
result_file = f'{SCREENSHOT_DIR}/v9fix_verify_results.json'
|
||
with open(result_file, 'w', encoding='utf-8') as f:
|
||
json.dump({
|
||
'timestamp': int(time.time()),
|
||
'time': time.strftime('%Y-%m-%d %H:%M:%S'),
|
||
'passed': passed_count,
|
||
'total': total_count,
|
||
'pass_rate': f"{passed_count/total_count*100:.0f}%",
|
||
'results': results,
|
||
'errors': errors
|
||
}, f, ensure_ascii=False, indent=2)
|
||
print(f"\n💾 结果已保存: {result_file}")
|
||
|
||
return passed_count, total_count
|
||
|
||
if __name__ == '__main__':
|
||
passed, total = main()
|
||
print(f"\n🎯 最终: {passed}/{total} ({passed/total*100:.0f}%)")
|
||
sys.exit(0 if passed >= total * 0.8 else 1)
|