for wechat 6.5.5

swigger · swigger · commit 154bfa450bad · 2017-02-27T12:38:51.000+08:00
参考了buginux的核心改动，节省调试工作。不过差异越来越大了。
to be tested...
diff --git a/Makefile b/Makefile
@@ -7,8 +7,10 @@ _THEOS_PLATFORM_DPKG_DEB_COMPRESSION=gzip
 include theos/makefiles/common.mk
 
 TWEAK_NAME = WeChatRedEnvelop
-WeChatRedEnvelop_FILES = Tweak.xm Debug.mm info.cpp
+WeChatRedEnvelop_FILES = Tweak.xm Debug.mm info.cpp xpruntime.c
 WeChatRedEnvelop_FRAMEWORKS = UIKit AudioToolbox
+WeChatRedEnvelop_CPPFLAGS=-std=c++11
+#CCFLAGS=-std=c++11
 
 include $(THEOS_MAKE_PATH)/tweak.mk
 
diff --git a/Tweak.xm b/Tweak.xm
@@ -5,6 +5,27 @@
 
 #define PL_PATH "/private/var/mobile/Library/Preferences/com.tencent.wcreopt.plist"
 
+
+static void * oc_rq = 0;
+
+static void ocrq_set(NSString* skey, NSObject* svalue)
+{
+	CFTypeRef cobj = (__bridge CFTypeRef)svalue;
+	if (cobj) CFRetain(cobj);
+	if (!oc_rq) oc_rq = rq_create();
+	rq_set(oc_rq, [skey UTF8String], (void*)cobj);
+}
+
+static Boolean ocrq_get(NSString* skey, NSObject** svalue)
+{
+	void * cobj = 0;
+	if (!skey || !svalue || !oc_rq) return FALSE;
+	rq_get(oc_rq, [skey UTF8String], &cobj);
+	if (!cobj) return FALSE;
+	*svalue = (__bridge NSObject*)cobj;
+	return TRUE;
+}
+
 NSMutableDictionary * loadSettings()
 {
 	static int64_t timeid = -1;
@@ -22,6 +43,42 @@ NSMutableDictionary * loadSettings()
 	return oldinfo;
 }
 
