//
//  WKJavaScriptBridgePluginAnnotation.m
//  WKJavaScriptBridge-demo
//
//  Created by Kevin on 2019/10/22.
//  Copyright © 2019 王凯. All rights reserved.
//

#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;

@end

@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;
}

@end