@devicelab/maestro-runner-darwin-x64 1.1.25

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Utilities/FBMjpegServer.h

@@ -0,0 +1,24 @@
+/**
+ * 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.
+ */
+
+#import "FBTCPSocket.h"
+
+NS_ASSUME_NONNULL_BEGIN
+
+@interface FBMjpegServer : NSObject <FBTCPSocketDelegate>
+
+/**
+ The default constructor for the screenshot bradcaster service.
+ This service sends low resolution screenshots 10 times per seconds
+ to all connected clients.
+ */
+- (instancetype)init;
+
+@end
+
+NS_ASSUME_NONNULL_END

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Utilities/FBMjpegServer.m

@@ -0,0 +1,151 @@
+/**
+ * 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.
+ */
+
+#import "FBMjpegServer.h"
+
+#import <mach/mach_time.h>
+@import UniformTypeIdentifiers;
+
+#import "GCDAsyncSocket.h"
+#import "FBConfiguration.h"
+#import "FBLogger.h"
+#import "FBScreenshot.h"
+#import "FBImageProcessor.h"
+#import "FBImageUtils.h"
+#import "XCUIScreen.h"
+
+static const NSUInteger MAX_FPS = 60;
+static const NSTimeInterval FRAME_TIMEOUT = 1.;
+
+static NSString *const SERVER_NAME = @"WDA MJPEG Server";
+static const char *QUEUE_NAME = "JPEG Screenshots Provider Queue";
+
+
+@interface FBMjpegServer()
+
+@property (nonatomic, readonly) dispatch_queue_t backgroundQueue;
+@property (nonatomic, readonly) NSMutableArray<GCDAsyncSocket *> *listeningClients;
+@property (nonatomic, readonly) FBImageProcessor *imageProcessor;
+@property (nonatomic, readonly) long long mainScreenID;
+
+@end
+
+
+@implementation FBMjpegServer
+
+- (instancetype)init
+{
+  if ((self = [super init])) {
+    _listeningClients = [NSMutableArray array];
+    dispatch_queue_attr_t queueAttributes = dispatch_queue_attr_make_with_qos_class(DISPATCH_QUEUE_SERIAL, QOS_CLASS_UTILITY, 0);
+    _backgroundQueue = dispatch_queue_create(QUEUE_NAME, queueAttributes);
+    dispatch_async(_backgroundQueue, ^{
+      [self streamScreenshot];
+    });
+    _imageProcessor = [[FBImageProcessor alloc] init];
+    _mainScreenID = [XCUIScreen.mainScreen displayID];
+  }
+  return self;
+}
+
+- (void)scheduleNextScreenshotWithInterval:(uint64_t)timerInterval timeStarted:(uint64_t)timeStarted
+{
+  uint64_t timeElapsed = clock_gettime_nsec_np(CLOCK_MONOTONIC_RAW) - timeStarted;
+  int64_t nextTickDelta = timerInterval - timeElapsed;
+  if (nextTickDelta > 0) {
+    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, nextTickDelta), self.backgroundQueue, ^{
+      [self streamScreenshot];
+    });
+  } else {
+    // Try to do our best to keep the FPS at a decent level
+    dispatch_async(self.backgroundQueue, ^{
+      [self streamScreenshot];
+    });
+  }
+}
+
+- (void)streamScreenshot
+{
+  NSUInteger framerate = FBConfiguration.mjpegServerFramerate;
+  uint64_t timerInterval = (uint64_t)(1.0 / ((0 == framerate || framerate > MAX_FPS) ? MAX_FPS : framerate) * NSEC_PER_SEC);
+  uint64_t timeStarted = clock_gettime_nsec_np(CLOCK_MONOTONIC_RAW);
+  @synchronized (self.listeningClients) {
+    if (0 == self.listeningClients.count) {
+      [self scheduleNextScreenshotWithInterval:timerInterval timeStarted:timeStarted];
+      return;
+    }
+  }
+
+  NSError *error;
+  CGFloat compressionQuality = MAX(FBMinCompressionQuality,
+                                   MIN(FBMaxCompressionQuality, FBConfiguration.mjpegServerScreenshotQuality / 100.0));
+  NSData *screenshotData = [FBScreenshot takeInOriginalResolutionWithScreenID:self.mainScreenID
+                                                           compressionQuality:compressionQuality
+                                                                          uti:UTTypeJPEG
+                                                                      timeout:FRAME_TIMEOUT
+                                                                        error:&error];
+  if (nil == screenshotData) {
+    [FBLogger logFmt:@"%@", error.description];
+    [self scheduleNextScreenshotWithInterval:timerInterval timeStarted:timeStarted];
+    return;
+  }
+
+  CGFloat scalingFactor = FBConfiguration.mjpegScalingFactor / 100.0;
+  [self.imageProcessor submitImageData:screenshotData
+                         scalingFactor:scalingFactor
+                     completionHandler:^(NSData * _Nonnull scaled) {
+    [self sendScreenshot:scaled];
+  }];
+
+  [self scheduleNextScreenshotWithInterval:timerInterval timeStarted:timeStarted];
+}
+
+- (void)sendScreenshot:(NSData *)screenshotData {
+  NSString *chunkHeader = [NSString stringWithFormat:@"--BoundaryString\r\nContent-type: image/jpeg\r\nContent-Length: %@\r\n\r\n", @(screenshotData.length)];
+  NSMutableData *chunk = [[chunkHeader dataUsingEncoding:NSUTF8StringEncoding] mutableCopy];
+  [chunk appendData:screenshotData];
+  [chunk appendData:(id)[@"\r\n\r\n" dataUsingEncoding:NSUTF8StringEncoding]];
+  @synchronized (self.listeningClients) {
+    for (GCDAsyncSocket *client in self.listeningClients) {
+      [client writeData:chunk withTimeout:-1 tag:0];
+    }
+  }
+}
+
+- (void)didClientConnect:(GCDAsyncSocket *)newClient
+{
+  [FBLogger logFmt:@"Got screenshots broadcast client connection at %@:%d", newClient.connectedHost, newClient.connectedPort];
+  // Start broadcast only after there is any data from the client
+  [newClient readDataWithTimeout:-1 tag:0];
+}
+
+- (void)didClientSendData:(GCDAsyncSocket *)client
+{
+  @synchronized (self.listeningClients) {
+    if ([self.listeningClients containsObject:client]) {
+      return;
+    }
+  }
+
+  [FBLogger logFmt:@"Starting screenshots broadcast for the client at %@:%d", client.connectedHost, client.connectedPort];
+  NSString *streamHeader = [NSString stringWithFormat:@"HTTP/1.0 200 OK\r\nServer: %@\r\nConnection: close\r\nMax-Age: 0\r\nExpires: 0\r\nCache-Control: no-cache, private\r\nPragma: no-cache\r\nContent-Type: multipart/x-mixed-replace; boundary=--BoundaryString\r\n\r\n", SERVER_NAME];
+  [client writeData:(id)[streamHeader dataUsingEncoding:NSUTF8StringEncoding] withTimeout:-1 tag:0];
+  @synchronized (self.listeningClients) {
+    [self.listeningClients addObject:client];
+  }
+}
+
+- (void)didClientDisconnect:(GCDAsyncSocket *)client
+{
+  @synchronized (self.listeningClients) {
+    [self.listeningClients removeObject:client];
+  }
+  [FBLogger log:@"Disconnected a client from screenshots broadcast"];
+}
+
+@end

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Utilities/FBNotificationsHelper.h