+
+%hook WCRedEnvelopesLogicMgr
+- (void)OnWCToHongbaoCommonResponse:(HongBaoRes *)arg1 Request:(HongBaoReq *)arg2
+{
+	%orig;
+
+	// 非参数查询请求
+	if (arg1.cgiCmdid != 3) return;
+
+	NSString *string = [[NSString alloc] initWithData:arg1.retText.buffer encoding:NSUTF8StringEncoding];
+	NSDictionary *dictionary = [string JSONDictionary];
+
+	// 自己已经抢过
+	if ([dictionary[@"receiveStatus"] integerValue] == 2)
+		return;
+
+	// 红包被抢完
+	if ([dictionary[@"hbStatus"] integerValue] == 4)
+		return;
+
+	// 没有这个字段会被判定为使用外挂
+	if (!dictionary[@"timingIdentifier"])
+		return;
+
+	NSString* xxkey = @"";
+	NSMutableDictionary * par_open = 0;
+	ocrq_get(xxkey, &par_open);
+	if (par_open)
+	{
+		par_open[@"timingIdentifier"] = dictionary[@"timingIdentifier"];
+		WCRedEnvelopesLogicMgr *logicMgr = [[objc_getClass("MMServiceCenter") defaultCenter] getService:[objc_getClass("WCRedEnvelopesLogicMgr") class]];
+		[logicMgr OpenRedEnvelopesRequest:par_open];
+	}
+}
+%end
+
 %hook CMessageMgr
 - (void)AsyncOnAddMsg:(NSString *)msg MsgWrap:(CMessageWrap *)wrap {
 	%orig;
@@ -35,16 +92,11 @@ NSMutableDictionary * loadSettings()
 	CContact *selfContact = [contactManager getSelfContact];
 
 	if (msgType != 49) return;
-	BOOL isMesasgeFromMe = NO;
+
 	if ([wrap.m_nsFromUsr isEqualToString:selfContact.m_nsUsrName]) {
-		isMesasgeFromMe = YES;
-	}
-	if (isMesasgeFromMe)
-	{
 		// 自己发的红包，就不需要机器人来抢了吧？？
 		return;
 	}
-
 	if ([wrap.m_nsContent rangeOfString:@"wxpay://"].location == NSNotFound)
 		return; //不是红包？？
 	if ([wrap.m_nsFromUsr rangeOfString:@"@chatroom"].location == NSNotFound)
@@ -114,21 +166,31 @@ NSMutableDictionary * loadSettings()
 		NSString *nativeUrl = [[wrap m_oWCPayInfoItem] m_c2cNativeUrl];
 		nativeUrl = [nativeUrl substringFromIndex:[@"wxpay://c2cbizmessagehandler/hongbao/receivehongbao?" length]];
 		NSDictionary *nativeUrlDict = [%c(WCBizUtil) dictionaryWithDecodedComponets:nativeUrl separator:@"&"];
-
-		/** 构造参数 */
-		NSMutableDictionary *params = [@{} mutableCopy];
-		params[@"msgType"] = nativeUrlDict[@"msgtype"] ?: @"1";
-		params[@"sendId"] = nativeUrlDict[@"sendid"] ?: @"";
+		WCRedEnvelopesLogicMgr *logicMgr = [[objc_getClass("MMServiceCenter") defaultCenter] getService:[objc_getClass("WCRedEnvelopesLogicMgr") class]];
+		sl_printf("%s 打开红包！延时%lf秒", ssig, (double) delayTime);
+
+		CContactMgr *contactManager = [[objc_getClass("MMServiceCenter") defaultCenter] getService:[objc_getClass("CContactMgr") class]];
+		CContact *selfContact = [contactManager getSelfContact];
+		NSMutableDictionary *par_open = [[NSMutableDictionary alloc] init];
+		par_open[@"msgType"] = nativeUrlDict[@"msgtype"] ?: @"1";
+		par_open[@"sendId"] = nativeUrlDict[@"sendid"] ?: @"";
+		par_open[@"channelId"] = nativeUrlDict[@"channelid"] ?: @"1";
+		par_open[@"nickName"] = [selfContact getContactDisplayName] ?: @"小锅";
+		par_open[@"headImg"] = [selfContact m_nsHeadImgUrl] ?: @"";
+		par_open[@"nativeUrl"] = [[wrap m_oWCPayInfoItem] m_c2cNativeUrl] ?: @"";
+		par_open[@"sessionUserName"] = wrap.m_nsFromUsr ?: @"";
+
+		NSString* xxkey = @"";
+		ocrq_set(xxkey, par_open);
+
+		NSMutableDictionary *params = [[NSMutableDictionary alloc]init];
+		params[@"agreeDuty"] = @"0";
 		params[@"channelId"] = nativeUrlDict[@"channelid"] ?: @"1";
-		params[@"nickName"] = [selfContact getContactDisplayName] ?: @"小锅";
-		params[@"headImg"] = [selfContact m_nsHeadImgUrl] ?: @"";
+		params[@"inWay"] = @"0";
+		params[@"msgType"] = nativeUrlDict[@"msgtype"] ?: @"1";
 		params[@"nativeUrl"] = [[wrap m_oWCPayInfoItem] m_c2cNativeUrl] ?: @"";
-		params[@"sessionUserName"] = wrap.m_nsFromUsr ?: @"";
-
-		MMServiceCenter * center = (MMServiceCenter*)[objc_getClass("MMServiceCenter") defaultCenter];
-		WCRedEnvelopesLogicMgr *logicMgr = [center getService:[objc_getClass("WCRedEnvelopesLogicMgr") class]];
-		sl_printf("%s 打开红包！！延时%lf秒", ssig, (double) delayTime);
-		[logicMgr OpenRedEnvelopesRequest:params];
+		params[@"sendId"] = nativeUrlDict[@"sendid"] ?: @"";
+		[logicMgr ReceiverQueryRedEnvelopesRequest:params];
     });
 }
 %end
diff --git a/WeChatRedEnvelop.h b/WeChatRedEnvelop.h
@@ -20,6 +20,11 @@ struct set_;
 
 typedef void (^CDUnknownBlockType)(void); // return type and parameters are unknown
 
+@interface NSString (NSString_SBJSON)
+- (id)JSONDictionary;
+- (id)JSONArray;
+- (id)JSONValue;
+@end
 
 @interface NSError (SimpleConstructor)
 + (_Bool)clearError:(id *)arg1;
diff --git a/info.cpp b/info.cpp
@@ -8,9 +8,12 @@
 
 #include <mach-o/dyld.h>
 #include <string>
+#include <set>
+using std::set;
 
 #include "info.h"
 #include "Debug.h"
+#include "xpruntime.h"
 
 using std::string;
 
@@ -67,3 +70,107 @@ int dont_open(const char * content, const char * kwds)
 	}
 	return 0;
 }
