WKScriptMessage+LeakFix.m

#import <mach-o/dyld.h>
#import <objc/runtime.h>
#import <WebKit/WebKit.h>

@interface WKScriptMessage (WKScriptMessageLeakFix)
@end

@implementation WKScriptMessage (WKScriptMessageLeakFix)

+ (void)load {
  int32_t version = NSVersionOfRunTimeLibrary("WebKit");
  int32_t majorVersion = version >> 16;
  if (majorVersion > 600) return;

  Method fixedDealloc = class_getInstanceMethod(self, @selector(fixedDealloc));
  IMP fixedDeallocIMP = method_getImplementation(fixedDealloc);
  class_addMethod(self, @selector(dealloc), fixedDeallocIMP, method_getTypeEncoding(fixedDealloc));
}

- (void)fixedDealloc {
  [self.body release];
  [super dealloc];
}

@end