@@ -0,0 +1,37 @@
+/**
+ * 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.
+ */
+
+#import <XCTest/XCTest.h>
+
+NS_ASSUME_NONNULL_BEGIN
+
+@interface FBNotificationsHelper : NSObject
+
+/**
+ Creates an expectation that is fulfilled when an expected NSNotification is received
+
+ @param name The name of the awaited notification
+ @param timeout The maximum amount of float seconds to wait for the expectation
+ @return The appropriate waiter result
+ */
++ (XCTWaiterResult)waitForNotificationWithName:(NSNotificationName)name
+                                       timeout:(NSTimeInterval)timeout;
+
+/**
+ Creates an expectation that is fulfilled when an expected Darwin notification is received
+
+ @param name The name of the awaited notification
+ @param timeout The maximum amount of float seconds to wait for the expectation
+ @return The appropriate waiter result
+ */
++ (XCTWaiterResult)waitForDarwinNotificationWithName:(NSString *)name
+                                             timeout:(NSTimeInterval)timeout;
+
+@end
+
+NS_ASSUME_NONNULL_END

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Utilities/FBNotificationsHelper.m

@@ -0,0 +1,29 @@
+/**
+ * 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.
+ */
+
+#import "FBNotificationsHelper.h"
+
+@implementation FBNotificationsHelper
+
++ (XCTWaiterResult)waitForNotificationWithName:(NSNotificationName)name
+                                       timeout:(NSTimeInterval)timeout
+{
+  XCTNSNotificationExpectation *expectation = [[XCTNSNotificationExpectation alloc]
+                                               initWithName:name];
+  return [XCTWaiter waitForExpectations:@[expectation] timeout:timeout];
+}
+
++ (XCTWaiterResult)waitForDarwinNotificationWithName:(NSString *)name
+                                             timeout:(NSTimeInterval)timeout
+{
+  XCTDarwinNotificationExpectation *expectation = [[XCTDarwinNotificationExpectation alloc]
+                                                   initWithNotificationName:name];
+  return [XCTWaiter waitForExpectations:@[expectation] timeout:timeout];
+}
+
+@end

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Utilities/FBPasteboard.h

