// -*- Mode: ObjC; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4; fill-column: 100 -*- // // This Source Code Form is subject to the terms of the Mozilla Public // License, v. 2.0. If a copy of the MPL was not distributed with this // file, You can obtain one at http://mozilla.org/MPL/2.0/. #import "config.h" #import #import #import #import #import #import #import #import "ios.h" #import "FakeSocket.hpp" #import "LOOLWSD.hpp" #import "Log.hpp" #import "MobileApp.hpp" #import "SigUtil.hpp" #import "Util.hpp" #import "DocumentViewController.h" @interface DocumentViewController() { int closeNotificationPipeForForwardingThread[2]; NSURL *downloadAsTmpURL; } @end // From https://gist.github.com/myell0w/d8dfabde43f8da543f9c static BOOL isExternalKeyboardAttached() { BOOL externalKeyboardAttached = NO; @try { NSString *keyboardClassName = [@[@"UI", @"Key", @"boa", @"rd", @"Im", @"pl"] componentsJoinedByString:@""]; Class c = NSClassFromString(keyboardClassName); SEL sharedInstanceSEL = NSSelectorFromString(@"sharedInstance"); if (c == Nil || ![c respondsToSelector:sharedInstanceSEL]) { return NO; } #pragma clang diagnostic push #pragma clang diagnostic ignored "-Warc-performSelector-leaks" id sharedKeyboardInstance = [c performSelector:sharedInstanceSEL]; #pragma clang diagnostic pop if (![sharedKeyboardInstance isKindOfClass:NSClassFromString(keyboardClassName)]) { return NO; } NSString *externalKeyboardSelectorName = [@[@"is", @"InH", @"ardw", @"areK", @"eyb", @"oard", @"Mode"] componentsJoinedByString:@""]; SEL externalKeyboardSEL = NSSelectorFromString(externalKeyboardSelectorName); if (![sharedKeyboardInstance respondsToSelector:externalKeyboardSEL]) { return NO; } externalKeyboardAttached = ((BOOL ( *)(id, SEL))objc_msgSend)(sharedKeyboardInstance, externalKeyboardSEL); } @catch(__unused NSException *ex) { externalKeyboardAttached = NO; } return externalKeyboardAttached; } @implementation DocumentViewController static IMP standardImpOfInputAccessoryView = nil; - (void)viewDidLoad { [super viewDidLoad]; WKWebViewConfiguration *configuration = [[WKWebViewConfiguration alloc] init]; WKUserContentController *userContentController = [[WKUserContentController alloc] init]; [userContentController addScriptMessageHandler:self name:@"debug"]; [userContentController addScriptMessageHandler:self name:@"lool"]; [userContentController addScriptMessageHandler:self name:@"error"]; configuration.userContentController = userContentController; self.webView = [[WKWebView alloc] initWithFrame:CGRectZero configuration:configuration]; self.webView.translatesAutoresizingMaskIntoConstraints = NO; self.webView.allowsLinkPreview = NO; // Prevent the WebView from scrolling. Sadly I couldn't figure out how to do it in the JS, // so the problem is still there when using Online from Mobile Safari. self.webView.scrollView.scrollEnabled = NO; // Prevent the user from zooming the WebView by assigning ourselves as the delegate, and // stopping any zoom attempt in scrollViewWillBeginZooming: below. (The zooming of the document // contents is handled fully in JavaScript, the WebView has no knowledge of that.) self.webView.scrollView.delegate = self; [self.view addSubview:self.webView]; self.webView.navigationDelegate = self; self.webView.UIDelegate = self; // Hack for tdf#129380: Don't show the "shortcut bar" if a hardware keyboard is used. // From https://inneka.com/programming/objective-c/hide-shortcut-keyboard-bar-for-uiwebview-in-ios-9/ Class webBrowserClass = NSClassFromString(@"WKContentView"); Method method = class_getInstanceMethod(webBrowserClass, @selector(inputAccessoryView)); if (isExternalKeyboardAttached()) { IMP newImp = imp_implementationWithBlock(^(id _s) { if ([self.webView respondsToSelector:@selector(inputAssistantItem)]) { UITextInputAssistantItem *inputAssistantItem = [self.webView inputAssistantItem]; inputAssistantItem.leadingBarButtonGroups = @[]; inputAssistantItem.trailingBarButtonGroups = @[]; } return nil; }); IMP oldImp = method_setImplementation(method, newImp); if (standardImpOfInputAccessoryView == nil) standardImpOfInputAccessoryView = oldImp; } else { // If the external keyboard has been disconnected, restore the normal behaviour. if (standardImpOfInputAccessoryView != nil) { method_setImplementation(method, standardImpOfInputAccessoryView); } // Hack to make the on-screen keyboard pop up more eagerly when focus set to the textarea // using JavaScript. // From https://stackoverflow.com/questions/32449870/programmatically-focus-on-a-form-in-a-webview-wkwebview/32845699 static bool doneThisAlready = false; if (!doneThisAlready) { const char * methodSignature; doneThisAlready = true; if ([[NSProcessInfo processInfo] isOperatingSystemAtLeastVersion: (NSOperatingSystemVersion){13, 0, 0}]) { methodSignature = "_elementDidFocus:userIsInteracting:blurPreviousNode:activityStateChanges:userObject:"; } else { methodSignature = "_elementDidFocus:userIsInteracting:blurPreviousNode:changingActivityState:userObject:"; } // Override that internal method with an own wrapper that always passes the // userIsInteracting parameter as TRUE. That will cause the on-screen keyboard to pop up // when we call the focus() method on the textarea element in JavaScript. SEL selector = sel_getUid(methodSignature); Method method = class_getInstanceMethod(webBrowserClass, selector); if (method != nil) { IMP original = method_getImplementation(method); IMP override = imp_implementationWithBlock(^void(id me, void* arg0, BOOL arg1, BOOL arg2, BOOL arg3, id arg4) { ((void (*)(id, SEL, void*, BOOL, BOOL, BOOL, id))original)(me, selector, arg0, TRUE, arg2, arg3, arg4); }); method_setImplementation(method, override); } } } WKWebView *webViewP = self.webView; NSDictionary *views = NSDictionaryOfVariableBindings(webViewP); [self.view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"H:|-0-[webViewP(>=0)]-0-|" options:0 metrics:nil views:views]]; [self.view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"V:|-0-[webViewP(>=0)]-0-|" options:0 metrics:nil views:views]]; } - (void)viewWillAppear:(BOOL)animated { [super viewWillAppear:animated]; // When the user uses the camer to insert a photo, when the camera is displayed, this view is // removed. After the photo is taken it is then added back to the hierarchy. Our Document object // is still there intact, however, so no need to re-open the document when we re-appear. // Check whether the Document object is an already initialised one. if (self.document->fakeClientFd >= 0) return; [self.document openWithCompletionHandler:^(BOOL success) { if (success) { // Display the content of the document } else { // Make sure to handle the failed import appropriately, e.g., by presenting an error message to the user. } }]; } - (IBAction)dismissDocumentViewController { [self dismissViewControllerAnimated:YES completion:^ { [self.document closeWithCompletionHandler:^(BOOL success){ LOG_TRC("close completion handler gets " << (success?"YES":"NO")); [self.webView.configuration.userContentController removeScriptMessageHandlerForName:@"debug"]; [self.webView.configuration.userContentController removeScriptMessageHandlerForName:@"lool"]; [self.webView.configuration.userContentController removeScriptMessageHandlerForName:@"error"]; self.webView.configuration.userContentController = nil; [self.webView removeFromSuperview]; self.webView = nil; }]; }]; } - (void)webView:(WKWebView *)webView didCommitNavigation:(WKNavigation *)navigation { LOG_TRC("didCommitNavigation: " << [[navigation description] UTF8String]); } - (void)webView:(WKWebView *)webView didFailNavigation:(WKNavigation *)navigation withError:(NSError *)error { LOG_TRC("didFailNavigation: " << [[navigation description] UTF8String]); } - (void)webView:(WKWebView *)webView didFailProvisionalNavigation:(WKNavigation *)navigation withError:(NSError *)error { LOG_TRC("didFailProvisionalNavigation: " << [[navigation description] UTF8String]); } - (void)webView:(WKWebView *)webView didFinishNavigation:(WKNavigation *)navigation { LOG_TRC("didFinishNavigation: " << [[navigation description] UTF8String]); } - (void)webView:(WKWebView *)webView didReceiveServerRedirectForProvisionalNavigation:(WKNavigation *)navigation { LOG_TRC("didReceiveServerRedirectForProvisionalNavigation: " << [[navigation description] UTF8String]); } - (void)webView:(WKWebView *)webView didStartProvisionalNavigation:(WKNavigation *)navigation { LOG_TRC("didStartProvisionalNavigation: " << [[navigation description] UTF8String]); } - (void)webView:(WKWebView *)webView decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction decisionHandler:(void (^)(WKNavigationActionPolicy))decisionHandler { LOG_TRC("decidePolicyForNavigationAction: " << [[navigationAction description] UTF8String]); decisionHandler(WKNavigationActionPolicyAllow); } - (void)webView:(WKWebView *)webView decidePolicyForNavigationResponse:(WKNavigationResponse *)navigationResponse decisionHandler:(void (^)(WKNavigationResponsePolicy))decisionHandler { LOG_TRC("decidePolicyForNavigationResponse: " << [[navigationResponse description] UTF8String]); decisionHandler(WKNavigationResponsePolicyAllow); } - (WKWebView *)webView:(WKWebView *)webView createWebViewWithConfiguration:(WKWebViewConfiguration *)configuration forNavigationAction:(WKNavigationAction *)navigationAction windowFeatures:(WKWindowFeatures *)windowFeatures { LOG_TRC("createWebViewWithConfiguration"); return webView; } - (void)webView:(WKWebView *)webView runJavaScriptAlertPanelWithMessage:(NSString *)message initiatedByFrame:(WKFrameInfo *)frame completionHandler:(void (^)(void))completionHandler { LOG_TRC("runJavaScriptAlertPanelWithMessage: " << [message UTF8String]); // UIAlertView *alert = [[UIAlertView alloc] initWithTitle:@"" // message:message // delegate:nil // cancelButtonTitle:nil // otherButtonTitles:@"OK", nil]; // [alert show]; completionHandler(); } - (void)webView:(WKWebView *)webView runJavaScriptConfirmPanelWithMessage:(NSString *)message initiatedByFrame:(WKFrameInfo *)frame completionHandler:(void (^)(BOOL result))completionHandler { LOG_TRC("runJavaScriptConfirmPanelWithMessage: " << [message UTF8String]); completionHandler(YES); } - (void)webView:(WKWebView *)webView runJavaScriptTextInputPanelWithPrompt:(NSString *)prompt defaultText:(NSString *)defaultText initiatedByFrame:(WKFrameInfo *)frame completionHandler:(void (^)(NSString *result))completionHandler { LOG_TRC("runJavaScriptTextInputPanelWithPrompt: " << [prompt UTF8String]); completionHandler(@"Something happened."); } - (void)webViewWebContentProcessDidTerminate:(WKWebView *)webView { LOG_ERR("WebContent process terminated! What should we do?"); } - (void)userContentController:(WKUserContentController *)userContentController didReceiveScriptMessage:(WKScriptMessage *)message { int rc; struct pollfd p; if ([message.name isEqualToString:@"error"]) { LOG_ERR("Error from WebView: " << [message.body UTF8String]); } else if ([message.name isEqualToString:@"debug"]) { std::cerr << "==> " << [message.body UTF8String] << std::endl; } else if ([message.name isEqualToString:@"lool"]) { NSString *subBody = [message.body substringToIndex:std::min(100ul, ((NSString*)message.body).length)]; if (subBody.length < ((NSString*)message.body).length) subBody = [subBody stringByAppendingString:@"..."]; LOG_TRC("To Online: " << [subBody UTF8String]); if ([message.body isEqualToString:@"HULLO"]) { // Now we know that the JS has started completely // Contact the permanently (during app lifetime) listening LOOLWSD server // "public" socket assert(loolwsd_server_socket_fd != -1); rc = fakeSocketConnect(self.document->fakeClientFd, loolwsd_server_socket_fd); assert(rc != -1); // Create a socket pair to notify the below thread when the document has been closed fakeSocketPipe2(closeNotificationPipeForForwardingThread); // Start another thread to read responses and forward them to the JavaScript dispatch_async(dispatch_get_global_queue( DISPATCH_QUEUE_PRIORITY_DEFAULT, 0), ^{ Util::setThreadName("app2js"); while (true) { struct pollfd p[2]; p[0].fd = self.document->fakeClientFd; p[0].events = POLLIN; p[1].fd = self->closeNotificationPipeForForwardingThread[1]; p[1].events = POLLIN; if (fakeSocketPoll(p, 2, -1) > 0) { if (p[1].revents == POLLIN) { // The code below handling the "BYE" fake Websocket // message has closed the other end of the // closeNotificationPipeForForwardingThread. Let's close // the other end too just for cleanliness, even if a // FakeSocket as such is not a system resource so nothing // is saved by closing it. fakeSocketClose(self->closeNotificationPipeForForwardingThread[1]); // Close our end of the fake socket connection to the // ClientSession thread, so that it terminates fakeSocketClose(self.document->fakeClientFd); return; } if (p[0].revents == POLLIN) { int n = fakeSocketAvailableDataLength(self.document->fakeClientFd); // I don't want to check for n being -1 here, even if // that will lead to a crash (std::length_error from the // below std::vector constructor), as n being -1 is a // sign of something being wrong elsewhere anyway, and I // prefer to fix the root cause. Let's see how well this // works out. See tdf#122543 for such a case. if (n == 0) return; std::vector buf(n); n = fakeSocketRead(self.document->fakeClientFd, buf.data(), n); [self.document send2JS:buf.data() length:n]; } } else break; } assert(false); }); // First we simply send the Online C++ parts the URL and the appDocId. This corresponds // to the GET request with Upgrade to WebSocket. std::string url([[self.document->copyFileURL absoluteString] UTF8String]); p.fd = self.document->fakeClientFd; p.events = POLLOUT; fakeSocketPoll(&p, 1, -1); std::string message(url + " " + std::to_string(self.document->appDocId)); fakeSocketWrite(self.document->fakeClientFd, message.c_str(), message.size()); return; } else if ([message.body isEqualToString:@"BYE"]) { LOG_TRC("Document window terminating on JavaScript side. Closing our end of the socket."); [self bye]; return; } else if ([message.body isEqualToString:@"SLIDESHOW"]) { // Create the SVG for the slideshow. self.slideshowFile = FileUtil::createRandomTmpDir() + "/slideshow.svg"; self.slideshowURL = [NSURL fileURLWithPath:[NSString stringWithUTF8String:self.slideshowFile.c_str()] isDirectory:NO]; getDocumentDataForMobileAppDocId(self.document->appDocId).loKitDocument->saveAs([[self.slideshowURL absoluteString] UTF8String], "svg", nullptr); // Add a new full-screen WebView displaying the slideshow. WKWebViewConfiguration *configuration = [[WKWebViewConfiguration alloc] init]; WKUserContentController *userContentController = [[WKUserContentController alloc] init]; [userContentController addScriptMessageHandler:self name:@"lool"]; configuration.userContentController = userContentController; self.slideshowWebView = [[WKWebView alloc] initWithFrame:CGRectZero configuration:configuration]; [self.slideshowWebView becomeFirstResponder]; self.slideshowWebView.contentMode = UIViewContentModeScaleAspectFit; self.slideshowWebView.translatesAutoresizingMaskIntoConstraints = NO; self.slideshowWebView.navigationDelegate = self; self.slideshowWebView.UIDelegate = self; self.webView.hidden = true; [self.view addSubview:self.slideshowWebView]; [self.view bringSubviewToFront:self.slideshowWebView]; WKWebView *slideshowWebViewP = self.slideshowWebView; NSDictionary *views = NSDictionaryOfVariableBindings(slideshowWebViewP); [self.view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"H:|-0-[slideshowWebViewP(>=0)]-0-|" options:0 metrics:nil views:views]]; [self.view addConstraints:[NSLayoutConstraint constraintsWithVisualFormat:@"V:|-0-[slideshowWebViewP(>=0)]-0-|" options:0 metrics:nil views:views]]; [self.slideshowWebView loadRequest:[NSURLRequest requestWithURL:self.slideshowURL]]; return; } else if ([message.body isEqualToString:@"EXITSLIDESHOW"]) { std::remove(self.slideshowFile.c_str()); [self.slideshowWebView removeFromSuperview]; self.slideshowWebView = nil; self.webView.hidden = false; return; } else if ([message.body isEqualToString:@"PRINT"]) { // Create the PDF to print. std::string printFile = FileUtil::createRandomTmpDir() + "/print.pdf"; NSURL *printURL = [NSURL fileURLWithPath:[NSString stringWithUTF8String:printFile.c_str()] isDirectory:NO]; getDocumentDataForMobileAppDocId(self.document->appDocId).loKitDocument->saveAs([[printURL absoluteString] UTF8String], "pdf", nullptr); UIPrintInteractionController *pic = [UIPrintInteractionController sharedPrintController]; UIPrintInfo *printInfo = [UIPrintInfo printInfo]; printInfo.outputType = UIPrintInfoOutputGeneral; printInfo.orientation = UIPrintInfoOrientationPortrait; // FIXME Check the document? printInfo.jobName = @"Document"; // FIXME pic.printInfo = printInfo; pic.printingItem = printURL; [pic presentFromRect:CGRectZero inView:self.webView animated:YES completionHandler:^(UIPrintInteractionController *pic, BOOL completed, NSError *error) { LOG_TRC("print completion handler gets " << (completed?"YES":"NO")); std::remove(printFile.c_str()); }]; return; } else if ([message.body isEqualToString:@"FOCUSIFHWKBD"]) { if (isExternalKeyboardAttached()) { NSString *hwKeyboardMagic = @"{" " if (window.MagicToGetHWKeyboardWorking) {" " window.MagicToGetHWKeyboardWorking();" " }" "}"; [self.webView evaluateJavaScript:hwKeyboardMagic completionHandler:^(id _Nullable obj, NSError * _Nullable error) { if (error) { LOG_ERR("Error after " << [hwKeyboardMagic UTF8String] << ": " << [[error localizedDescription] UTF8String]); NSString *jsException = error.userInfo[@"WKJavaScriptExceptionMessage"]; if (jsException != nil) LOG_ERR("JavaScript exception: " << [jsException UTF8String]); } } ]; } return; } else if ([message.body hasPrefix:@"HYPERLINK"]) { NSArray *messageBodyItems = [message.body componentsSeparatedByString:@" "]; if ([messageBodyItems count] >= 2) { NSURL *url = [[NSURL alloc] initWithString:messageBodyItems[1]]; UIApplication *application = [UIApplication sharedApplication]; [application openURL:url options:@{} completionHandler:nil]; return; } } else if ([message.body isEqualToString:@"FONTPICKER"]) { UIFontPickerViewControllerConfiguration *configuration = [[UIFontPickerViewControllerConfiguration alloc] init]; configuration.includeFaces = YES; UIFontPickerViewController *picker = [[UIFontPickerViewController alloc] initWithConfiguration:configuration]; picker.delegate = self; [self presentViewController:picker animated:YES completion:nil]; return; } else if ([message.body hasPrefix:@"downloadas "]) { NSArray *messageBodyItems = [message.body componentsSeparatedByString:@" "]; NSString *format = nil; if ([messageBodyItems count] >= 2) { for (int i = 1; i < [messageBodyItems count]; i++) { if ([messageBodyItems[i] hasPrefix:@"format="]) format = [messageBodyItems[i] substringFromIndex:[@"format=" length]]; } if (format == nil) return; // Warn? // First save it in the requested format to a temporary location. First remove any // leftover identically named temporary file. NSURL *tmpFileDirectory = [[NSFileManager.defaultManager temporaryDirectory] URLByAppendingPathComponent:@"export"]; if (![NSFileManager.defaultManager createDirectoryAtURL:tmpFileDirectory withIntermediateDirectories:YES attributes:nil error:nil]) { LOG_ERR("Could not create directory " << [[tmpFileDirectory path] UTF8String]); return; } NSString *tmpFileName = [[[self.document->copyFileURL lastPathComponent] stringByDeletingPathExtension] stringByAppendingString:[@"." stringByAppendingString:format]]; downloadAsTmpURL = [tmpFileDirectory URLByAppendingPathComponent:tmpFileName]; std::remove([[downloadAsTmpURL path] UTF8String]); getDocumentDataForMobileAppDocId(self.document->appDocId).loKitDocument->saveAs([[downloadAsTmpURL absoluteString] UTF8String], [format UTF8String], nullptr); // Then verify that it indeed was saved, and then use an // UIDocumentPickerViewController to ask the user where to store the exported // document. struct stat statBuf; if (stat([[downloadAsTmpURL path] UTF8String], &statBuf) == -1) { LOG_ERR("Could apparently not save to '" << [[downloadAsTmpURL path] UTF8String] << "'"); return; } UIDocumentPickerViewController *picker = [[UIDocumentPickerViewController alloc] initWithURL:downloadAsTmpURL inMode:UIDocumentPickerModeExportToService]; picker.delegate = self; [self presentViewController:picker animated:YES completion:nil]; return; } } else if ([message.body hasPrefix:@"REMOVE "]) { // Sent from the img element's onload event handler. Remove tile file once it has been loaded. NSArray *messageBodyItems = [message.body componentsSeparatedByString:@" "]; assert([messageBodyItems count] == 2); NSURL *tile = [NSURL URLWithString:messageBodyItems[1]]; if (unlink([[tile path] UTF8String]) == -1) { LOG_SYS("Could not unlink tile " << [[tile path] UTF8String]); } return; } const char *buf = [message.body UTF8String]; p.fd = self.document->fakeClientFd; p.events = POLLOUT; fakeSocketPoll(&p, 1, -1); fakeSocketWrite(self.document->fakeClientFd, buf, strlen(buf)); } else { LOG_ERR("Unrecognized kind of message received from WebView: " << [message.name UTF8String] << ":" << [message.body UTF8String]); } } - (void)documentPicker:(UIDocumentPickerViewController *)controller didPickDocumentsAtURLs:(NSArray *)urls { std::remove([[downloadAsTmpURL path] UTF8String]); std::remove([[[downloadAsTmpURL URLByDeletingLastPathComponent] path] UTF8String]); } - (void)documentPickerWasCancelled:(UIDocumentPickerViewController *)controller { std::remove([[downloadAsTmpURL path] UTF8String]); std::remove([[[downloadAsTmpURL URLByDeletingLastPathComponent] path] UTF8String]); } - (void)scrollViewWillBeginZooming:(UIScrollView *)scrollView withView:(UIView *)view { scrollView.pinchGestureRecognizer.enabled = NO; } - (void)fontPickerViewControllerDidPickFont:(UIFontPickerViewController *)viewController { NSLog(@"Picked font: %@", [viewController selectedFontDescriptor]); NSDictionary *attribs = [[viewController selectedFontDescriptor] fontAttributes]; NSString *family = attribs[UIFontDescriptorFamilyAttribute]; if (family && [family length] > 0) { NSString *js = [[@"window.MagicFontNameCallback('" stringByAppendingString:family] stringByAppendingString:@"');"]; [self.webView evaluateJavaScript:js completionHandler:^(id _Nullable obj, NSError * _Nullable error) { if (error) { LOG_ERR("Error after " << [js UTF8String] << ": " << [[error localizedDescription] UTF8String]); NSString *jsException = error.userInfo[@"WKJavaScriptExceptionMessage"]; if (jsException != nil) LOG_ERR("JavaScript exception: " << [jsException UTF8String]); } } ]; } } - (void)bye { // Close one end of the socket pair, that will wake up the forwarding thread above fakeSocketClose(closeNotificationPipeForForwardingThread[0]); // deallocateDocumentDataForMobileAppDocId(self.document->appDocId); [[NSFileManager defaultManager] removeItemAtURL:self.document->copyFileURL error:nil]; // The dismissViewControllerAnimated must be done on the main queue. dispatch_async(dispatch_get_main_queue(), ^{ [self dismissDocumentViewController]; }); } @end // vim:set shiftwidth=4 softtabstop=4 expandtab: