#import "WKJavaScriptBridgePluginAnnotation.h"

#import "WKJavaScriptBridge.h"

#include <mach-o/getsect.h>

#include <mach-o/loader.h>

#include <mach-o/dyld.h>

#include <dlfcn.h>

#import <objc/runtime.h>

#import <objc/message.h>

#include <mach-o/ldsyms.h>

@interface WKJavaScriptBridgePluginAnnotation ()

@property (nonatomic, weak) WKJavaScriptBridge *bridge;

@implementation WKJavaScriptBridgePluginAnnotation

- (instancetype)initWithBridge:(WKJavaScriptBridge *)bridge {

if (self = [super init]) {

self.bridge = bridge;

}

return self;

}

- (void)getAllRegisterPluginName {

[self registedAnnotationModules];

}

- (void)registedAnnotationModules {

NSArray <NSString *>*whiteListPlugin = [WKJavaScriptBridgePluginAnnotation AnnotationWhiteListPlugin];

if (whiteListPlugin.count > 0) {

NSString *plugin = whiteListPlugin[0];

if (plugin.length) {

[self.bridge registerSecurityPlugin:plugin];

}

}

NSArray <NSString *>*securityInfo = [WKJavaScriptBridgePluginAnnotation AnnotationPlugins];

[self.bridge registerPluginSecurityInfomation:securityInfo];

}

+ (NSArray<NSString *> *)AnnotationPlugins {

static NSArray<NSString *> *securityInfo = nil;

static dispatch_once_t onceToken;

dispatch_once(&onceToken, ^{

securityInfo = BHReadConfiguration(SecurityConfig);

});

return securityInfo;

}

+ (NSArray<NSString *> *)AnnotationWhiteListPlugin {

static NSArray<NSString *> *plugin = nil;

static dispatch_once_t onceToken;

dispatch_once(&onceToken, ^{

plugin = BHReadConfiguration(SecurityPlugin);

});

return plugin;

}

static NSArray<NSString *>* BHReadConfiguration(char *section) {

NSMutableArray *configs = [NSMutableArray array];

Dl_info info;

dladdr(BHReadConfiguration, &info);

#ifndef __LP64__

const struct mach_header *mhp = (struct mach_header*)info.dli_fbase;

unsigned long size = 0;

uint32_t *memory = (uint32_t*)getsectiondata(mhp, "__DATA", section, & size);

#else /* defined(__LP64__) */

const struct mach_header_64 *mhp = (struct mach_header_64*)info.dli_fbase;

unsigned long size = 0;

uint64_t *memory = (uint64_t*)getsectiondata(mhp, "__DATA", section, & size);

#endif /* defined(__LP64__) */

for(int idx = 0; idx < size/sizeof(void*); ++idx){

char *string = (char*)memory[idx];

NSString *str = [NSString stringWithUTF8String:string];

if(!str)continue;

#ifdef DEBUG

NSLog(@"config = %@", str);

#endif

if(str) [configs addObject:str];

}

return configs;

}