@@ -0,0 +1,40 @@
+/**
+ * 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.
+ */
+
+#import <XCTest/XCTest.h>
+
+NS_ASSUME_NONNULL_BEGIN
+
+#if !TARGET_OS_TV
+@interface FBPasteboard : NSObject
+
+/**
+ Sets data to the general pasteboard
+
+ @param data base64-encoded string containing the data chunk which is going to be written to the pasteboard
+ @param type one of the possible data types to set: plaintext, url, image
+ @param error If there is an error, upon return contains an NSError object that describes the problem
+ @return YES if the operation was successful
+ */
++ (BOOL)setData:(NSData *)data forType:(NSString *)type error:(NSError **)error;
+
+/**
+ Gets the data contained in the general pasteboard
+
+ @param type one of the possible data types to get: plaintext, url, image
+ @param error If there is an error, upon return contains an NSError object that describes the problem
+ @return NSData object, containing the pasteboard content or an empty string if the pasteboard is empty.
+ nil is returned if there was an error while getting the data from the pasteboard
+ */
++ (nullable NSData *)dataForType:(NSString *)type error:(NSError **)error;
+
+@end
+#endif
+
+NS_ASSUME_NONNULL_END
+

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Utilities/FBPasteboard.m

@@ -0,0 +1,172 @@
+/**
+ * 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.
+ */
+
+#import "FBPasteboard.h"
+
+#import <mach/mach_time.h>
+#import "FBAlert.h"
+#import "FBErrorBuilder.h"
+#import "FBMacros.h"
+#import "XCUIApplication+FBHelpers.h"
+#import "XCUIApplication+FBAlert.h"
+
+#define ALERT_TIMEOUT_SEC 30
+// Must not be less than FB_MONTORING_INTERVAL in FBAlertsMonitor
+#define ALERT_CHECK_INTERVAL_SEC 2
+
+#if !TARGET_OS_TV
+@implementation FBPasteboard
+
++ (BOOL)setData:(NSData *)data forType:(NSString *)type error:(NSError **)error
+{
+  UIPasteboard *pb = UIPasteboard.generalPasteboard;
+  if ([type.lowercaseString isEqualToString:@"plaintext"]) {
+    pb.string = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
+  } else if ([type.lowercaseString isEqualToString:@"image"]) {
+    UIImage *image = [UIImage imageWithData:data];
+    if (nil == image) {
+      NSString *description = @"No image can be parsed from the given pasteboard data";
+      if (error) {
+        *error = [[FBErrorBuilder.builder withDescription:description] build];
+      }
+      return NO;
+    }
+    pb.image = image;
+  } else if ([type.lowercaseString isEqualToString:@"url"]) {
+    NSString *urlString = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
+    NSURL *url = [[NSURL alloc] initWithString:urlString];
+    if (nil == url) {
+      NSString *description = @"No URL can be parsed from the given pasteboard data";
+      if (error) {
+        *error = [[FBErrorBuilder.builder withDescription:description] build];
+      }
+      return NO;
+    }
+    pb.URL = url;
+  } else {
+    NSString *description = [NSString stringWithFormat:@"Unsupported content type: %@", type];
+    if (error) {
+      *error = [[FBErrorBuilder.builder withDescription:description] build];
+    }
+    return NO;
+  }
+  return YES;
+}
+
++ (nullable id)pasteboardContentForItem:(NSString *)item
+                               instance:(UIPasteboard *)pbInstance
+                                timeout:(NSTimeInterval)timeout
+                                  error:(NSError **)error
+{
+  SEL selector = NSSelectorFromString(item);
+  NSMethodSignature *methodSignature = [pbInstance methodSignatureForSelector:selector];
+  if (nil == methodSignature) {
+    NSString *description = [NSString stringWithFormat:@"Cannot retrieve '%@' from a UIPasteboard instance", item];
+    if (error) {
+      *error = [[FBErrorBuilder.builder withDescription:description] build];
+    }
+    return nil;
+  }
+  NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:methodSignature];
+  [invocation setSelector:selector];
+  [invocation setTarget:pbInstance];
+  if (SYSTEM_VERSION_LESS_THAN(@"16.0")) {
+    [invocation invoke];
+    id __unsafe_unretained result;
+    [invocation getReturnValue:&result];
+    return result;
+  }
+
+  // https://github.com/appium/appium/issues/17392
+  __block id pasteboardContent;
+  dispatch_queue_t backgroundQueue = dispatch_queue_create("GetPasteboard", NULL);
+  __block BOOL didFinishGetPasteboard = NO;
+  dispatch_async(backgroundQueue, ^{
+    [invocation invoke];
+    id __unsafe_unretained result;
+    [invocation getReturnValue:&result];
+    pasteboardContent = result;
+    didFinishGetPasteboard = YES;
+  });
+  uint64_t timeStarted = clock_gettime_nsec_np(CLOCK_MONOTONIC_RAW);
+  while (!didFinishGetPasteboard) {
+    [[NSRunLoop currentRunLoop] runUntilDate:[NSDate dateWithTimeIntervalSinceNow:ALERT_CHECK_INTERVAL_SEC]];
+    if (didFinishGetPasteboard) {
+      break;
+    }
+
+    XCUIElement *alertElement = XCUIApplication.fb_systemApplication.fb_alertElement;
+    if (nil != alertElement) {
+      FBAlert *alert = [FBAlert alertWithElement:alertElement];
+      [alert acceptWithError:nil];
+    }
+    uint64_t timeElapsed = clock_gettime_nsec_np(CLOCK_MONOTONIC_RAW) - timeStarted;
+    if (timeElapsed / NSEC_PER_SEC > timeout) {
+      NSString *description = [NSString stringWithFormat:@"Cannot handle pasteboard alert within %@s timeout", @(timeout)];
+      if (error) {
+        *error = [[FBErrorBuilder.builder withDescription:description] build];
+      }
+      return nil;
+    }
+  }
+  return pasteboardContent;
+}
+
++ (NSData *)dataForType:(NSString *)type error:(NSError **)error
+{
+  UIPasteboard *pb = UIPasteboard.generalPasteboard;
+  if ([type.lowercaseString isEqualToString:@"plaintext"]) {
+    if (pb.hasStrings) {
+      id result = [self.class pasteboardContentForItem:@"strings"
+                                              instance:pb
+                                               timeout:ALERT_TIMEOUT_SEC
+                                                 error:error
+      ];
+      return nil == result
+        ? nil
+        : [[(NSArray *)result componentsJoinedByString:@"\n"] dataUsingEncoding:NSUTF8StringEncoding];
+    }
+  } else if ([type.lowercaseString isEqualToString:@"image"]) {
+    if (pb.hasImages) {
+      id result = [self.class pasteboardContentForItem:@"image"
+                                              instance:pb
+                                               timeout:ALERT_TIMEOUT_SEC
+                                                 error:error
+      ];
+      return nil == result ? nil : UIImagePNGRepresentation((UIImage *)result);
+    }
+  } else if ([type.lowercaseString isEqualToString:@"url"]) {
+    if (pb.hasURLs) {
+      id result = [self.class pasteboardContentForItem:@"URLs"
+                                              instance:pb
+                                               timeout:ALERT_TIMEOUT_SEC
+                                                 error:error
+      ];
+      if (nil == result) {
+        return nil;
+      }
+      NSMutableArray<NSString *> *urls = [NSMutableArray array];
+      for (NSURL *url in (NSArray *)result) {
+        if (nil != url.absoluteString) {
+          [urls addObject:(id)url.absoluteString];
+        }
+      }
+      return [[urls componentsJoinedByString:@"\n"] dataUsingEncoding:NSUTF8StringEncoding];
+    }
+  } else {
+    NSString *description = [NSString stringWithFormat:@"Unsupported content type: %@", type];
+    if (error) {
+      *error = [[FBErrorBuilder.builder withDescription:description] build];
+    }
+    return nil;
+  }
+  return [@"" dataUsingEncoding:NSUTF8StringEncoding];
+}
+
+@end
+#endif

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Utilities/FBProtocolHelpers.h

