/*
 * Copyright (c) 2015 - present Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the BSD style license found in the
 * LICENSE file in the root directory of this source tree. An additional grant
 * of patent rights can be found in the PATENTS file in the same directory.
 */

#import <Foundation/Foundation.h>
#import <UIKit/UIKit.h>

BOOL ExampleSanitizer(NSURL* u, int f) {
  if (f)
    __set_untaint_attribute(u);
  return f;
}

@interface ExampleViewController : NSObject
- (void)loadURL:(NSURL*)URL trackingCodes:(NSArray*)trackingCodes;
@end

@implementation ExampleViewController
- (void)loadURL:(NSURL*)URL
  trackingCodes:(NSArray*)trackingCodes{
                    // Require untainted URL
                };
@end

@interface B : NSObject
- (void)another_url_pass:(NSURL*)u;
@end

@implementation B
- (void)another_url_pass:(NSURL*)u {
  ExampleViewController* vc = [[ExampleViewController alloc] init];
  [vc loadURL:u trackingCodes:nil];
}
@end

@interface A : NSObject
- (void)pass_url_arond:(NSURL*)u;
@end

@implementation A
- (void)pass_url_arond:(NSURL*)u {
  B* b = [[B alloc] init];
  [b another_url_pass:u];
}
@end

@interface ExampleDelegate : NSObject
- (BOOL)application:(UIApplication*)application
            openURL:(NSURL*)URL
  sourceApplication:(NSString*)sourceApplication
         annotation:(id)annotation;
@end

@implementation ExampleDelegate
- (BOOL)application:(UIApplication*)application
            openURL:(NSURL*)URL
  sourceApplication:(NSString*)sourceApplication
         annotation:(id)annotation {
  // Assume tainted URL;
  A* a = [[A alloc] init];
  if (!ExampleSanitizer(URL, 0)) {
    [a pass_url_arond:URL]; // report taint
  }
  if (!ExampleSanitizer(URL, 1)) {
    [a pass_url_arond:URL]; // No taint
  }
  return YES;
}
@end