/* -*- Mode: C++; 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/. */ #include #include "DocumentBroker.hpp" #include #include #include #include #include #include #include #include #include #include #include #include #include #include "Admin.hpp" #include "ClientSession.hpp" #include "Exceptions.hpp" #include "JailUtil.hpp" #include "COOLWSD.hpp" #include "SenderQueue.hpp" #include "Storage.hpp" #include "TileCache.hpp" #include "ProxyProtocol.hpp" #include "Util.hpp" #include "QuarantineUtil.hpp" #include #include #include #include #include #include #include #if !MOBILEAPP #include #endif #include #include #define TILES_ON_FLY_MIN_UPPER_LIMIT 10.0f using namespace COOLProtocol; using Poco::JSON::Object; #if ENABLE_DEBUG # define ADD_DEBUG_RENDERID (" renderid=cached\n") #else # define ADD_DEBUG_RENDERID ("\n") #endif void ChildProcess::setDocumentBroker(const std::shared_ptr& docBroker) { assert(docBroker && "Invalid DocumentBroker instance."); _docBroker = docBroker; // Add the prisoner socket to the docBroker poll. docBroker->addSocketToPoll(getSocket()); if (UnitWSD::isUnitTesting()) { UnitWSD::get().onDocBrokerAttachKitProcess(docBroker->getDocKey(), getPid()); } } void DocumentBroker::broadcastLastModificationTime( const std::shared_ptr& session) const { if (_storageManager.getLastModifiedTime().empty()) // No time from the storage (e.g., SharePoint 2013 and 2016) -> don't send return; std::ostringstream stream; stream << "lastmodtime: " << _storageManager.getLastModifiedTime(); const std::string message = stream.str(); // While loading, the current session is not yet added to // the sessions container, so we need to send to it directly. if (session) session->sendTextFrame(message); broadcastMessage(message); } /// The Document Broker Poll - one of these in a thread per document class DocumentBroker::DocumentBrokerPoll final : public TerminatingPoll { /// The DocumentBroker owning us. DocumentBroker& _docBroker; public: DocumentBrokerPoll(const std::string &threadName, DocumentBroker& docBroker) : TerminatingPoll(threadName), _docBroker(docBroker) { } void pollingThread() override { // Delegate to the docBroker. _docBroker.pollThread(); } }; std::atomic DocumentBroker::DocBrokerId(1); DocumentBroker::DocumentBroker(ChildType type, const std::string& uri, const Poco::URI& uriPublic, const std::string& docKey, unsigned mobileAppDocId) : _limitLifeSeconds(std::chrono::seconds::zero()), _uriOrig(uri), _type(type), _uriPublic(uriPublic), _docKey(docKey), _docId(Util::encodeId(DocBrokerId++, 3)), _documentChangedInStorage(false), _isViewFileExtension(false), _isModified(false), _cursorPosX(0), _cursorPosY(0), _cursorWidth(0), _cursorHeight(0), _poll(new DocumentBrokerPoll("doc" SHARED_DOC_THREADNAME_SUFFIX + _docId, *this)), _stop(false), _lockCtx(new LockContext()), _tileVersion(0), _debugRenderedTileCount(0), _wopiDownloadDuration(0), _mobileAppDocId(mobileAppDocId) { assert(!_docKey.empty()); assert(!COOLWSD::ChildRoot.empty()); #ifdef IOS assert(_mobileAppDocId > 0); #endif LOG_INF("DocumentBroker [" << COOLWSD::anonymizeUrl(_uriPublic.toString()) << "] created with docKey [" << _docKey << ']'); if (UnitWSD::isUnitTesting()) { UnitWSD::get().onDocBrokerCreate(_docKey); } } void DocumentBroker::setupPriorities() { #if !MOBILEAPP if (_type == ChildType::Batch) { int prio = COOLWSD::getConfigValue("per_document.batch_priority", 5); Util::setProcessAndThreadPriorities(_childProcess->getPid(), prio); } #endif } void DocumentBroker::setupTransfer(SocketDisposition &disposition, SocketDisposition::MoveFunction transferFn) { disposition.setTransfer(*_poll, transferFn); } void DocumentBroker::assertCorrectThread() const { _poll->assertCorrectThread(); } // The inner heart of the DocumentBroker - our poll loop. void DocumentBroker::pollThread() { LOG_INF("Starting docBroker polling thread for docKey [" << _docKey << "]."); _threadStart = std::chrono::steady_clock::now(); // Request a kit process for this doc. #if !MOBILEAPP do { static constexpr std::chrono::milliseconds timeoutMs(COMMAND_TIMEOUT_MS * 5); _childProcess = getNewChild_Blocks(); if (_childProcess || std::chrono::duration_cast( std::chrono::steady_clock::now() - _threadStart) > timeoutMs) break; // Nominal time between retries, lest we busy-loop. getNewChild could also wait, so don't double that here. std::this_thread::sleep_for(std::chrono::milliseconds(CHILD_REBALANCE_INTERVAL_MS / 10)); } while (!_stop && _poll->continuePolling() && !SigUtil::getTerminationFlag() && !SigUtil::getShutdownRequestFlag()); #else #ifdef IOS assert(_mobileAppDocId > 0); #endif _childProcess = getNewChild_Blocks(_mobileAppDocId); #endif if (!_childProcess) { // Let the client know we can't serve now. LOG_ERR("Failed to get new child."); // FIXME: need to notify all clients and shut this down ... // FIXME: return something good down the websocket ... #if 0 const std::string msg = SERVICE_UNAVAILABLE_INTERNAL_ERROR; ws.sendMessage(msg); // abnormal close frame handshake ws.shutdown(WebSocketHandler::StatusCodes::ENDPOINT_GOING_AWAY); #endif stop("Failed to get new child."); // Stop to mark it done and cleanup. _poll->stop(); _poll->removeSockets(); // Async cleanup. COOLWSD::doHousekeeping(); LOG_INF("Finished docBroker polling thread for docKey [" << _docKey << "]."); return; } // We have a child process. _childProcess->setDocumentBroker(shared_from_this()); LOG_INF("Doc [" << _docKey << "] attached to child [" << _childProcess->getPid() << "]."); setupPriorities(); #if !MOBILEAPP static const std::size_t IdleDocTimeoutSecs = COOLWSD::getConfigValue("per_document.idle_timeout_secs", 3600); // Used to accumulate B/W deltas. uint64_t adminSent = 0; uint64_t adminRecv = 0; auto lastBWUpdateTime = std::chrono::steady_clock::now(); auto lastClipboardHashUpdateTime = std::chrono::steady_clock::now(); const int limit_load_secs = #if ENABLE_DEBUG // paused waiting for a debugger to attach // ignore load time out std::getenv("PAUSEFORDEBUGGER") ? -1 : #endif COOLWSD::getConfigValue("per_document.limit_load_secs", 100); auto loadDeadline = std::chrono::steady_clock::now() + std::chrono::seconds(limit_load_secs); #endif const auto limStoreFailures = COOLWSD::getConfigValue("per_document.limit_store_failures", 5); // Main polling loop goodness. while (!_stop && _poll->continuePolling() && !SigUtil::getTerminationFlag()) { // Poll more frequently while unloading to cleanup sooner. const bool unloading = isMarkedToDestroy() || _docState.isUnloadRequested(); _poll->poll(unloading ? SocketPoll::DefaultPollTimeoutMicroS / 16 : SocketPoll::DefaultPollTimeoutMicroS); if (EnableExperimental && _stop) { LOG_DBG("Doc [" << _docKey << "] is flagged to stop after returning from poll."); break; } #if !MOBILEAPP const auto now = std::chrono::steady_clock::now(); // a tile's data is ~8k, a 4k screen is ~128 256x256 tiles if (_tileCache) _tileCache->setMaxCacheSize(8 * 1024 * 128 * _sessions.size()); if (isInteractive()) { // Extend the deadline while we are interactiving with the user. loadDeadline = now + std::chrono::seconds(limit_load_secs); continue; } if (!isLoaded() && (limit_load_secs > 0) && (now > loadDeadline)) { LOG_ERR("Doc [" << _docKey << "] is taking too long to load. Will kill process [" << _childProcess->getPid() << "]. per_document.limit_load_secs set to " << limit_load_secs << " secs."); broadcastMessage("error: cmd=load kind=docloadtimeout"); // Brutal but effective. if (_childProcess) _childProcess->terminate(); stop("Doc lifetime expired"); continue; } // Check if we had a sunset time and expired. if (_limitLifeSeconds > std::chrono::seconds::zero() && std::chrono::duration_cast(now - _threadStart) > _limitLifeSeconds) { LOG_WRN("Doc [" << _docKey << "] is taking too long to convert. Will kill process [" << _childProcess->getPid() << "]. per_document.limit_convert_secs set to " << _limitLifeSeconds.count() << " secs."); broadcastMessage("error: cmd=load kind=docexpired"); // Brutal but effective. if (_childProcess) _childProcess->terminate(); stop("Convert-to timed out"); continue; } if (std::chrono::duration_cast (now - lastBWUpdateTime).count() >= COMMAND_TIMEOUT_MS) { lastBWUpdateTime = now; uint64_t sent = 0, recv = 0; getIOStats(sent, recv); uint64_t deltaSent = 0, deltaRecv = 0; // connection drop transiently reduces this. if (sent > adminSent) { deltaSent = sent - adminSent; adminSent = sent; } if (recv > deltaRecv) { deltaRecv = recv - adminRecv; adminRecv = recv; } LOG_TRC("Doc [" << _docKey << "] added stats sent: +" << deltaSent << ", recv: +" << deltaRecv << " bytes to totals."); // send change since last notification. Admin::instance().addBytes(getDocKey(), deltaSent, deltaRecv); } if (_storage && _lockCtx->needsRefresh(now)) refreshLock(); #endif LOG_TRC("Poll: current activity: " << DocumentState::toString(_docState.activity())); switch (_docState.activity()) { case DocumentState::Activity::None: { // Check if there are queued activities. if (!_renameFilename.empty() && !_renameSessionId.empty()) { startRenameFileCommand(); // Nothing more to do until the save is complete. continue; } #if !MOBILEAPP // Remove idle documents after 1 hour. if (isLoaded() && getIdleTimeSecs() >= IdleDocTimeoutSecs) { autoSaveAndStop("idle"); } else #endif if (_sessions.empty() && (isLoaded() || _docState.isMarkedToDestroy())) { if (!isLoaded()) { // Nothing to do; no sessions, not loaded, marked to destroy. stop("dead"); } else if (_saveManager.isSaving() || isAsyncUploading()) { LOG_DBG("Don't terminate dead DocumentBroker: async saving in progress for " "docKey [" << getDocKey() << "]."); continue; } autoSaveAndStop("dead"); } else if (_docState.isUnloadRequested() || SigUtil::getShutdownRequestFlag() || _docState.isCloseRequested()) { if (limStoreFailures > 0 && (_saveManager.saveFailureCount() >= static_cast(limStoreFailures) || _storageManager.uploadFailureCount() >= static_cast(limStoreFailures))) { LOG_ERR("Failed to store the document and reached maximum retry count of " << limStoreFailures << ". Giving up. The document should be recoverable from the " "quarantine. Save failures: " << _saveManager.saveFailureCount() << ", Upload failures: " << _storageManager.uploadFailureCount()); stop("storefailed"); continue; } const std::string reason = SigUtil::getShutdownRequestFlag() ? "recycling" : (!_closeReason.empty() ? _closeReason : "unloading"); autoSaveAndStop(reason); } else if (!_stop && _saveManager.needAutosaveCheck()) { LOG_TRC("Triggering an autosave."); autoSave(false); } } break; case DocumentState::Activity::Save: case DocumentState::Activity::SaveAs: { if (_docState.isDisconnected()) { // We will never save. No need to wait for timeout. _saveManager.setLastSaveResult(false); endActivity(); } } break; // We have some activity ongoing. default: { constexpr std::chrono::seconds postponeAutosaveDuration(30); LOG_TRC("Postponing autosave check by " << postponeAutosaveDuration); _saveManager.postponeAutosave(postponeAutosaveDuration); } break; } #if !MOBILEAPP if (std::chrono::duration_cast(now - lastClipboardHashUpdateTime).count() >= 2) { for (auto &it : _sessions) { if (it.second->staleWaitDisconnect(now)) { std::string id = it.second->getId(); LOG_WRN("Unusual, Kit session " + id + " failed its disconnect handshake, killing"); finalRemoveSession(id); break; // it invalid. } } } if (std::chrono::duration_cast(now - lastClipboardHashUpdateTime).count() >= 5) { LOG_TRC("Rotating clipboard keys"); for (auto &it : _sessions) it.second->rotateClipboardKey(true); lastClipboardHashUpdateTime = now; } #endif } LOG_INF("Finished polling doc [" << _docKey << "]. stop: " << _stop << ", continuePolling: " << _poll->continuePolling() << ", ShutdownRequestFlag: " << SigUtil::getShutdownRequestFlag() << ", TerminationFlag: " << SigUtil::getTerminationFlag() << ", closeReason: " << _closeReason << ". Flushing socket."); // If the document is modified at exit, dump the state and warn. // Unless, that is, the user discarded the changes. if (isModified() && !(_docState.isCloseRequested() && _documentChangedInStorage)) { std::stringstream state; dumpState(state); LOG_WRN("DocumentBroker stopping although flagged as modified " << state.str()); if (UnitWSD::isUnitTesting()) UnitWSD::get().fail("Unsaved data detected while exiting DocBroker [" + _docKey + ']'); } else if (UnitWSD::isUnitTesting() && UnitWSD::get().isFinished() && UnitWSD::get().failed()) { std::stringstream state; dumpState(state); LOG_WRN("Test failed with doc [" << _docKey << "]: " << state.str()); } // Flush socket data first. constexpr auto flushTimeoutMicroS = std::chrono::microseconds(POLL_TIMEOUT_MICRO_S * 2); // ~1000ms LOG_INF("Flushing socket " << _poll->getSocketCount() << " for doc [" << _docKey << "] for " << flushTimeoutMicroS << ". stop: " << _stop << ", continuePolling: " << _poll->continuePolling() << ", ShutdownRequestFlag: " << SigUtil::getShutdownRequestFlag() << ", TerminationFlag: " << SigUtil::getTerminationFlag() << ". Terminating child with reason: [" << _closeReason << "]."); const auto flushStartTime = std::chrono::steady_clock::now(); while (_poll->getSocketCount()) { const auto now = std::chrono::steady_clock::now(); const auto elapsedMicroS = std::chrono::duration_cast(now - flushStartTime); if (elapsedMicroS > flushTimeoutMicroS) break; const std::chrono::microseconds timeoutMicroS = std::min(flushTimeoutMicroS - elapsedMicroS, std::chrono::microseconds(POLL_TIMEOUT_MICRO_S / 5)); _poll->poll(timeoutMicroS); } LOG_INF("Finished flushing socket for doc [" << _docKey << "]. stop: " << _stop << ", continuePolling: " << _poll->continuePolling() << ", ShutdownRequestFlag: " << SigUtil::getShutdownRequestFlag() << ", TerminationFlag: " << SigUtil::getTerminationFlag() << ". Terminating child with reason: [" << _closeReason << "]."); // Terminate properly while we can. terminateChild(_closeReason); // Stop to mark it done and cleanup. _poll->stop(); _poll->removeSockets(); #if !MOBILEAPP // Async cleanup. COOLWSD::doHousekeeping(); #endif if (_tileCache) _tileCache->clear(); LOG_INF("Finished docBroker polling thread for docKey [" << _docKey << "]."); } bool DocumentBroker::isAlive() const { if (!_stop || _poll->isAlive()) return true; // Polling thread not started or still running. // Shouldn't have live child process outside of the polling thread. return _childProcess && _childProcess->isAlive(); } DocumentBroker::~DocumentBroker() { assertCorrectThread(); LOG_INF("~DocumentBroker [" << _docKey << "] destroyed with " << _sessions.size() << " sessions left."); // Do this early - to avoid operating on _childProcess from two threads. _poll->joinThread(); if (!_sessions.empty()) LOG_WRN("Destroying DocumentBroker [" << _docKey << "] while having unremoved sessions."); // Need to first make sure the child exited, socket closed, // and thread finished before we are destroyed. _childProcess.reset(); #if !MOBILEAPP // Remove from the admin last, to avoid racing the next test. Admin::instance().rmDoc(_docKey); #endif if (UnitWSD::isUnitTesting()) { UnitWSD::get().onDocBrokerDestroy(_docKey); } } void DocumentBroker::joinThread() { _poll->joinThread(); } void DocumentBroker::stop(const std::string& reason) { if (_closeReason.empty() || _closeReason == reason) { LOG_DBG("Stopping DocumentBroker for docKey [" << _docKey << "] with reason: " << reason); _closeReason = reason; // used later in the polling loop } else { LOG_DBG("Stopping DocumentBroker for docKey [" << _docKey << "] with existing close reason: " << _closeReason << " (ignoring requested reason: " << reason << ')'); } _stop = true; _poll->wakeup(); } bool DocumentBroker::download(const std::shared_ptr& session, const std::string& jailId) { assertCorrectThread(); const std::string sessionId = session->getId(); LOG_INF("Loading [" << _docKey << "] for session [" << sessionId << "] in jail [" << jailId << ']'); { bool result; if (UnitWSD::get().filterLoad(sessionId, jailId, result)) return result; } if (_docState.isMarkedToDestroy()) { // Tearing down. LOG_WRN("Will not load document marked to destroy. DocKey: [" << _docKey << "]."); return false; } _jailId = jailId; // The URL is the publicly visible one, not visible in the chroot jail. // We need to map it to a jailed path and copy the file there. // user/doc/jailId const Poco::Path jailPath(JAILED_DOCUMENT_ROOT, jailId); const std::string jailRoot = getJailRoot(); LOG_INF("jailPath: " << jailPath.toString() << ", jailRoot: " << jailRoot); bool firstInstance = false; if (_storage == nullptr) { _docState.setStatus(DocumentState::Status::Downloading); // Pass the public URI to storage as it needs to load using the token // and other storage-specific data provided in the URI. const Poco::URI& uriPublic = session->getPublicUri(); LOG_DBG("Loading, and creating new storage instance for URI [" << COOLWSD::anonymizeUrl(uriPublic.toString()) << "]."); try { _storage = StorageBase::create(uriPublic, jailRoot, jailPath.toString(), /*takeOwnership=*/isConvertTo()); } catch (...) { session->sendMessage("loadstorage: failed"); throw; } if (_storage == nullptr) { // We should get an exception, not null. LOG_ERR("Failed to create Storage instance for [" << _docKey << "] in " << jailPath.toString()); return false; } firstInstance = true; } LOG_ASSERT(_storage); // Call the storage specific fileinfo functions std::string userId, username; std::string userExtraInfo; std::string watermarkText; std::string templateSource; #if !MOBILEAPP std::chrono::milliseconds checkFileInfoCallDurationMs = std::chrono::milliseconds::zero(); WopiStorage* wopiStorage = dynamic_cast(_storage.get()); if (wopiStorage != nullptr) { std::chrono::steady_clock::time_point start = std::chrono::steady_clock::now(); std::unique_ptr wopifileinfo = wopiStorage->getWOPIFileInfo(session->getAuthorization(), *_lockCtx); checkFileInfoCallDurationMs = std::chrono::duration_cast( std::chrono::steady_clock::now() - start); userId = wopifileinfo->getUserId(); username = wopifileinfo->getUsername(); userExtraInfo = wopifileinfo->getUserExtraInfo(); watermarkText = wopifileinfo->getWatermarkText(); templateSource = wopifileinfo->getTemplateSource(); _isViewFileExtension = COOLWSD::IsViewFileExtension(wopiStorage->getFileExtension()); if (CommandControl::LockManager::isLockedReadOnlyUser() || !wopifileinfo->getUserCanWrite() || _isViewFileExtension) { LOG_DBG("Setting the session as readonly"); session->setReadOnly(); if (COOLWSD::IsViewWithCommentsFileExtension(wopiStorage->getFileExtension())) { LOG_DBG("Allow session to change comments"); session->setAllowChangeComments(); } } else if (wopifileinfo->getUserCanWrite()) { session->setReadOnly(false); session->setAllowChangeComments(true); } // We will send the client about information of the usage type of the file. // Some file types may be treated differently than others. session->sendFileMode(session->isReadOnly(), session->isAllowChangeComments()); // Construct a JSON containing relevant WOPI host properties Object::Ptr wopiInfo = new Object(); if (!wopifileinfo->getPostMessageOrigin().empty()) { // Update the scheme to https if ssl or ssl termination is on if (wopifileinfo->getPostMessageOrigin().substr(0, 7) == "http://" && (COOLWSD::isSSLEnabled() || COOLWSD::isSSLTermination())) { wopifileinfo->getPostMessageOrigin().replace(0, 4, "https"); LOG_DBG("Updating PostMessageOrigin scheme to HTTPS. Updated origin is [" << wopifileinfo->getPostMessageOrigin() << "]."); } wopiInfo->set("PostMessageOrigin", wopifileinfo->getPostMessageOrigin()); } // If print, export are disabled, order client to hide these options in the UI if (wopifileinfo->getDisablePrint()) wopifileinfo->setHidePrintOption(true); if (wopifileinfo->getDisableExport()) wopifileinfo->setHideExportOption(true); wopiInfo->set("BaseFileName", wopiStorage->getFileInfo().getFilename()); if (wopifileinfo->getBreadcrumbDocName().size()) wopiInfo->set("BreadcrumbDocName", wopifileinfo->getBreadcrumbDocName()); if (!wopifileinfo->getTemplateSaveAs().empty()) wopiInfo->set("TemplateSaveAs", wopifileinfo->getTemplateSaveAs()); if (!templateSource.empty()) wopiInfo->set("TemplateSource", templateSource); wopiInfo->set("HidePrintOption", wopifileinfo->getHidePrintOption()); wopiInfo->set("HideSaveOption", wopifileinfo->getHideSaveOption()); wopiInfo->set("HideExportOption", wopifileinfo->getHideExportOption()); wopiInfo->set("DisablePrint", wopifileinfo->getDisablePrint()); wopiInfo->set("DisableExport", wopifileinfo->getDisableExport()); wopiInfo->set("DisableCopy", wopifileinfo->getDisableCopy()); wopiInfo->set("DisableInactiveMessages", wopifileinfo->getDisableInactiveMessages()); wopiInfo->set("DownloadAsPostMessage", wopifileinfo->getDownloadAsPostMessage()); wopiInfo->set("UserCanNotWriteRelative", wopifileinfo->getUserCanNotWriteRelative()); wopiInfo->set("EnableInsertRemoteImage", wopifileinfo->getEnableInsertRemoteImage()); wopiInfo->set("EnableShare", wopifileinfo->getEnableShare()); wopiInfo->set("HideUserList", wopifileinfo->getHideUserList()); wopiInfo->set("SupportsRename", wopifileinfo->getSupportsRename()); wopiInfo->set("UserCanRename", wopifileinfo->getUserCanRename()); wopiInfo->set("FileUrl", wopifileinfo->getFileUrl()); if (wopifileinfo->getHideChangeTrackingControls() != WopiStorage::WOPIFileInfo::TriState::Unset) wopiInfo->set("HideChangeTrackingControls", wopifileinfo->getHideChangeTrackingControls() == WopiStorage::WOPIFileInfo::TriState::True); std::ostringstream ossWopiInfo; wopiInfo->stringify(ossWopiInfo); const std::string wopiInfoString = ossWopiInfo.str(); LOG_TRC("Sending wopi info to client: " << wopiInfoString); // Contains PostMessageOrigin property which is necessary to post messages to parent // frame. Important to send this message immediately and not enqueue it so that in case // document load fails, cool is able to tell its parent frame via PostMessage API. session->sendMessage("wopi: " + wopiInfoString); // Mark the session as 'Document owner' if WOPI hosts supports it if (userId == _storage->getFileInfo().getOwnerId()) { LOG_DBG("Session [" << sessionId << "] is the document owner"); session->setDocumentOwner(true); } // Pass the ownership to client session session->setWopiFileInfo(wopifileinfo); } else #endif { LocalStorage* localStorage = dynamic_cast(_storage.get()); if (localStorage != nullptr) { std::unique_ptr localfileinfo = localStorage->getLocalFileInfo(); userId = localfileinfo->getUserId(); username = localfileinfo->getUsername(); if (COOLWSD::IsViewFileExtension(localStorage->getFileExtension())) { LOG_DBG("Setting the session as readonly"); session->setReadOnly(); if (COOLWSD::IsViewWithCommentsFileExtension(localStorage->getFileExtension())) { LOG_DBG("Allow session to change comments"); session->setAllowChangeComments(); } } session->sendFileMode(session->isReadOnly(), session->isAllowChangeComments()); } } #ifdef ENABLE_FEATURE_LOCK Object::Ptr lockInfo = new Object(); lockInfo->set("IsLockedUser", CommandControl::LockManager::isLockedUser()); lockInfo->set("IsLockReadOnly", CommandControl::LockManager::isLockReadOnly()); // Poco:Dynamic:Var does not support std::unordred_set so converted to std::vector std::vector lockedCommandList( CommandControl::LockManager::getLockedCommandList().begin(), CommandControl::LockManager::getLockedCommandList().end()); lockInfo->set("LockedCommandList", lockedCommandList); lockInfo->set("UnlockTitle", CommandControl::LockManager::getUnlockTitle()); lockInfo->set("UnlockLink", CommandControl::LockManager::getUnlockLink()); lockInfo->set("UnlockDescription", CommandControl::LockManager::getUnlockDescription()); lockInfo->set("WriterHighlights", CommandControl::LockManager::getWriterHighlights()); lockInfo->set("CalcHighlights", CommandControl::LockManager::getCalcHighlights()); lockInfo->set("ImpressHighlights", CommandControl::LockManager::getImpressHighlights()); lockInfo->set("DrawHighlights", CommandControl::LockManager::getDrawHighlights()); std::ostringstream ossLockInfo; lockInfo->stringify(ossLockInfo); const std::string lockInfoString = ossLockInfo.str(); LOG_TRC("Sending feature locking info to client: " << lockInfoString); session->sendMessage("featurelock: " + lockInfoString); #endif #ifdef ENABLE_FEATURE_RESTRICTION Object::Ptr restrictionInfo = new Object(); restrictionInfo->set("IsRestrictedUser", CommandControl::RestrictionManager::isRestrictedUser()); // Poco:Dynamic:Var does not support std::unordred_set so converted to std::vector std::vector restrictedCommandList(CommandControl::RestrictionManager::getRestrictedCommandList().begin(), CommandControl::RestrictionManager::getRestrictedCommandList().end()); restrictionInfo->set("RestrictedCommandList", restrictedCommandList); std::ostringstream ossRestrictionInfo; restrictionInfo->stringify(ossRestrictionInfo); const std::string restrictionInfoString = ossRestrictionInfo.str(); LOG_TRC("Sending command restriction info to client: " << restrictionInfoString); session->sendMessage("restrictedCommands: " + restrictionInfoString); #endif #if ENABLE_SUPPORT_KEY if (!COOLWSD::OverrideWatermark.empty()) watermarkText = COOLWSD::OverrideWatermark; #endif session->setUserId(userId); session->setUserName(username); session->setUserExtraInfo(userExtraInfo); session->setWatermarkText(watermarkText); session->createCanonicalViewId(_sessions); LOG_DBG("Setting username [" << COOLWSD::anonymizeUsername(username) << "] and userId [" << COOLWSD::anonymizeUsername(userId) << "] for session [" << sessionId << "] is canonical id " << session->getCanonicalViewId()); // Basic file information was stored by the above getWOPIFileInfo() or getLocalFileInfo() calls const StorageBase::FileInfo fileInfo = _storage->getFileInfo(); if (!fileInfo.isValid()) { LOG_ERR("Invalid fileinfo for URI [" << session->getPublicUri().toString() << "]."); return false; } if (firstInstance) { _storageManager.setLastModifiedTime(fileInfo.getLastModifiedTime()); LOG_DBG("Document timestamp: " << _storageManager.getLastModifiedTime()); } else { // Check if document has been modified by some external action LOG_TRC("Document modified time: " << fileInfo.getLastModifiedTime()); if (!_storageManager.getLastModifiedTime().empty() && !fileInfo.getLastModifiedTime().empty() && _storageManager.getLastModifiedTime() != fileInfo.getLastModifiedTime()) { LOG_DBG("Document " << _docKey << "] has been modified behind our back. " << "Informing all clients. Expected: " << _storageManager.getLastModifiedTime() << ", Actual: " << fileInfo.getLastModifiedTime()); _documentChangedInStorage = true; const std::string message = isModified() ? "error: cmd=storage kind=documentconflict" : "close: documentconflict"; session->sendTextFrame(message); broadcastMessage(message); } } broadcastLastModificationTime(session); // Let's download the document now, if not downloaded. std::chrono::milliseconds getFileCallDurationMs = std::chrono::milliseconds::zero(); if (!_storage->isDownloaded()) { std::chrono::steady_clock::time_point start = std::chrono::steady_clock::now(); std::string localPath = _storage->downloadStorageFileToLocal(session->getAuthorization(), *_lockCtx, templateSource); if (localPath.empty()) { throw std::runtime_error("Failed to retrieve document from storage"); } getFileCallDurationMs = std::chrono::duration_cast( std::chrono::steady_clock::now() - start); _docState.setStatus(DocumentState::Status::Loading); // Done downloading. // Only lock the document on storage for editing sessions // FIXME: why not lock before downloadStorageFileToLocal? Would also prevent race conditions if (!session->isReadOnly() && !_storage->updateLockState(session->getAuthorization(), *_lockCtx, true)) { LOG_ERR("Failed to lock!"); session->setLockFailed(_lockCtx->_lockFailureReason); // TODO: make this "read-only" a special one with a notification (infobar? balloon tip?) // and a button to unlock } #if !MOBILEAPP // Check if we have a prefilter "plugin" for this document format for (const auto& plugin : COOLWSD::PluginConfigurations) { try { const std::string extension(plugin->getString("prefilter.extension")); const std::string newExtension(plugin->getString("prefilter.newextension")); std::string commandLine(plugin->getString("prefilter.commandline")); if (localPath.length() > extension.length()+1 && strcasecmp(localPath.substr(localPath.length() - extension.length() -1).data(), (std::string(".") + extension).data()) == 0) { // Extension matches, try the conversion. We convert the file to another one in // the same (jail) directory, with just the new extension tacked on. const std::string newRootPath = _storage->getRootFilePath() + '.' + newExtension; // The commandline must contain the space-separated substring @INPUT@ that is // replaced with the input file name, and @OUTPUT@ for the output file name. int inputs(0), outputs(0); std::string input("@INPUT"); std::size_t pos = commandLine.find(input); if (pos != std::string::npos) { commandLine.replace(pos, input.length(), _storage->getRootFilePath()); ++inputs; } std::string output("@OUTPUT@"); pos = commandLine.find(output); if (pos != std::string::npos) { commandLine.replace(pos, output.length(), newRootPath); ++outputs; } StringVector args(StringVector::tokenize(commandLine, ' ')); std::string command(args[0]); args.erase(args.begin()); // strip the command if (inputs != 1 || outputs != 1) throw std::exception(); int process = Util::spawnProcess(command, args); int status = -1; const int rc = ::waitpid(process, &status, 0); if (rc != 0) { LOG_ERR("Conversion from " << extension << " to " << newExtension << " failed (" << rc << ")."); return false; } _storage->setRootFilePath(newRootPath); localPath += '.' + newExtension; } // We successfully converted the file to something LO can use; break out of the for // loop. break; } catch (const std::exception&) { // This plugin is not a proper prefilter one } } #endif const std::string localFilePath = Poco::Path(getJailRoot(), localPath).toString(); std::ifstream istr(localFilePath, std::ios::binary); Poco::SHA1Engine sha1; Poco::DigestOutputStream dos(sha1); Poco::StreamCopier::copyStream(istr, dos); dos.close(); LOG_INF("SHA1 for DocKey [" << _docKey << "] of [" << COOLWSD::anonymizeUrl(localPath) << "]: " << Poco::DigestEngine::digestToHex(sha1.digest())); std::string localPathEncoded; Poco::URI::encode(localPath, "#?", localPathEncoded); _uriJailed = Poco::URI(Poco::URI("file://"), localPathEncoded).toString(); _uriJailedAnonym = Poco::URI(Poco::URI("file://"), COOLWSD::anonymizeUrl(localPathEncoded)).toString(); _filename = fileInfo.getFilename(); #if !MOBILEAPP Quarantine::quarantineFile(this, _filename); #endif if (!templateSource.empty()) { // Invalid timestamp for templates, to force uploading once we save-after-loading. _saveManager.setLastModifiedTime(std::chrono::system_clock::time_point()); _storageManager.setLastUploadedFileModifiedTime( std::chrono::system_clock::time_point()); } else { // Use the local temp file's timestamp. const auto timepoint = FileUtil::Stat(localFilePath).modifiedTimepoint(); _saveManager.setLastModifiedTime(timepoint); _storageManager.setLastUploadedFileModifiedTime(timepoint); // Used to detect modifications. } bool dontUseCache = false; #if MOBILEAPP // avoid memory consumption for single-user local bits. // FIXME: arguably should/could do this for single user documents too. dontUseCache = true; #endif _tileCache = Util::make_unique(_storage->getUri().toString(), _saveManager.getLastModifiedTime(), dontUseCache); _tileCache->setThreadOwner(std::this_thread::get_id()); } #if !MOBILEAPP COOLWSD::dumpNewSessionTrace(getJailId(), sessionId, _uriOrig, _storage->getRootFilePath()); // Since document has been loaded, send the stats if its WOPI if (wopiStorage != nullptr) { // Add the time taken to load the file from storage and to check file info. _wopiDownloadDuration += getFileCallDurationMs + checkFileInfoCallDurationMs; const auto downloadSecs = _wopiDownloadDuration.count() / 1000.; const std::string msg = "stats: wopiloadduration " + std::to_string(downloadSecs); // In seconds. LOG_TRC("Sending to Client [" << msg << "]."); session->sendTextFrame(msg); } #endif return true; } std::string DocumentBroker::handleRenameFileCommand(std::string sessionId, std::string newFilename) { if (newFilename.empty()) return "error: cmd=renamefile kind=invalid"; //TODO: better filename validation. if (_docState.activity() == DocumentState::Activity::Rename) { if (_renameFilename == newFilename) return std::string(); // Nothing to do, it's a duplicate. else return "error: cmd=renamefile kind=conflict"; // Renaming in progress. } _renameFilename = std::move(newFilename); _renameSessionId = std::move(sessionId); if (_docState.activity() == DocumentState::Activity::None) { // We can start by saving now. startRenameFileCommand(); } return std::string(); } void DocumentBroker::startRenameFileCommand() { LOG_TRC("Starting renamefile command execution."); if (_renameSessionId.empty() || _renameFilename.empty()) { assert(!"Saving before renaming without valid filename or sessionId."); LOG_DBG("Error: Trying to saveBeforeRename with invalid filename [" << _renameFilename << "] and/or sessionId [" << _renameSessionId << "]"); return; } // Transition. if (!startActivity(DocumentState::Activity::Rename)) { return; } blockUI("rename"); // Prevent user interaction while we start renaming. constexpr bool dontTerminateEdit = false; // We will save, rename, and reload: terminate. constexpr bool dontSaveIfUnmodified = true; constexpr bool isAutosave = false; constexpr bool isExitSave = false; sendUnoSave(_renameSessionId, dontTerminateEdit, dontSaveIfUnmodified, isAutosave, isExitSave); } void DocumentBroker::endRenameFileCommand() { LOG_TRC("Ending renamefile command execution."); _renameSessionId.clear(); _renameFilename.clear(); unblockUI(); endActivity(); } bool DocumentBroker::attemptLock(const ClientSession& session, std::string& failReason) { const bool bResult = _storage->updateLockState(session.getAuthorization(), *_lockCtx, true); if (!bResult) failReason = _lockCtx->_lockFailureReason; return bResult; } DocumentBroker::NeedToUpload DocumentBroker::needToUploadToStorage() const { if (_storage == nullptr) { // This can happen when we reject the connection (unauthorized). return NeedToUpload::No; } // When destroying, we might have to force uploading if always_save_on_exit=true. // If unloadRequested is set, assume we will unload after uploading and exit. if (isMarkedToDestroy() || _docState.isUnloadRequested()) { static const bool always_save = COOLWSD::getConfigValue("per_document.always_save_on_exit", false); if (always_save) { if (_documentChangedInStorage) { LOG_WRN("Need to upload per always_save_on_exit config, but the document has a " "conflict. Cannot force uploading."); return NeedToUpload::Yes; // Still try. } LOG_INF("Need to upload per always_save_on_exit config " << (isMarkedToDestroy() ? "MarkedToDestroy" : "Unloading")); return NeedToUpload::Force; } } // Force uploading only for retryable failures, not conflicts. See FIXME below. if (!_storageManager.lastUploadSuccessful() && !_documentChangedInStorage) { //FIXME: Forcing is used when overwriting a 'document conflict' and // for uploading when otherwise we might not have an immediate // reason. We shouldn't use a single flag for both these uses. if (_documentChangedInStorage) { LOG_WRN("Last upload failed due to a conflict. Cannot force uploading."); return NeedToUpload::Yes; // Still try. } LOG_INF("Enabling forced uploading to storage as last attempt had failed."); return NeedToUpload::Force; } // Finally, see if we have a newer version than storage. if (isStorageOutdated()) return NeedToUpload::Yes; // Timestamp changed, upload. return NeedToUpload::No; // No reason to upload, seems up-to-date. } bool DocumentBroker::isStorageOutdated() const { // Get the modified-time of the file on disk. const auto st = FileUtil::Stat(_storage->getRootFilePathUploading()); if (!st.exists()) { LOG_TRC("File to upload to storage [" << _storage->getRootFilePathUploading() << "] does not exist."); return false; } const std::chrono::system_clock::time_point currentModifiedTime = st.modifiedTimepoint(); const std::chrono::system_clock::time_point lastModifiedTime = _storageManager.getLastUploadedFileModifiedTime(); LOG_TRC("File to upload to storage [" << _storage->getRootFilePathUploading() << "] was modified at " << currentModifiedTime << " and the last uploaded file was modified at " << lastModifiedTime << ", which are " << (currentModifiedTime == lastModifiedTime ? "identical." : "different.")); // Compare to the last uploaded file's modified-time. return currentModifiedTime != lastModifiedTime; } void DocumentBroker::handleSaveResponse(const std::string& sessionId, bool success, const std::string& result) { assertCorrectThread(); if (success) LOG_DBG("Save result from Core: saved (during " << DocumentState::toString(_docState.activity()) << ')'); else LOG_INF("Save result from Core (failure): " << result << " (during " << DocumentState::toString(_docState.activity()) << ')'); #if !MOBILEAPP // Create the 'upload' file regardless of success or failure, // because we don't know if the last upload worked or not. // DocBroker will have to decide to upload or skip. const std::string oldName = _storage->getRootFilePathToUpload(); const std::string newName = _storage->getRootFilePathUploading(); if (rename(oldName.c_str(), newName.c_str()) < 0) { // It's not an error if there was no file to rename, when the document isn't modified. const auto onrre = errno; LOG_TRC("Failed to renamed [" << oldName << "] to [" << newName << "] (" << Util::symbolicErrno(onrre) << ": " << std::strerror(onrre) << ')'); } else { LOG_TRC("Renamed [" << oldName << "] to [" << newName << ']'); } Quarantine::quarantineFile(this, Util::splitLast(newName, '/').second); #endif //!MOBILEAPP // Record that we got a response to avoid timing out on saving. _saveManager.setLastSaveResult(success || result == "unmodified"); // The the clients know of any save failures. if (!success && result != "unmodified") { const auto it = _sessions.find(sessionId); if (it != _sessions.end()) { LOG_DBG("Failed to save docKey [" << _docKey << "] as .uno:Save has failed in LOK. Notifying client " << sessionId); it->second->sendTextFrameAndLogError("error: cmd=storage kind=savefailed"); broadcastSaveResult(false, "Could not save the document"); } } checkAndUploadToStorage(sessionId); } // This is called when either we just got save response, or, // there was nothing to save and want to check for uploading. void DocumentBroker::checkAndUploadToStorage(const std::string& sessionId) { LOG_TRC("checkAndUploadToStorage with session " << sessionId); // See if we have anything to upload. NeedToUpload needToUploadState = needToUploadToStorage(); // Handle activity-specific logic. switch (_docState.activity()) { case DocumentState::Activity::None: break; case DocumentState::Activity::Rename: { // If we have nothing to upload, do the rename now. if (needToUploadState == NeedToUpload::No) { LOG_DBG("Renaming in storage as there is no new version to upload first."); std::string uploadAsPath; constexpr bool isRename = true; uploadAsToStorage(_renameSessionId, uploadAsPath, _renameFilename, isRename); endRenameFileCommand(); return; } } break; case DocumentState::Activity::Save: { // Done saving. endActivity(); } default: break; } #if !MOBILEAPP // Avoid multiple uploads during unloading if we know we need to save a new version. if (_docState.isUnloadRequested() && isPossiblyModified()) { // We are unloading but have possible modifications. Save again (done in poll). LOG_DBG("Document [" << getDocKey() << "] is unloading, but was possibly modified during saving. Skipping " "upload to save again before unloading."); return; } #endif if (needToUploadState != NeedToUpload::No) { uploadToStorage(sessionId, /*force=*/needToUploadState == NeedToUpload::Force); } else { // If the session is disconnected, remove. const auto it = _sessions.find(sessionId); if (it != _sessions.end() && it->second->isCloseFrame()) disconnectSessionInternal(sessionId); // If marked to destroy, then this was the last session. if (_docState.isMarkedToDestroy() || _sessions.empty()) { // Stop so we get cleaned up and removed. LOG_DBG("Stopping after saving because " << (_sessions.empty() ? "there are no active sessions left." : "the document is marked to destroy.")); stop("unloading"); } } } void DocumentBroker::uploadToStorage(const std::string& sessionId, bool force) { assertCorrectThread(); if (force && _storage) { _storage->forceSave(); } constexpr bool isRename = false; uploadToStorageInternal(sessionId, /*saveAsPath*/ std::string(), /*saveAsFilename*/ std::string(), isRename, force); // If marked to destroy, or session is disconnected, remove. const auto it = _sessions.find(sessionId); if (_docState.isMarkedToDestroy() || (it != _sessions.end() && it->second->isCloseFrame())) disconnectSessionInternal(sessionId); } void DocumentBroker::uploadAsToStorage(const std::string& sessionId, const std::string& uploadAsPath, const std::string& uploadAsFilename, const bool isRename) { assertCorrectThread(); uploadToStorageInternal(sessionId, uploadAsPath, uploadAsFilename, isRename, /*force=*/false); } void DocumentBroker::uploadAfterLoadingTemplate(const std::string& sessionId) { #if !MOBILEAPP // Create the 'upload' file as it gets created only when // handling .uno:Save, which isn't issued for templates // (save is done in Kit right after loading a template). const std::string oldName = _storage->getRootFilePathToUpload(); const std::string newName = _storage->getRootFilePathUploading(); if (rename(oldName.c_str(), newName.c_str()) < 0) { // It's not an error if there was no file to rename, when the document isn't modified. LOG_SYS("Expected to renamed the document [" << oldName << "] after template-loading to [" << newName << ']'); } else { LOG_TRC("Renamed [" << oldName << "] to [" << newName << ']'); } #endif //!MOBILEAPP uploadToStorage(sessionId, /*force=*/false); } void DocumentBroker::uploadToStorageInternal(const std::string& sessionId, const std::string& saveAsPath, const std::string& saveAsFilename, const bool isRename, const bool force) { assertCorrectThread(); LOG_DBG("Uploading to storage docKey [" << _docKey << "] for session [" << sessionId << "]. Force: " << force); const auto it = _sessions.find(sessionId); if (it == _sessions.end()) { LOG_ERR("Session with sessionId [" << sessionId << "] not found while storing document docKey [" << _docKey << "]. The document will not be uploaded to storage at this time."); broadcastSaveResult(false, "Session not found"); return; } const bool isSaveAs = !saveAsPath.empty(); auto session = it->second; const std::string uri = isSaveAs ? saveAsPath : session->getPublicUri().toString(); // Map the FileId from the docKey to the new filename to anonymize the new filename as the FileId. const std::string newFilename = Util::getFilenameFromURL(uri); const std::string fileId = Util::getFilenameFromURL(_docKey); if (COOLWSD::AnonymizeUserData) { LOG_DBG("New filename [" << COOLWSD::anonymizeUrl(newFilename) << "] will be known by its fileId [" << fileId << ']'); Util::mapAnonymized(newFilename, fileId); } assert(_storage && "Must have a valid Storage instance"); const std::string uriAnonym = COOLWSD::anonymizeUrl(uri); // If the file timestamp hasn't changed, skip uploading. const std::string filePath = _storage->getRootFilePathUploading(); const std::chrono::system_clock::time_point newFileModifiedTime = FileUtil::Stat(filePath).modifiedTimepoint(); if (!isSaveAs && newFileModifiedTime == _saveManager.getLastModifiedTime() && !isRename && !force) { // Nothing to do. const auto timeInSec = std::chrono::duration_cast( std::chrono::system_clock::now() - _saveManager.getLastModifiedTime()); LOG_DBG("Skipping unnecessary uploading to URI [" << uriAnonym << "] with docKey [" << _docKey << "]. File last modified " << timeInSec.count() << " seconds ago, timestamp unchanged."); _poll->wakeup(); broadcastSaveResult(true, "unmodified"); return; } LOG_DBG("Uploading [" << _docKey << "] after saving to URI [" << uriAnonym << "]."); _uploadRequest = Util::make_unique(uriAnonym, newFileModifiedTime, session, isSaveAs, isRename); StorageBase::AsyncUploadCallback asyncUploadCallback = [this](const StorageBase::AsyncUpload& asyncUp) { switch (asyncUp.state()) { case StorageBase::AsyncUpload::State::Running: LOG_TRC("Async upload of [" << _docKey << "] is in progress during " << DocumentState::toString(_docState.activity())); return; case StorageBase::AsyncUpload::State::Complete: { LOG_TRC("Finished uploading [" << _docKey << "] during " << DocumentState::toString(_docState.activity()) << ", processing results."); return handleUploadToStorageResponse(asyncUp.result()); } case StorageBase::AsyncUpload::State::None: // Unexpected: fallback. case StorageBase::AsyncUpload::State::Error: default: broadcastSaveResult(false, "Could not upload document to storage"); } //FIXME: flag the failure so we retry. LOG_WRN("Failed to upload [" << _docKey << "] asynchronously. " << DocumentState::toString(_docState.activity())); switch (_docState.activity()) { case DocumentState::Activity::None: break; case DocumentState::Activity::Rename: { LOG_DBG("Failed to renameFile because uploading post-save failed."); const std::string renameSessionId = _renameSessionId; endRenameFileCommand(); auto pair = _sessions.find(renameSessionId); if (pair != _sessions.end() && pair->second) pair->second->sendTextFrameAndLogError("error: cmd=renamefile kind=failed"); } break; default: break; } }; _storage->uploadLocalFileToStorageAsync(session->getAuthorization(), *_lockCtx, saveAsPath, saveAsFilename, isRename, *_poll, asyncUploadCallback); } void DocumentBroker::handleUploadToStorageResponse(const StorageBase::UploadResult& uploadResult) { if (!_uploadRequest) { // We shouldn't get here if there is no active upload request. LOG_ERR("No active upload request while handling upload result."); return; } // Storage upload is considered successful only when storage returns OK. const bool lastUploadSuccessful = uploadResult.getResult() == StorageBase::UploadResult::Result::OK; LOG_TRC("lastUploadSuccessful: " << lastUploadSuccessful); _storageManager.setLastUploadResult(lastUploadSuccessful); #if !MOBILEAPP if (lastUploadSuccessful && !isModified()) { // Flag the document as un-modified in the admin console. // But only when we have uploaded successfully and the document // is current not flagged as modified by Core. Admin::instance().modificationAlert(_docKey, getPid(), false); } #endif if (uploadResult.getResult() == StorageBase::UploadResult::Result::OK) { LOG_DBG("Last upload result: OK"); #if !MOBILEAPP WopiStorage* wopiStorage = dynamic_cast(_storage.get()); if (wopiStorage != nullptr) Admin::instance().setDocWopiUploadDuration(_docKey, std::chrono::duration_cast(wopiStorage->getWopiSaveDuration())); #endif if (!_uploadRequest->isSaveAs() && !_uploadRequest->isRename()) { // Saved and stored; update flags. _saveManager.setLastModifiedTime(_uploadRequest->newFileModifiedTime()); _storageManager.markLastUploadTime(); // Save the storage timestamp. _storageManager.setLastModifiedTime(_storage->getFileInfo().getLastModifiedTime()); // Set the timestamp of the file we uploaded, to detect changes. _storageManager.setLastUploadedFileModifiedTime(_uploadRequest->newFileModifiedTime()); // After a successful save, we are sure that document in the storage is same as ours _documentChangedInStorage = false; LOG_DBG("Uploaded docKey [" << _docKey << "] to URI [" << _uploadRequest->uriAnonym() << "] and updated timestamps. Document modified timestamp: " << _storageManager.getLastModifiedTime() << ". Current Activity: " << DocumentState::toString(_docState.activity())); // Handle activity-specific logic. switch (_docState.activity()) { case DocumentState::Activity::Rename: { LOG_DBG("Renaming in storage after uploading the last saved version."); std::string uploadAsPath; constexpr bool isRename = true; uploadAsToStorage(_renameSessionId, uploadAsPath, _renameFilename, isRename); endRenameFileCommand(); } break; default: { // Check stop conditions. } break; } // Resume polling. _poll->wakeup(); } else if (_uploadRequest->isRename()) { // encode the name const std::string& filename = uploadResult.getSaveAsName(); auto uri = Poco::URI(uploadResult.getSaveAsUrl()); // Remove the access_token, which belongs to the renaming user. Poco::URI::QueryParameters queryParams = uri.getQueryParameters(); queryParams.erase(std::remove_if(queryParams.begin(), queryParams.end(), [](const std::pair& pair) { return pair.first == "access_token"; }), queryParams.end()); uri.setQueryParameters(queryParams); const std::string url = uri.toString(); std::string encodedName; Poco::URI::encode(filename, "", encodedName); const std::string filenameAnonym = COOLWSD::anonymizeUrl(filename); std::ostringstream oss; oss << "renamefile: " << "filename=" << encodedName << " url=" << url; broadcastMessage(oss.str()); broadcastMessage("close: reloadafterrename"); } else { // normalize the url (mainly to " " -> "%20") const std::string url = Poco::URI(uploadResult.getSaveAsUrl()).toString(); const std::string& filename = uploadResult.getSaveAsName(); // encode the name std::string encodedName; Poco::URI::encode(filename, "", encodedName); const std::string filenameAnonym = COOLWSD::anonymizeUrl(filename); const auto session = _uploadRequest->session(); if (session) { LOG_DBG("Uploaded SaveAs docKey [" << _docKey << "] to URI [" << COOLWSD::anonymizeUrl(url) << "] with name [" << filenameAnonym << "] successfully."); std::ostringstream oss; oss << "saveas: url=" << url << " filename=" << encodedName << " xfilename=" << filenameAnonym; session->sendTextFrame(oss.str()); const auto fileExtension = _filename.substr(_filename.find_last_of('.')); if (!strcasecmp(fileExtension.c_str(), ".csv") || !strcasecmp(fileExtension.c_str(), ".txt")) { broadcastMessageToOthers("warn: " + oss.str() + " username=" + session->getUserName(), session); } } else { LOG_DBG("Uploaded SaveAs docKey [" << _docKey << "] to URI [" << COOLWSD::anonymizeUrl(url) << "] with name [" << filenameAnonym << "] successfully, but the client session is closed."); } } broadcastLastModificationTime(); if (_docState.isUnloadRequested()) { // We just uploaded, flag to destroy if unload is requested. LOG_DBG("Unload requested after uploading, marking to destroy."); _docState.markToDestroy(); } // If marked to destroy, and there are no late-arriving modifications, then stop. if ((_docState.isMarkedToDestroy() || _sessions.empty()) && !isPossiblyModified()) { // Stop so we get cleaned up and removed. LOG_DBG("Stopping after uploading because " << (_sessions.empty() ? "there are no active sessions left." : "the document is marked to destroy.")); stop("unloading"); } return; } else if (uploadResult.getResult() == StorageBase::UploadResult::Result::DISKFULL) { LOG_WRN("Disk full while uploading docKey [" << _docKey << "] to URI [" << _uploadRequest->uriAnonym() << "]. Making all sessions on doc read-only and notifying clients."); // Make everyone readonly and tell everyone that storage is low on diskspace. for (const auto& sessionIt : _sessions) { sessionIt.second->sendTextFrameAndLogError("error: cmd=storage kind=savediskfull"); } broadcastSaveResult(false, "Disk full", uploadResult.getReason()); } else if (uploadResult.getResult() == StorageBase::UploadResult::Result::UNAUTHORIZED) { LOG_DBG("Last upload result: UNAUTHORIZED"); const auto session = _uploadRequest->session(); if (session) { LOG_ERR("Cannot upload docKey [" << _docKey << "] to storage URI [" << _uploadRequest->uriAnonym() << "]. Invalid or expired access token. Notifying client."); session->sendTextFrameAndLogError("error: cmd=storage kind=saveunauthorized"); } else { LOG_ERR("Cannot upload docKey [" << _docKey << "] to storage URI [" << _uploadRequest->uriAnonym() << "]. Invalid or expired access token. The client session is closed."); } broadcastSaveResult(false, "Invalid or expired access token"); } else if (uploadResult.getResult() == StorageBase::UploadResult::Result::FAILED) { LOG_DBG("Last upload result: FAILED"); //TODO: Should we notify all clients? const auto session = _uploadRequest->session(); if (session) { LOG_ERR("Failed to upload docKey [" << _docKey << "] to URI [" << _uploadRequest->uriAnonym() << "]. Notifying client."); const std::string msg = std::string("error: cmd=storage kind=") + (_uploadRequest->isRename() ? "renamefailed" : "savefailed"); session->sendTextFrame(msg); } else { LOG_ERR("Failed to upload docKey [" << _docKey << "] to URI [" << _uploadRequest->uriAnonym() << "]. The client session is closed."); } broadcastSaveResult(false, "Save failed", uploadResult.getReason()); } else if (uploadResult.getResult() == StorageBase::UploadResult::Result::DOC_CHANGED || uploadResult.getResult() == StorageBase::UploadResult::Result::CONFLICT) { LOG_ERR("PutFile says that Document [" << _docKey << "] changed in storage"); _documentChangedInStorage = true; const std::string message = isPossiblyModified() ? "error: cmd=storage kind=documentconflict" : "close: documentconflict"; const std::size_t activeClients = broadcastMessage(message); broadcastSaveResult(false, "Conflict: Document changed in storage", uploadResult.getReason()); LOG_TRC("There are " << activeClients << " active clients after broadcasting documentconflict"); if (activeClients == 0) { // No clients were contacted; we will never resolve this conflict. stop("conflict"); std::stringstream state; dumpState(state); LOG_WRN("The document [" << _docKey << "] could not be uploaded to storage because there is a newer version there, " "and no active clients exist to resolve the conflict. The document should " "be recoverable from the quarantine. State:\n" << state.str()); } } } void DocumentBroker::broadcastSaveResult(bool success, const std::string& result, const std::string& errorMsg) { const std::string resultstr = success ? "true" : "false"; // Some sane limit, otherwise we get problems transferring this to the client with large strings (can be a whole webpage) std::string errorMsgFormatted = COOLProtocol::getAbbreviatedMessage(errorMsg); // Replace reserved characters errorMsgFormatted = Poco::translate(errorMsgFormatted, "\"", "'"); broadcastMessage("commandresult: { \"command\": \"save\", \"success\": " + resultstr + ", \"result\": \"" + result + "\", \"errorMsg\": \"" + errorMsgFormatted + "\"}"); } void DocumentBroker::setLoaded() { if (!isLoaded()) { _docState.setLive(); _loadDuration = std::chrono::duration_cast( std::chrono::steady_clock::now() - _threadStart); LOG_TRC("Document loaded in " << _loadDuration); } } void DocumentBroker::setInteractive(bool value) { if (isInteractive() != value) { _docState.setInteractive(value); LOG_TRC("Document has interactive dialogs before load"); } } std::string DocumentBroker::getWriteableSessionId() const { assertCorrectThread(); std::string savingSessionId; for (const auto& sessionIt : _sessions) { const auto& session = sessionIt.second; // Save the document using an editable and loaded session, or first ... // Note that isViewLoaded() precludes inWaitDisconnected(). if ((session->isViewLoaded() && session->isWritable()) || savingSessionId.empty()) { savingSessionId = session->getId(); } // or if any of the sessions is document owner, use that. //FIXME: can the owner be read-only? if (session->isDocumentOwner()) { savingSessionId = session->getId(); break; } } return savingSessionId; } void DocumentBroker::refreshLock() { assertCorrectThread(); const std::string savingSessionId = getWriteableSessionId(); LOG_TRC("Refresh lock " << _lockCtx->_lockToken << " with session [" << savingSessionId << ']'); auto it = _sessions.find(savingSessionId); if (it == _sessions.end()) LOG_ERR("No write-able session to refresh lock with"); else { std::shared_ptr session = it->second; if (!session || !_storage->updateLockState(session->getAuthorization(), *_lockCtx, true)) LOG_ERR("Failed to refresh lock"); } } bool DocumentBroker::autoSave(const bool force, const bool dontSaveIfUnmodified) { assertCorrectThread(); _saveManager.autosaveChecked(); LOG_TRC("autoSave(): forceful? " << force << ", dontSaveIfUnmodified: " << dontSaveIfUnmodified); if (_sessions.empty() || _storage == nullptr || !isLoaded() || (!isModified() && !force)) { // Nothing to do. LOG_TRC("Nothing to autosave [" << _docKey << "]."); return false; } if (_docState.isDisconnected()) { LOG_DBG("Cannot autosave when disconnected from Kit."); return false; } // Which session to use when auto saving ? // Prefer the last editing view, if still valid, otherwise, find the first writable sessionId. // Note: a loaded view cannot be disconnecting. const auto itLastEditingSession = _sessions.find(_lastEditingSessionId); const std::string savingSessionId = (itLastEditingSession != _sessions.end() && itLastEditingSession->second->isWritable() && itLastEditingSession->second->isViewLoaded()) ? _lastEditingSessionId : getWriteableSessionId(); // Remember the last save time, since this is the predicate. LOG_TRC("Checking to autosave [" << _docKey << "] using session [" << savingSessionId << ']'); bool sent = false; if (force) { LOG_TRC("Sending forced save command for [" << _docKey << "]."); // Don't terminate editing as this can be invoked by the admin OOM, but otherwise force saving anyway. // Flag isAutosave=false so the WOPI host wouldn't think this is a regular checkpoint and // potentially optimize it away. This is as good as user-issued save, since this is // triggered when the document is closed. In the case of network disconnection or browser crash // most users would want to have had the chance to hit save before the document unloaded. sent = sendUnoSave(savingSessionId, /*dontTerminateEdit=*/true, dontSaveIfUnmodified, /*isAutosave=*/false, /*isExitSave=*/true); } else if (isModified()) { // The configured maximum idle duration before saving. Zero to disable. static const std::chrono::milliseconds MaxIdleSaveDurationMs = std::chrono::seconds( COOLWSD::getConfigValue("per_document.idlesave_duration_secs", 30)); // The configured maximum duration before saving. Zero to disable. static const std::chrono::milliseconds MaxAutoSaveDurationMs = std::chrono::seconds( COOLWSD::getConfigValue("per_document.autosave_duration_secs", 300)); const std::chrono::steady_clock::time_point now = std::chrono::steady_clock::now(); const std::chrono::milliseconds inactivityTimeMs = std::chrono::duration_cast(now - _lastActivityTime); const auto timeSinceLastSaveMs = std::chrono::duration_cast( now - _storageManager.getLastUploadTime()); LOG_TRC("Time since last save of docKey [" << _docKey << "] is " << timeSinceLastSaveMs << " and most recent activity was " << inactivityTimeMs << " ago."); bool save = false; // Zero or negative config value disables save. // Either we've been idle long enough, or it's auto-save time. if (MaxIdleSaveDurationMs > std::chrono::milliseconds::zero() && inactivityTimeMs >= MaxIdleSaveDurationMs) { save = true; } if (MaxAutoSaveDurationMs > std::chrono::milliseconds::zero() && timeSinceLastSaveMs >= MaxAutoSaveDurationMs) { save = true; } if (save) { LOG_TRC("Sending timed save command for [" << _docKey << "]."); sent = sendUnoSave(savingSessionId, /*dontTerminateEdit=*/true, /*dontSaveIfUnmodified=*/true, /*isAutosave=*/true, /*isExitSave=*/false); } } return sent; } void DocumentBroker::autoSaveAndStop(const std::string& reason) { LOG_TRC("autoSaveAndStop for docKey [" << getDocKey() << "]: " << reason); if (_saveManager.isSaving() || isAsyncUploading()) { LOG_TRC("Async saving/uploading in progress for docKey [" << getDocKey() << ']'); return; } if (_docState.isDisconnected() && !isStorageOutdated()) { LOG_DBG("Disconnected from Kit and nothing to upload. Stopping"); stop(reason); return; } bool canStop = false; if (needToUploadToStorage() == NeedToUpload::No) { // Here we don't check for the modified flag because it can come in // very late, or not at all. We care that there is nothing to upload // and the last save succeeded, possibly because there was no // modifications, and there has been no activity since. if (!haveActivityAfterSaveRequest() && _saveManager.lastSaveRequestTime() < _saveManager.lastSaveResponseTime() && _saveManager.lastSaveSuccessful()) { // We can stop, but the modified flag is set. Delayed ModifiedStatus? if (isModified()) { if (_saveManager.timeSinceLastSaveResponse() < std::chrono::seconds(2)) { LOG_INF("Can stop " << reason << " DocumentBroker for docKey [" << getDocKey() << "] but will wait for isModified to clear."); return; } LOG_WRN("Will stop " << reason << " DocumentBroker for docKey [" << getDocKey() << "] even with isModified, which is not clearing."); } // Nothing to upload and last save was successful; stop. canStop = true; LOG_TRC("autoSaveAndStop for docKey [" << getDocKey() << "]: no modifications since last successful save. Stopping."); } else if (!isPossiblyModified()) { // Nothing to upload and no modifications; stop. canStop = true; LOG_TRC("autoSaveAndStop for docKey [" << getDocKey() << "]: not modified. Stopping."); } } // Don't hammer on saving. if (!canStop && _saveManager.canSaveNow(std::chrono::milliseconds(500))) { // Stop if there is nothing to save. const bool possiblyModified = isPossiblyModified(); LOG_INF("Autosaving " << reason << " DocumentBroker for docKey [" << getDocKey() << "] before terminating. isPossiblyModified: " << (possiblyModified ? "yes" : "no") << ", conflict: " << (_documentChangedInStorage ? "yes" : "no")); if (!autoSave(possiblyModified)) { // Nothing to save. Try to upload if necessary. const std::string sessionId = getWriteableSessionId(); if (!sessionId.empty()) { checkAndUploadToStorage(sessionId); if (isAsyncUploading()) { LOG_DBG("Uploading document before stopping."); return; } } else { // There is nothing to do here except to detect data-loss and stop. if (isStorageOutdated()) { std::stringstream state; dumpState(state); LOG_WRN("The document [" << _docKey << "] could not be uploaded to storage because there are no writable " "sessions to upload. The document should be recoverable from the " "quarantine. Stopping. State:\n" << state.str()); } canStop = true; } } } else if (!canStop) { LOG_TRC("Too soon to issue another save on [" << getDocKey() << "]: " << _saveManager.timeSinceLastSaveRequest() << " since last save request and " << _saveManager.timeSinceLastSaveRequest() << " since last save response"); } if (canStop) { // Nothing to save, nothing to upload, and no modifications. Stop. LOG_INF("Nothing to save or upload. Terminating " << reason << " DocumentBroker for docKey [" << getDocKey() << ']'); stop(reason); } } bool DocumentBroker::sendUnoSave(const std::string& sessionId, bool dontTerminateEdit, bool dontSaveIfUnmodified, bool isAutosave, bool isExitSave, const std::string& extendedData) { assertCorrectThread(); LOG_INF("Saving doc [" << _docKey << "] using session [" << sessionId << "]."); if (_sessions.find(sessionId) != _sessions.end()) { // Invalidate the timestamp to force persisting. _saveManager.setLastModifiedTime(std::chrono::system_clock::time_point()); // We do not want save to terminate editing mode if we are in edit mode now std::ostringstream oss; // arguments init oss << '{'; if (dontTerminateEdit) { oss << "\"DontTerminateEdit\":" "{" "\"type\":\"boolean\"," "\"value\":true" "}"; } if (dontSaveIfUnmodified) { if (dontTerminateEdit) oss << ','; oss << "\"DontSaveIfUnmodified\":" "{" "\"type\":\"boolean\"," "\"value\":true" "}"; } // arguments end oss << '}'; assert(_storage); _storage->setIsAutosave(isAutosave || UnitWSD::get().isAutosave()); _storage->setIsExitSave(isExitSave); _storage->setExtendedData(extendedData); const std::string saveArgs = oss.str(); LOG_TRC(".uno:Save arguments: " << saveArgs); const auto command = "uno .uno:Save " + saveArgs; if (forwardToChild(sessionId, command)) { _saveManager.markLastSaveRequestTime(); if (_docState.activity() == DocumentState::Activity::None) { // If we aren't in the midst of any particular activity, // then this is a generic save on its own. startActivity(DocumentState::Activity::Save); } } return true; } LOG_ERR("Failed to save doc [" << _docKey << "]: No valid session [" << sessionId << "]."); return false; } std::string DocumentBroker::getJailRoot() const { assert(!_jailId.empty()); return Poco::Path(COOLWSD::ChildRoot, _jailId).toString(); } std::size_t DocumentBroker::addSession(const std::shared_ptr& session) { try { return addSessionInternal(session); } catch (const std::exception& exc) { LOG_ERR("Failed to add session to [" << _docKey << "] with URI [" << COOLWSD::anonymizeUrl(session->getPublicUri().toString()) << "]: " << exc.what()); if (_sessions.empty()) { LOG_INF("Doc [" << _docKey << "] has no more sessions. Marking to destroy."); _docState.markToDestroy(); } throw; } } std::size_t DocumentBroker::addSessionInternal(const std::shared_ptr& session) { assertCorrectThread(); try { // First, download the document, since this can fail. if (!download(session, _childProcess->getJailId())) { const auto msg = "Failed to load document with URI [" + session->getPublicUri().toString() + "]."; LOG_ERR(msg); throw std::runtime_error(msg); } } catch (const StorageSpaceLowException&) { LOG_ERR("Out of storage while loading document with URI [" << session->getPublicUri().toString() << "]."); // We use the same message as is sent when some of cool's own locations are full, // even if in this case it might be a totally different location (file system, or // some other type of storage somewhere). This message is not sent to all clients, // though, just to all sessions of this document. alertAllUsers("internal", "diskfull"); throw; } catch (const std::exception& exc) { LOG_ERR("loading document exception: " << exc.what()); throw; } const std::string id = session->getId(); // Request a new session from the child kit. const std::string aMessage = "session " + id + ' ' + _docKey + ' ' + _docId + ' ' + std::to_string(session->getCanonicalViewId()); _childProcess->sendTextFrame(aMessage); #if !MOBILEAPP // Tell the admin console about this new doc std::string wopiHost = _storage->getUri().Poco::URI::getHost(); Admin::instance().addDoc(_docKey, getPid(), getFilename(), id, session->getUserName(), session->getUserId(), _childProcess->getSMapsFD(), wopiHost); Admin::instance().setDocWopiDownloadDuration(_docKey, _wopiDownloadDuration); #endif // Add and attach the session. _sessions.emplace(session->getId(), session); session->setState(ClientSession::SessionState::LOADING); const std::size_t count = _sessions.size(); LOG_TRC("Added " << (session->isReadOnly() ? "readonly" : "non-readonly") << " session [" << id << "] to docKey [" << _docKey << "] to have " << count << " sessions."); if (UnitWSD::isUnitTesting()) { UnitWSD::get().onDocBrokerAddSession(_docKey, session); } return count; } std::size_t DocumentBroker::removeSession(const std::string& id) { assertCorrectThread(); try { const auto it = _sessions.find(id); if (it == _sessions.end()) { LOG_ERR("Invalid or unknown session [" << id << "] to remove."); return _sessions.size(); } std::shared_ptr session = it->second; const std::size_t activeSessionCount = countActiveSessions(); const bool lastEditableSession = session->isWritable() && !haveAnotherEditableSession(id); static const bool dontSaveIfUnmodified = !COOLWSD::getConfigValue("per_document.always_save_on_exit", false); LOG_INF("Removing session [" << id << "] on docKey [" << _docKey << "]. Have " << _sessions.size() << " sessions (" << activeSessionCount << " active). IsLive: " << session->isLive() << ", IsReadOnly: " << session->isReadOnly() << ", IsAllowChangeComments: " << session->isAllowChangeComments() << ", IsWritable: " << session->isWritable() << ", Unloading: " << _docState.isUnloadRequested() << ", MarkToDestroy: " << _docState.isMarkedToDestroy() << ", LastEditableSession: " << lastEditableSession << ", DontSaveIfUnmodified: " << dontSaveIfUnmodified << ", IsPossiblyModified: " << isPossiblyModified()); // In theory, we almost could do this here: // #if MOBILEAPP // There is always just one "session" in a mobile app, and the same one process continues // running, so no need to delay the disconnectSessionInternal() call. Doing it like this // will also get rid of the docbroker and lokit_main thread for the document quicker. // But, in reality it has unintended side effects on iOS because if you have done changes to // the document, it does get saved, but that is only to the temporary copy. It is only in // the document callback handler for LOK_CALLBACK_UNO_COMMAND_RESULT that we then call the // system API to save that copy back to where it came from. See the // LOK_CALLBACK_UNO_COMMAND_RESULT case in ChildSession::loKitCallback() in // ChildSession.cpp. If we did use the below code snippet here, the document callback would // get unregistered right away in Document::onUnload in Kit.cpp. // autoSave(isPossiblyModified(), dontSaveIfUnmodified); // disconnectSessionInternal(id); // stop("stopped"); // So just go down the same code path as for normal Online: // If last editable, save and don't remove until after uploading to storage. if (!lastEditableSession || !autoSave(isPossiblyModified(), dontSaveIfUnmodified)) disconnectSessionInternal(id); // Last view going away; can destroy? if (activeSessionCount <= 1) { if (_saveManager.isSaving() || isAsyncUploading()) { // Don't destroy just yet, wait until save and upload are done. // Notice that the save and/or upload could have been triggered // earlier, and not necessarily here when removing this last session. _docState.setUnloadRequested(); LOG_DBG("Removing last session and will unload after saving and uploading. Setting " "UnloadRequested flag."); } else if (_sessions.empty()) { // Nothing to save, and we were the last. _docState.markToDestroy(); LOG_DBG("No more sessions after removing last. Setting MarkToDestroy flag."); } } else if (activeSessionCount > 0) { LOG_ASSERT_MSG(!_docState.isMarkedToDestroy(), "Have active sessions while marked to destroy"); } } catch (const std::exception& ex) { LOG_ERR("Error while removing session [" << id << "]: " << ex.what()); } return _sessions.size(); } void DocumentBroker::disconnectSessionInternal(const std::string& id) { assertCorrectThread(); try { #if !MOBILEAPP Admin::instance().rmDoc(_docKey, id); #endif auto it = _sessions.find(id); if (it != _sessions.end()) { #if !MOBILEAPP COOLWSD::dumpEndSessionTrace(getJailId(), id, _uriOrig); #endif if (_docState.isUnloadRequested()) { // We must be the last session, flag to destroy if unload is requested. LOG_ASSERT_MSG(countActiveSessions() <= 1, "Unload-requested with multiple sessions"); _docState.markToDestroy(); LOG_TRC("Unload requested while disconnecting session [" << id << "], having " << _sessions.size() << " sessions, marking to destroy."); } std::shared_ptr session = it->second; const bool lastEditableSession = session->isWritable() && !haveAnotherEditableSession(id); LOG_TRC("Disconnect session internal " << id << ", LastEditableSession: " << lastEditableSession << " destroy? " << _docState.isMarkedToDestroy() << " locked? " << _lockCtx->_isLocked); // Unlock the document, if last editable sessions, before we lose a token that can unlock. if (lastEditableSession && _lockCtx->_isLocked && _storage) { if (!_storage->updateLockState(it->second->getAuthorization(), *_lockCtx, false)) LOG_ERR("Failed to unlock!"); } bool hardDisconnect; if (it->second->inWaitDisconnected()) { LOG_TRC("hard disconnecting while waiting for disconnected handshake."); hardDisconnect = true; } else { hardDisconnect = it->second->disconnectFromKit(); if (isLoaded() || _sessions.size() > 1) { // Let the child know the client has disconnected. const std::string msg("child-" + id + " disconnect"); _childProcess->sendTextFrame(msg); } else { // We aren't even loaded and no other views--kill. // If we send disconnect, we risk hanging because // we flag Core for quiting via unipoll, but Core // would still continue loading. If at the end of // loading it shows a dialog (such as the macro or // csv import dialogs), it will wait for their // dismissal indefinetely. Neither would our // load-timeout kick in, since we would be gone. LOG_INF("Session [" << id << "] disconnected but DocKey [" << _docKey << "] isn't loaded yet. Terminating the child roughly."); _childProcess->terminate(); } } if (hardDisconnect) finalRemoveSession(id); // else wait for disconnected. } else { LOG_TRC("Session [" << id << "] not found to disconnect from docKey [" << _docKey << "]. Have " << _sessions.size() << " sessions."); } } catch (const std::exception& ex) { LOG_ERR("Error while disconnecting session [" << id << "]: " << ex.what()); } } void DocumentBroker::finalRemoveSession(const std::string& id) { assertCorrectThread(); try { auto it = _sessions.find(id); if (it != _sessions.end()) { const bool readonly = (it->second ? it->second->isReadOnly() : false); // Remove. The caller must have a reference to the session // in question, lest we destroy from underneath them. it->second->dispose(); _sessions.erase(it); const std::size_t count = _sessions.size(); Log::StreamLogger logger = Log::trace(); if (logger.enabled()) { logger << "Removed " << (readonly ? "readonly" : "non-readonly") << " session [" << id << "] from docKey [" << _docKey << "] to have " << count << " sessions:"; for (const auto& pair : _sessions) logger << pair.second->getId() << ' '; LOG_END(logger, true); } if (UnitWSD::isUnitTesting()) { UnitWSD::get().onDocBrokerRemoveSession(_docKey, it->second); } } else { LOG_TRC("Session [" << id << "] not found to remove from docKey [" << _docKey << "]. Have " << _sessions.size() << " sessions."); } } catch (const std::exception& ex) { LOG_ERR("Error while removing session [" << id << "]: " << ex.what()); } } std::shared_ptr DocumentBroker::createNewClientSession( const std::shared_ptr &ws, const std::string& id, const Poco::URI& uriPublic, const bool isReadOnly, const RequestDetails &requestDetails) { try { if (isMarkedToDestroy() || _docState.isUnloadRequested()) { LOG_INF("DocumentBroker [" << getDocKey() << "] is marked to destroy and will not create new client sessions."); return nullptr; } // Now we have a DocumentBroker and we're ready to process client commands. if (ws) { const std::string statusReady = "statusindicator: ready"; LOG_TRC("Sending to Client [" << statusReady << "]."); ws->sendTextMessage(statusReady.c_str(), statusReady.size()); } // In case of WOPI, if this session is not set as readonly, it might be set so // later after making a call to WOPI host which tells us the permission on files // (UserCanWrite param). auto session = std::make_shared(ws, id, shared_from_this(), uriPublic, isReadOnly, requestDetails); session->construct(); return session; } catch (const std::exception& exc) { LOG_ERR("Exception while preparing session [" << id << "]: " << exc.what()); } return nullptr; } void DocumentBroker::addCallback(const SocketPoll::CallbackFn& fn) { _poll->addCallback(fn); } void DocumentBroker::addSocketToPoll(const std::shared_ptr& socket) { _poll->insertNewSocket(socket); } void DocumentBroker::alertAllUsers(const std::string& msg) { assertCorrectThread(); if (UnitWSD::get().filterAlertAllusers(msg)) return; auto payload = std::make_shared(msg, Message::Dir::Out); LOG_DBG("Alerting all users of [" << _docKey << "]: " << msg); for (auto& it : _sessions) { if (!it.second->inWaitDisconnected()) it.second->enqueueSendMessage(payload); } } void DocumentBroker::setKitLogLevel(const std::string& level) { assertCorrectThread(); _childProcess->sendTextFrame("setloglevel " + level); } std::string DocumentBroker::getDownloadURL(const std::string& downloadId) { auto aFound = _registeredDownloadLinks.find(downloadId); if (aFound != _registeredDownloadLinks.end()) return aFound->second; return ""; } void DocumentBroker::unregisterDownloadId(const std::string& downloadId) { auto aFound = _registeredDownloadLinks.find(downloadId); if (aFound != _registeredDownloadLinks.end()) _registeredDownloadLinks.erase(aFound); } /// Handles input from the prisoner / child kit process bool DocumentBroker::handleInput(const std::vector& payload) { auto message = std::make_shared(payload.data(), payload.size(), Message::Dir::Out); LOG_TRC("DocumentBroker handling child message: [" << message->abbr() << "]."); #if !MOBILEAPP if (COOLWSD::TraceDumper) COOLWSD::dumpOutgoingTrace(getJailId(), "0", message->abbr()); #endif if (COOLProtocol::getFirstToken(message->forwardToken(), '-') == "client") { forwardToClient(message); } else { if (message->firstTokenMatches("tile:")) { handleTileResponse(payload); } else if (message->firstTokenMatches("tilecombine:")) { handleTileCombinedResponse(payload); } else if (message->firstTokenMatches("errortoall:")) { LOG_CHECK_RET(message->tokens().size() == 3, false); std::string cmd, kind; COOLProtocol::getTokenString((*message)[1], "cmd", cmd); LOG_CHECK_RET(cmd != "", false); COOLProtocol::getTokenString((*message)[2], "kind", kind); LOG_CHECK_RET(kind != "", false); Util::alertAllUsers(cmd, kind); } else if (message->firstTokenMatches("registerdownload:")) { LOG_CHECK_RET(message->tokens().size() == 3, false); std::string downloadid, url; COOLProtocol::getTokenString((*message)[1], "downloadid", downloadid); LOG_CHECK_RET(downloadid != "", false); COOLProtocol::getTokenString((*message)[2], "url", url); LOG_CHECK_RET(url != "", false); _registeredDownloadLinks[downloadid] = url; } else if (message->firstTokenMatches("traceevent:")) { LOG_CHECK_RET(message->tokens().size() == 1, false); if (COOLWSD::TraceEventFile != NULL && TraceEvent::isRecordingOn()) { const auto newLine = static_cast(memchr(payload.data(), '\n', payload.size())); if (newLine) COOLWSD::writeTraceEventRecording(newLine + 1, payload.size() - (newLine + 1 - payload.data())); } } else if (message->firstTokenMatches("forcedtraceevent:")) { LOG_CHECK_RET(message->tokens().size() == 1, false); if (COOLWSD::TraceEventFile != NULL) { const auto newLine = static_cast(memchr(payload.data(), '\n', payload.size())); if (newLine) COOLWSD::writeTraceEventRecording(newLine + 1, payload.size() - (newLine + 1 - payload.data())); } } else { LOG_ERR("Unexpected message: [" << message->abbr() << "]."); return false; } } return true; } std::size_t DocumentBroker::getMemorySize() const { return sizeof(DocumentBroker) + (!!_tileCache ? _tileCache->getMemorySize() : 0) + _sessions.size() * sizeof(ClientSession); } // Expected to be legacy, ~all new requests are tilecombinedRequests void DocumentBroker::handleTileRequest(const StringVector &tokens, const std::shared_ptr& session) { assertCorrectThread(); std::unique_lock lock(_mutex); TileDesc tile = TileDesc::parse(tokens); tile.setNormalizedViewId(session->getCanonicalViewId()); tile.setVersion(++_tileVersion); const std::string tileMsg = tile.serialize(); LOG_TRC("Tile request for " << tileMsg); if (!hasTileCache()) { LOG_WRN("Tile request without a loaded document?"); return; } TileCache::Tile cachedTile = _tileCache->lookupTile(tile); if (cachedTile) { const std::string response = tile.serialize("tile:", ADD_DEBUG_RENDERID); session->sendTile(response, cachedTile); return; } auto now = std::chrono::steady_clock::now(); if (tile.getBroadcast()) { for (auto& it: _sessions) { if (!it.second->inWaitDisconnected()) { tile.setNormalizedViewId(it.second->getCanonicalViewId()); tileCache().subscribeToTileRendering(tile, it.second, now); } } } else { tileCache().subscribeToTileRendering(tile, session, now); } // Forward to child to render. LOG_DBG("Sending render request for tile (" << tile.getPart() << ',' << tile.getTilePosX() << ',' << tile.getTilePosY() << ")."); const std::string request = "tile " + tileMsg; _childProcess->sendTextFrame(request); _debugRenderedTileCount++; } void DocumentBroker::handleTileCombinedRequest(TileCombined& tileCombined, const std::shared_ptr& session) { std::unique_lock lock(_mutex); LOG_TRC("TileCombined request for " << tileCombined.serialize()); if (!hasTileCache()) { LOG_WRN("Combined tile request without a loaded document?"); return; } // Check which newly requested tiles need rendering. const auto now = std::chrono::steady_clock::now(); std::vector tilesNeedsRendering; for (auto& tile : tileCombined.getTiles()) { tile.setVersion(++_tileVersion); TileCache::Tile cachedTile = _tileCache->lookupTile(tile); if(!cachedTile) { // Not cached, needs rendering. tilesNeedsRendering.push_back(tile); _debugRenderedTileCount++; tileCache().subscribeToTileRendering(tile, session, now); } } // Send rendering request, prerender before we actually send the tiles if (!tilesNeedsRendering.empty()) { TileCombined newTileCombined = TileCombined::create(tilesNeedsRendering); // Forward to child to render. const std::string req = newTileCombined.serialize("tilecombine"); LOG_TRC("Sending uncached residual tilecombine request to Kit: " << req); _childProcess->sendTextFrame(req); } // Accumulate tiles std::deque& requestedTiles = session->getRequestedTiles(); if (requestedTiles.empty()) { requestedTiles = std::deque(tileCombined.getTiles().begin(), tileCombined.getTiles().end()); } // Drop duplicated tiles, but use newer version number else { for (const auto& newTile : tileCombined.getTiles()) { const TileDesc& firstOldTile = *(requestedTiles.begin()); if(!session->isTextDocument() && newTile.getPart() != firstOldTile.getPart()) { LOG_WRN("Different part numbers in tile requests"); } else if (newTile.getTileWidth() != firstOldTile.getTileWidth() || newTile.getTileHeight() != firstOldTile.getTileHeight() ) { LOG_WRN("Different tile sizes in tile requests"); } bool tileFound = false; for (auto& oldTile : requestedTiles) { if(oldTile.getTilePosX() == newTile.getTilePosX() && oldTile.getTilePosY() == newTile.getTilePosY() && oldTile.getNormalizedViewId() == newTile.getNormalizedViewId()) { oldTile.setVersion(newTile.getVersion()); oldTile.setOldWireId(newTile.getOldWireId()); oldTile.setWireId(newTile.getWireId()); tileFound = true; break; } } if(!tileFound) requestedTiles.push_back(newTile); } } lock.unlock(); lock.release(); sendRequestedTiles(session); } /// lookup in global clipboard cache and send response, send error if missing if @sendError bool DocumentBroker::lookupSendClipboardTag(const std::shared_ptr &socket, const std::string &tag, bool sendError) { LOG_TRC("Clipboard request " << tag << " not for a live session - check cache."); #if !MOBILEAPP std::shared_ptr saved = COOLWSD::SavedClipboards->getClipboard(tag); if (saved) { std::ostringstream oss; oss << "HTTP/1.1 200 OK\r\n" << "Last-Modified: " << Util::getHttpTimeNow() << "\r\n" << "User-Agent: " << WOPI_AGENT_STRING << "\r\n" << "Content-Length: " << saved->length() << "\r\n" << "Content-Type: application/octet-stream\r\n" << "X-Content-Type-Options: nosniff\r\n" << "\r\n"; oss.write(saved->c_str(), saved->length()); socket->setSocketBufferSize( std::min(saved->length() + 256, std::size_t(Socket::MaximumSendBufferSize))); socket->send(oss.str()); socket->shutdown(); LOG_INF("Found and queued clipboard response for send of size " << saved->length()); return true; } #endif if (!sendError) return false; #if !MOBILEAPP // Bad request. HttpHelper::sendError(400, socket, "Failed to find this clipboard"); #endif socket->shutdown(); socket->ignoreInput(); return false; } void DocumentBroker::handleClipboardRequest(ClipboardRequest type, const std::shared_ptr &socket, const std::string &viewId, const std::string &tag, const std::shared_ptr &data) { for (auto& it : _sessions) { if (it.second->matchesClipboardKeys(viewId, tag)) { it.second->handleClipboardRequest(type, socket, tag, data); return; } } if (!lookupSendClipboardTag(socket, tag, true)) LOG_ERR("Could not find matching session to handle clipboard request for " << viewId << " tag: " << tag); } void DocumentBroker::sendRequestedTiles(const std::shared_ptr& session) { std::unique_lock lock(_mutex); // How many tiles we have on the visible area, set the upper limit accordingly Util::Rectangle normalizedVisArea = session->getNormalizedVisibleArea(); float tilesOnFlyUpperLimit = 0; if (normalizedVisArea.hasSurface() && session->getTileWidthInTwips() != 0 && session->getTileHeightInTwips() != 0) { const int tilesFitOnWidth = std::ceil(normalizedVisArea.getRight() / session->getTileWidthInTwips()) - std::ceil(normalizedVisArea.getLeft() / session->getTileWidthInTwips()) + 1; const int tilesFitOnHeight = std::ceil(normalizedVisArea.getBottom() / session->getTileHeightInTwips()) - std::ceil(normalizedVisArea.getTop() / session->getTileHeightInTwips()) + 1; const int tilesInVisArea = tilesFitOnWidth * tilesFitOnHeight; tilesOnFlyUpperLimit = std::max(TILES_ON_FLY_MIN_UPPER_LIMIT, tilesInVisArea * 1.1f); } else { tilesOnFlyUpperLimit = 200; // Have a big number here to get all tiles requested by file opening } // Drop tiles which we are waiting for too long session->removeOutdatedTilesOnFly(); auto now = std::chrono::steady_clock::now(); // All tiles were processed on client side that we sent last time, so we can send // a new batch of tiles which was invalidated / requested in the meantime std::deque& requestedTiles = session->getRequestedTiles(); if (!requestedTiles.empty() && hasTileCache()) { std::size_t delayedTiles = 0; std::vector tilesNeedsRendering; std::size_t beingRendered = _tileCache->countTilesBeingRenderedForSession(session, now); while (session->getTilesOnFlyCount() + beingRendered < tilesOnFlyUpperLimit && !requestedTiles.empty() && // If we delayed all tiles we don't send any tile (we will when next tileprocessed message arrives) delayedTiles < requestedTiles.size()) { TileDesc& tile = *(requestedTiles.begin()); // We already sent out two versions of the same tile, let's not send the third one // until we get a tileprocessed message for this specific tile. if (session->countIdenticalTilesOnFly(tile) >= 2) { LOG_DBG("Requested tile " << tile.getWireId() << " was delayed (already sent a version)!"); requestedTiles.push_back(requestedTiles.front()); requestedTiles.pop_front(); delayedTiles += 1; continue; } // Satisfy as many tiles from the cache. TileCache::Tile cachedTile = _tileCache->lookupTile(tile); if (cachedTile) { //TODO: Combine the response to reduce latency. const std::string response = tile.serialize("tile:", ADD_DEBUG_RENDERID); session->sendTile(response, cachedTile); } else { // Not cached, needs rendering. if (!tileCache().hasTileBeingRendered(tile, &now) || // There is no in progress rendering of the given tile tileCache().getTileBeingRenderedVersion(tile) < tile.getVersion()) // We need a newer version { tile.setVersion(++_tileVersion); tilesNeedsRendering.push_back(tile); _debugRenderedTileCount++; } tileCache().subscribeToTileRendering(tile, session, now); beingRendered++; } requestedTiles.pop_front(); } // Send rendering request for those tiles which were not prerendered if (!tilesNeedsRendering.empty()) { TileCombined newTileCombined = TileCombined::create(tilesNeedsRendering); // Forward to child to render. const std::string req = newTileCombined.serialize("tilecombine"); LOG_TRC("Some of the tiles were not prerendered. Sending residual tilecombine: " << req); _childProcess->sendTextFrame(req); } } } void DocumentBroker::cancelTileRequests(const std::shared_ptr& session) { std::unique_lock lock(_mutex); // Clear tile requests session->clearTilesOnFly(); session->getRequestedTiles().clear(); session->resetWireIdMap(); if (!hasTileCache()) return; const std::string canceltiles = tileCache().cancelTiles(session); if (!canceltiles.empty()) { LOG_DBG("Forwarding canceltiles request: " << canceltiles); _childProcess->sendTextFrame(canceltiles); } } void DocumentBroker::handleTileResponse(const std::vector& payload) { const std::string firstLine = getFirstLine(payload); LOG_DBG("Handling tile: " << firstLine); try { const std::size_t length = payload.size(); if (firstLine.size() < static_cast(length) - 1) { const TileDesc tile = TileDesc::parse(firstLine); const char* buffer = payload.data(); const std::size_t offset = firstLine.size() + 1; std::unique_lock lock(_mutex); tileCache().saveTileAndNotify(tile, buffer + offset, length - offset); } else { LOG_WRN("Dropping empty tile response: " << firstLine); // They will get re-issued if we don't forget them. } } catch (const std::exception& exc) { LOG_ERR("Failed to process tile response [" << firstLine << "]: " << exc.what() << '.'); } } void DocumentBroker::handleTileCombinedResponse(const std::vector& payload) { const std::string firstLine = getFirstLine(payload); LOG_DBG("Handling tile combined: " << firstLine); try { const std::size_t length = payload.size(); if (firstLine.size() <= static_cast(length) - 1) { const TileCombined tileCombined = TileCombined::parse(firstLine); const char* buffer = payload.data(); std::size_t offset = firstLine.size() + 1; std::unique_lock lock(_mutex); for (const auto& tile : tileCombined.getTiles()) { tileCache().saveTileAndNotify(tile, buffer + offset, tile.getImgSize()); offset += tile.getImgSize(); } } else { LOG_INF("Dropping empty tilecombine response: " << firstLine); // They will get re-issued if we don't forget them. } } catch (const std::exception& exc) { LOG_ERR("Failed to process tile response [" << firstLine << "]: " << exc.what() << '.'); } } bool DocumentBroker::haveAnotherEditableSession(const std::string& id) const { assertCorrectThread(); for (const auto& it : _sessions) { if (it.second->getId() != id && it.second->isViewLoaded() && it.second->isWritable()) { // This is a loaded session that is non-readonly. return true; } } // None found. return false; } std::size_t DocumentBroker::countActiveSessions() const { assertCorrectThread(); std::size_t count = 0; for (const auto& it : _sessions) { if (it.second->isLive()) { ++count; } } return count; } void DocumentBroker::setModified(const bool value) { #if !MOBILEAPP if (value) { // Flag the document as modified in the admin console. // But only flag it as unmodified when we do upload it. Admin::instance().modificationAlert(_docKey, getPid(), value); } #endif if (value || _storageManager.lastUploadSuccessful()) { // Set the X-COOL-WOPI-IsModifiedByUser header flag sent to storage. // But retain the last value if we failed to upload, so we don't clobber it. _storage->setUserModified(value); } LOG_DBG("Modified state set to " << value << " for Doc [" << _docId << ']'); _isModified = value; } bool DocumentBroker::isInitialSettingSet(const std::string& name) const { return _isInitialStateSet.find(name) != _isInitialStateSet.end(); } void DocumentBroker::setInitialSetting(const std::string& name) { _isInitialStateSet.emplace(name); } bool DocumentBroker::forwardToChild(const std::string& viewId, const std::string& message) { assertCorrectThread(); // Ignore userinactive, useractive message until document is loaded if (!isLoaded() && (message == "userinactive" || message == "useractive")) { return true; } LOG_TRC("Forwarding payload to child [" << viewId << "]: " << getAbbreviatedMessage(message)); if (Log::traceEnabled() && Util::startsWith(message, "paste ")) LOG_TRC("Logging paste payload (" << message.size() << " bytes) '" << message << "' end paste"); std::string msg = "child-" + viewId + ' ' + message; const auto it = _sessions.find(viewId); if (it != _sessions.end()) { assert(!_uriJailed.empty()); StringVector tokens = StringVector::tokenize(msg); if (tokens.size() > 1 && tokens.equals(1, "load")) { // The json options must come last. msg = tokens[0] + ' ' + tokens[1] + ' ' + tokens[2]; msg += " jail=" + _uriJailed; msg += " xjail=" + _uriJailedAnonym; msg += ' ' + tokens.cat(' ', 3); } _childProcess->sendTextFrame(msg); return true; } // try the not yet created sessions LOG_WRN("Child session [" << viewId << "] not found to forward message: " << getAbbreviatedMessage(message)); return false; } bool DocumentBroker::forwardToClient(const std::shared_ptr& payload) { assertCorrectThread(); const std::string& prefix = payload->forwardToken(); LOG_TRC("Forwarding payload to [" << prefix << "]: " << payload->abbr()); std::string name; std::string sid; if (COOLProtocol::parseNameValuePair(payload->forwardToken(), name, sid, '-') && name == "client") { const auto& data = payload->data().data(); const auto& size = payload->size(); if (sid == "all") { // Broadcast to all. // Events could cause the removal of sessions. std::map> sessions(_sessions); for (const auto& it : _sessions) { if (!it.second->inWaitDisconnected()) it.second->handleKitToClientMessage(data, size); } } else { const auto it = _sessions.find(sid); if (it != _sessions.end()) { // Take a ref as the session could be removed from _sessions // if it's the save confirmation keeping a stopped session alive. std::shared_ptr session = it->second; return session->handleKitToClientMessage(data, size); } else { LOG_WRN("Client session [" << sid << "] not found to forward message: " << payload->abbr()); } } } else { LOG_ERR("Unexpected prefix of forward-to-client message: " << prefix); } return false; } void DocumentBroker::shutdownClients(const std::string& closeReason) { assertCorrectThread(); LOG_INF("Terminating " << _sessions.size() << " clients of doc [" << _docKey << "] with reason: " << closeReason); // First copy into local container, since removeSession // will erase from _sessions, but will leave the last. std::map> sessions = _sessions; for (const auto& pair : sessions) { std::shared_ptr session = pair.second; try { if (session->inWaitDisconnected()) finalRemoveSession(session->getId()); else { // Notify the client and disconnect. session->shutdownGoingAway(closeReason); // Remove session, save, and mark to destroy. removeSession(session->getId()); } } catch (const std::exception& exc) { LOG_ERR("Error while shutting down client [" << session->getName() << "]: " << exc.what()); } } } void DocumentBroker::terminateChild(const std::string& closeReason) { assertCorrectThread(); #if !MOBILEAPP Quarantine::quarantineFile(this, _filename); #endif LOG_INF("Terminating doc [" << _docKey << "] with reason: " << closeReason); // Close all running sessions first. shutdownClients(closeReason); if (_childProcess) { LOG_INF("Terminating child [" << getPid() << "] of doc [" << _docKey << "]."); _childProcess->close(); } stop(closeReason); } void DocumentBroker::closeDocument(const std::string& reason) { assertCorrectThread(); _docState.setCloseRequested(); _closeReason = reason; if (_documentChangedInStorage) { // Discarding changes in the face of conflict in storage. LOG_DBG("Closing DocumentBroker for docKey [" << _docKey << "] and discarding changes with reason: " << reason); stop(reason); } else { LOG_DBG("Closing DocumentBroker for docKey [" << _docKey << "] with reason: " << reason); } } void DocumentBroker::disconnectedFromKit() { LOG_DBG("DocBroker " << _docKey << " Disconnected from Kit. Flagging to close."); _docState.setDisconnected(); closeDocument("docdisconnected"); } std::size_t DocumentBroker::broadcastMessage(const std::string& message) const { assertCorrectThread(); LOG_DBG("Broadcasting message [" << message << "] to all " << _sessions.size() << " sessions."); std::size_t count = 0; for (const auto& sessionIt : _sessions) { count += sessionIt.second->sendTextFrame(message); } return count; } void DocumentBroker::broadcastMessageToOthers(const std::string& message, const std::shared_ptr& _session) const { assertCorrectThread(); LOG_DBG("Broadcasting message [" << message << "] to all, except for " << _session->getId() << _sessions.size() << " sessions."); for (const auto& sessionIt : _sessions) { if (sessionIt.second == _session) continue; sessionIt.second->sendTextFrame(message); } } void DocumentBroker::updateLastActivityTime() { _lastActivityTime = std::chrono::steady_clock::now(); #if !MOBILEAPP Admin::instance().updateLastActivityTime(_docKey); #endif } void DocumentBroker::getIOStats(uint64_t &sent, uint64_t &recv) { sent = 0; recv = 0; assertCorrectThread(); for (const auto& sessionIt : _sessions) { uint64_t s = 0, r = 0; sessionIt.second->getIOStats(s, r); sent += s; recv += r; } } #if !MOBILEAPP void StatelessBatchBroker::removeFile(const std::string &uriOrig) { // Remove and report errors on failure. FileUtil::removeFile(uriOrig); const std::string dir = Poco::Path(uriOrig).parent().toString(); if (FileUtil::isEmptyDirectory(dir)) FileUtil::removeFile(dir); } static std::atomic gConvertToBrokerInstanceCouter; std::size_t ConvertToBroker::getInstanceCount() { return gConvertToBrokerInstanceCouter; } ConvertToBroker::ConvertToBroker(const std::string& uri, const Poco::URI& uriPublic, const std::string& docKey, const std::string& format, const std::string& sOptions) : StatelessBatchBroker(uri, uriPublic, docKey) , _format(format) , _sOptions(sOptions) { LOG_TRC("Created ConvertToBroker: uri: [" << uri << "], uriPublic: [" << uriPublic.toString() << "], docKey: [" << docKey << "], format: [" << format << "], options: [" << sOptions << "]."); static const std::chrono::seconds limit_convert_secs( COOLWSD::getConfigValue("per_document.limit_convert_secs", 100)); _limitLifeSeconds = limit_convert_secs; ++gConvertToBrokerInstanceCouter; } ConvertToBroker::~ConvertToBroker() {} bool ConvertToBroker::startConversion(SocketDisposition &disposition, const std::string &id) { std::shared_ptr docBroker = std::static_pointer_cast(shared_from_this()); // Create a session to load the document. const bool isReadOnly = true; // FIXME: associate this with moveSocket (?) std::shared_ptr nullPtr; RequestDetails requestDetails("convert-to"); _clientSession = std::make_shared(nullPtr, id, docBroker, getPublicUri(), isReadOnly, requestDetails); _clientSession->construct(); if (!_clientSession) return false; docBroker->setupTransfer(disposition, [docBroker] (const std::shared_ptr &moveSocket) { auto streamSocket = std::static_pointer_cast(moveSocket); docBroker->_clientSession->setSaveAsSocket(streamSocket); // First add and load the session. docBroker->addSession(docBroker->_clientSession); // Load the document manually and request saving in the target format. std::string encodedFrom; Poco::URI::encode(docBroker->getPublicUri().getPath(), "", encodedFrom); // add batch mode, no interactive dialogs const std::string _load = "load url=" + encodedFrom + " batch=true"; std::vector loadRequest(_load.begin(), _load.end()); docBroker->_clientSession->handleMessage(loadRequest); // Save is done in the setLoaded }); return true; } void ConvertToBroker::dispose() { if (!_uriOrig.empty()) { gConvertToBrokerInstanceCouter--; removeFile(_uriOrig); _uriOrig.clear(); } } void ConvertToBroker::setLoaded() { DocumentBroker::setLoaded(); // FIXME: Check for security violations. Poco::Path toPath(getPublicUri().getPath()); toPath.setExtension(_format); // file:///user/docs/filename.ext normally, file:////user/docs/filename.ext in the nocaps case const std::string toJailURL = "file://" + (COOLWSD::NoCapsForKit? getJailRoot(): "") + std::string(JAILED_DOCUMENT_ROOT) + toPath.getFileName(); std::string encodedTo; Poco::URI::encode(toJailURL, "", encodedTo); // Convert it to the requested format. const std::string saveAsCmd = "saveas url=" + encodedTo + " format=" + _format + " options=" + _sOptions; // Send the save request ... std::vector saveasRequest(saveAsCmd.begin(), saveAsCmd.end()); _clientSession->handleMessage(saveasRequest); } static std::atomic gRenderSearchResultBrokerInstanceCouter; std::size_t RenderSearchResultBroker::getInstanceCount() { return gRenderSearchResultBrokerInstanceCouter; } RenderSearchResultBroker::RenderSearchResultBroker( std::string const& uri, Poco::URI const& uriPublic, std::string const& docKey, std::shared_ptr> const& pSearchResultContent) : StatelessBatchBroker(uri, uriPublic, docKey) , _pSearchResultContent(pSearchResultContent) { LOG_TRC("Created RenderSearchResultBroker: uri: [" << uri << "], uriPublic: [" << uriPublic.toString() << "], docKey: [" << docKey << "]."); gConvertToBrokerInstanceCouter++; } RenderSearchResultBroker::~RenderSearchResultBroker() {} bool RenderSearchResultBroker::executeCommand(SocketDisposition& disposition, std::string const& id) { std::shared_ptr docBroker = std::static_pointer_cast(shared_from_this()); const bool isReadOnly = true; std::shared_ptr emptyProtocolHandler; RequestDetails requestDetails("render-search-result"); _clientSession = std::make_shared(emptyProtocolHandler, id, docBroker, getPublicUri(), isReadOnly, requestDetails); _clientSession->construct(); if (!_clientSession) return false; docBroker->setupTransfer(disposition, [docBroker] (std::shared_ptrconst & moveSocket) { docBroker->setResponseSocket(std::static_pointer_cast(moveSocket)); // First add and load the session. docBroker->addSession(docBroker->_clientSession); // Load the document manually. std::string encodedFrom; Poco::URI::encode(docBroker->getPublicUri().getPath(), "", encodedFrom); // add batch mode, no interactive dialogs const std::string _load = "load url=" + encodedFrom + " batch=true"; std::vector loadRequest(_load.begin(), _load.end()); docBroker->_clientSession->handleMessage(loadRequest); }); return true; } void RenderSearchResultBroker::setLoaded() { DocumentBroker::setLoaded(); // Send the rendersearchresult request ... const std::string renderSearchResultCmd = "rendersearchresult "; std::vector renderSearchResultRequest(renderSearchResultCmd.begin(), renderSearchResultCmd.end()); renderSearchResultRequest.resize(renderSearchResultCmd.size() + _pSearchResultContent->size()); std::copy(_pSearchResultContent->begin(), _pSearchResultContent->end(), renderSearchResultRequest.begin() + renderSearchResultCmd.size()); _clientSession->handleMessage(renderSearchResultRequest); } void RenderSearchResultBroker::dispose() { if (!_uriOrig.empty()) { gRenderSearchResultBrokerInstanceCouter--; removeFile(_uriOrig); _uriOrig.clear(); } } bool RenderSearchResultBroker::handleInput(const std::vector& payload) { bool bResult = DocumentBroker::handleInput(payload); if (bResult) { auto message = std::make_shared(payload.data(), payload.size(), Message::Dir::Out); auto const& messageData = message->data(); static std::string commandString = "rendersearchresult:"; static std::vector commandStringVector(commandString.begin(), commandString.end()); if (messageData.size() >= commandStringVector.size()) { bool bEquals = std::equal(commandStringVector.begin(), commandStringVector.end(), messageData.begin()); if (bEquals) { _aResposeData.resize(messageData.size() - commandStringVector.size()); std::copy(messageData.begin() + commandStringVector.size(), messageData.end(), _aResposeData.begin()); std::string aDataString(_aResposeData.data(), _aResposeData.size()); // really not ideal that the response works only with std::string http::Response httpResponse(http::StatusLine(200)); httpResponse.setBody(aDataString, "image/png"); httpResponse.set("Connection", "close"); _socket->sendAndShutdown(httpResponse); removeSession(_clientSession->getId()); stop("Finished RenderSearchResult handler."); } } } return bResult; } #endif std::vector> DocumentBroker::getSessionsTestOnlyUnsafe() { std::vector> result; for (auto& it : _sessions) result.push_back(it.second); return result; } void DocumentBroker::dumpState(std::ostream& os) { std::unique_lock lock(_mutex); uint64_t sent = 0, recv = 0; getIOStats(sent, recv); const auto now = std::chrono::steady_clock::now(); os << std::boolalpha; os << " Broker: " << COOLWSD::anonymizeUrl(_filename) << " pid: " << getPid(); if (_docState.isMarkedToDestroy()) os << " *** Marked to destroy ***"; else os << " has live sessions"; if (isLoaded()) os << "\n loaded in: " << _loadDuration; else os << "\n still loading... " << std::chrono::duration_cast(now - _threadStart); os << "\n sent: " << sent; os << "\n recv: " << recv; os << "\n jail id: " << _jailId; os << "\n filename: " << COOLWSD::anonymizeUrl(_filename); os << "\n public uri: " << _uriPublic.toString(); os << "\n jailed uri: " << COOLWSD::anonymizeUrl(_uriJailed); os << "\n doc key: " << _docKey; os << "\n doc id: " << _docId; os << "\n num sessions: " << _sessions.size(); os << "\n thread start: " << Util::getTimeForLog(now, _threadStart); os << "\n modified?: " << isModified(); os << "\n possibly-modified: " << isPossiblyModified(); os << "\n haveActivityAfterSaveRequest: " << haveActivityAfterSaveRequest(); os << "\n isViewFileExtension: " << _isViewFileExtension; if (_limitLifeSeconds > std::chrono::seconds::zero()) os << "\n life limit in seconds: " << _limitLifeSeconds.count(); os << "\n idle time: " << getIdleTimeSecs(); os << "\n cursor X: " << _cursorPosX << ", Y: " << _cursorPosY << ", W:" << _cursorWidth << ", H: " << _cursorHeight; os << "\n DocumentState:"; _docState.dumpState(os, "\n "); if (_docState.activity() == DocumentState::Activity::Rename) os << "\n (new name: " << _renameFilename << ')'; os << "\n SaveManager:"; _saveManager.dumpState(os, "\n "); os << "\n StorageManager:"; _storageManager.dumpState(os, "\n "); _lockCtx->dumpState(os); if (_tileCache) _tileCache->dumpState(os); _poll->dumpState(os); #if !MOBILEAPP // Bit nasty - need a cleaner way to dump state. os << "\n Sessions [" << _sessions.size() << "]:"; for (const auto &it : _sessions) { auto proto = it.second->getProtocol(); auto proxy = dynamic_cast(proto.get()); if (proxy) proxy->dumpProxyState(os); else std::static_pointer_cast(it.second)->dumpState(os); } #endif os << '\n'; } bool DocumentBroker::isAsyncUploading() const { if (!_storage) return false; StorageBase::AsyncUpload::State state = _storage->queryLocalFileToStorageAsyncUploadState().state(); return state == StorageBase::AsyncUpload::State::Running; } // not beautiful - but neither is editing mobile project files. #if MOBILEAPP # include "Exceptions.cpp" #endif /* vim:set shiftwidth=4 softtabstop=4 expandtab: */