@@ -0,0 +1,46 @@
+/**
+ * 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.
+ */
+
+#import <Foundation/Foundation.h>
+
+NS_ASSUME_NONNULL_BEGIN
+
+/**
+ Prepares an element dictionary, which could be then used in hybrid W3C/JWP responses
+
+ @param element Either element identifier or element object itself
+ @returns The resulting dictionary
+ */
+NSDictionary<NSString *, id> *FBToElementDict(id element);
+
+/**
+ Extracts element uuid from dictionary
+
+ @param src The source dictionary
+ @returns The resulting element or nil if no element keys are found
+ */
+id _Nullable FBExtractElement(NSDictionary *src);
+
+/**
+ Cleanup items having element keys from the dictionary
+ 
+ @param src The source dictionary
+ @returns The resulting dictionary
+ */
+NSDictionary *FBCleanupElements(NSDictionary *src);
+
+/**
+ Parses key/value pairs of valid W3C capabilities
+
+ @param caps The source capabilitites dictionary
+ @param error Is set if there was an error while parsing the source capabilities
+ @returns Parsed capabilitites mapping or nil in case of failure
+ */
+NSDictionary<NSString *, id> *_Nullable FBParseCapabilities(NSDictionary<NSString *, id> *caps, NSError **error);
+
+NS_ASSUME_NONNULL_END

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Utilities/FBProtocolHelpers.m