+
+class CRecentQueue
+{
+	enum {TIMEOUT=60000};
+	struct RQV{
+		string skey;
+		void * obj;
+		int64_t tick;
+		int64_t seq;
+		RQV(const char * key):skey(key){}
+	};
+	set<RQV*> bkey, bid;
+	long m_seqidx;
+protected:
+	RQV* _find(set<RQV*> & which, const char * key)
+	{
+		RQV val(key);
+		auto it = which.find(&val);
+		return (it == which.end()) ? 0 : *it;
+	}
+	void clear_to()
+	{
+		int32_t tick = GetTickCount64b();
+		for (auto it = bid.begin(); it != bid.end(); )
+		{
+			RQV * w = *it;
+			if (w->tick + TIMEOUT < tick)
+			{
+				bkey.erase(w);
+				it = bid.erase(it++);
+			}
+			else
+				break;
+		}
+	}
+
+public:
+	CRecentQueue()
+	{
+	}
+	~CRecentQueue(){}
+public:
+	void get(const char * key, void ** v)
+	{
+		clear_to();
+		RQV * r1 = _find(bkey, key);
+		*v = r1 ? r1->obj : 0;
+	}
+	void set(const char * key, void * v)
+	{
+		clear_to();
+		RQV * r1 = _find(bkey, key);
+		if (r1)
+		{
+			if (v)
+			{
+				bid.erase(r1);
+				r1->obj = v;
+				r1->seq = ++m_seqidx;
+				r1->tick = GetTickCount64b();
+				bid.insert(r1);
+			}
+			else
+			{
+				bid.erase(r1);
+				bkey.erase(r1);
+				delete r1;
+			}
+		}
+		else if (v)
+		{
+			//new add.
+			RQV * prqv = new RQV(key);
+			prqv->obj = v;
+			prqv->seq = ++m_seqidx;
+			prqv->tick = GetTickCount64b();
+			bid.insert(prqv);
+			bkey.insert(prqv);
+		}
+	}
+};
+
+void * rq_create()
+{
+	return new CRecentQueue();
+}
+
+void rq_set(void * rq, const char * key, void * obj)
+{
+	CRecentQueue * rq_ = (CRecentQueue*)rq;
+	return rq_->set(key, obj);
+}
+
+void rq_get(void * rq, const char * key, void ** pobj)
+{
+	CRecentQueue * rq_ = (CRecentQueue*)rq;
+	return rq_->get(key, pobj);
+}
+
+void rq_delete(void * rq)
+{
+	CRecentQueue * rq_ = (CRecentQueue*)rq;
+	delete rq_;
+}
diff --git a/info.h b/info.h
@@ -10,6 +10,11 @@ extern "C"{
 	// 特定文字是否抢红包
 	int dont_open(const char * content, const char * kwds);
 
+	void * rq_create();
+	void rq_set(void * rq, const char * key, void * obj);
+	void rq_get(void * rq, const char * key, void ** pobj);
+	void rq_delete(void * rq);
+
 #ifdef __cplusplus
 }
 #endif
diff --git a/xpruntime.c b/xpruntime.c
@@ -0,0 +1,52 @@
+#include "xpruntime.h"
+#include <unistd.h>
+#include <assert.h>
+#include <sys/sysctl.h>
+#include <mach/clock.h>
+#include <mach/clock_types.h>
+#include <mach/mach_host.h>
+#include <mach/mach.h>
+#include <sys/socket.h>
+#include <arpa/inet.h>
+#include <netinet/in.h>
+#include <stdio.h>
+#include <sys/types.h>
+#include <netdb.h>
+#include <stdlib.h>
+
+static int get_clock_ref()
+{
+    clock_serv_t clock_ref = 0;
+    host_get_clock_service(mach_host_self(), SYSTEM_CLOCK, &clock_ref);
+    return clock_ref;
+}
+
+static int darwin_clock_gettime_MONOTONIC(struct timespec *tp)
+{
+    static int clockinit = 0;
+    static clock_serv_t clock_ref = 0;
+    mach_timespec_t mach_tp = {0};
+    
+    if (clockinit == 0)
+    {
+        clockinit = 1;
+        clock_ref = get_clock_ref();
+    }
+    
+    kern_return_t ret = clock_get_time(clock_ref, &mach_tp);
+    if (ret != KERN_SUCCESS) return -1;
+    
+    tp->tv_sec = mach_tp.tv_sec;
+    tp->tv_nsec = mach_tp.tv_nsec;
+    return 0;
+}
+
+int64_t GetTickCount64b()
+{
+    struct timespec ts = {0};
+    int r = darwin_clock_gettime_MONOTONIC(&ts);
+    if (r<0) return 1;
+    int64_t a = (int64_t)ts.tv_sec * 1000 + ts.tv_nsec/1000000;
+    return a;
+}
+
diff --git a/xpruntime.h b/xpruntime.h
@@ -0,0 +1,10 @@
+#pragma once
+#include <stdint.h>
+
+#ifdef __cplusplus
+#define extc extern "C"
+#else
+#define extc
+#endif
+
+extc int64_t GetTickCount64b();