@@ -0,0 +1,144 @@
+/**
+ * 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.
+ */
+
+#import "FBProtocolHelpers.h"
+
+#import "FBErrorBuilder.h"
+#import "FBLogger.h"
+
+static NSString *const W3C_ELEMENT_KEY = @"element-6066-11e4-a52e-4f735466cecf";
+static NSString *const JSONWP_ELEMENT_KEY = @"ELEMENT";
+
+static NSString *const APPIUM_PREFIX = @"appium";
+static NSString *const ALWAYS_MATCH_KEY = @"alwaysMatch";
+static NSString *const FIRST_MATCH_KEY = @"firstMatch";
+
+
+NSDictionary<NSString *, id> *FBToElementDict(id element)
+{
+  return @{
+    W3C_ELEMENT_KEY: element,
+    JSONWP_ELEMENT_KEY: element
+  };
+}
+
+id FBExtractElement(NSDictionary *src)
+{
+  for (NSString* key in src) {
+    if ([key.lowercaseString isEqualToString:W3C_ELEMENT_KEY.lowercaseString]
+        || [key.lowercaseString isEqualToString:JSONWP_ELEMENT_KEY.lowercaseString]) {
+      return src[key];
+    }
+  }
+  return nil;
+}
+
+NSDictionary *FBCleanupElements(NSDictionary *src)
+{
+  NSMutableDictionary *result = src.mutableCopy;
+  for (NSString* key in src) {
+    if ([key.lowercaseString isEqualToString:W3C_ELEMENT_KEY.lowercaseString]
+        || [key.lowercaseString isEqualToString:JSONWP_ELEMENT_KEY.lowercaseString]) {
+      [result removeObjectForKey:key];
+    }
+  }
+  return result.copy;
+}
+
+NSArray<NSString *> *standardCapabilities(void)
+{
+  static NSArray<NSString *> *standardCaps;
+  static dispatch_once_t onceStandardCaps;
+  dispatch_once(&onceStandardCaps, ^{
+    standardCaps = @[
+      @"browserName",
+      @"browserVersion",
+      @"platformName",
+      @"acceptInsecureCerts",
+      @"pageLoadStrategy",
+      @"proxy",
+      @"setWindowRect",
+      @"timeouts",
+      @"unhandledPromptBehavior"
+    ];
+  });
+  return standardCaps;
+}
+
+BOOL isStandardCap(NSString *capName)
+{
+  return [standardCapabilities() containsObject:capName];
+}
+
+NSDictionary<NSString *, id> *_Nullable mergeCaps(NSDictionary<NSString *, id> *primary, NSDictionary<NSString *, id> *secondary, NSError **error)
+{
+  NSMutableDictionary<NSString *, id> *result = primary.mutableCopy;
+  for (NSString *capName in secondary) {
+    if (nil != result[capName]) {
+      [[[FBErrorBuilder builder]
+        withDescriptionFormat:@"Property '%@' should not exist on both primary (%@) and secondary (%@) objects", capName, primary, secondary]
+       buildError:error];
+      return nil;
+    }
+    [result setObject:(id) secondary[capName] forKey:capName];
+  }
+  return result.copy;
+}
+
+NSDictionary<NSString *, id> *_Nullable stripPrefixes(NSDictionary<NSString *, id> *caps, NSError **error)
+{
+  NSString* prefix = [NSString stringWithFormat:@"%@:", APPIUM_PREFIX];
+  NSMutableDictionary<NSString *, id> *filteredCaps = [NSMutableDictionary dictionary];
+  NSMutableArray<NSString *> *badPrefixedCaps = [NSMutableArray array];
+  for (NSString *capName in caps) {
+    if (![capName hasPrefix:prefix]) {
+      [filteredCaps setObject:(id) caps[capName] forKey:capName];
+      continue;
+    }
+
+    NSString *strippedName = [capName substringFromIndex:prefix.length];
+    [filteredCaps setObject:(id) caps[capName] forKey:strippedName];
+    if (isStandardCap(strippedName)) {
+      [badPrefixedCaps addObject:strippedName];
+    }
+  }
+  if (badPrefixedCaps.count > 0) {
+    [[[FBErrorBuilder builder]
+      withDescriptionFormat:@"The capabilities %@ are standard and should not have the '%@' prefix", badPrefixedCaps, prefix]
+     buildError:error];
+    return nil;
+  }
+  return filteredCaps.copy;
+}
+
+NSDictionary<NSString *, id> *FBParseCapabilities(NSDictionary<NSString *, id> *caps, NSError **error)
+{
+  NSDictionary<NSString *, id> *alwaysMatch = caps[ALWAYS_MATCH_KEY] ?: @{};
+  NSArray<NSDictionary<NSString *, id> *> *firstMatch = caps[FIRST_MATCH_KEY] ?: @[];
+  NSArray<NSDictionary<NSString *, id> *> *allFirstMatchCaps = firstMatch.count == 0 ? @[@{}] : firstMatch;
+  NSDictionary<NSString *, id> *requiredCaps;
+  if (nil == (requiredCaps = stripPrefixes(alwaysMatch, error))) {
+    return nil;
+  }
+  for (NSDictionary<NSString *, id> *fmc in allFirstMatchCaps) {
+    NSDictionary<NSString *, id> *strippedCaps;
+    if (nil == (strippedCaps = stripPrefixes(fmc, error))) {
+      return nil;
+    }
+    NSDictionary<NSString *, id> *mergedCaps;
+    if (nil == (mergedCaps = mergeCaps(requiredCaps, strippedCaps, error))) {
+      [FBLogger logFmt:@"%@", (*error).description];
+      continue;
+    }
+    return mergedCaps;
+  }
+  [[[FBErrorBuilder builder]
+    withDescriptionFormat:@"Could not find matching capabilities from %@", caps]
+   buildError:error];
+  return nil;
+}

package/drivers/ios/WebDriverAgent/WebDriverAgentLib/Utilities/FBReflectionUtils.h

@@ -0,0 +1,23 @@
+/**
+ * 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.
+ */
+
+#import <Foundation/Foundation.h>
+
+NS_ASSUME_NONNULL_BEGIN
+
+/**
+ * Swizzles the implemntation of originalSelector with the swizzledSelector for the given class.
+ * Both methods must belong to this class.
+ *
+ * @param cls The class where to swizzle
+ * @param originalSelector original method selector
+ * @paramswizzledSelector swizzled method selector
+ */
+void FBReplaceMethod(Class cls, SEL originalSelector, SEL swizzledSelector);
+
+NS_ASSUME_NONNULL_END