| /* |
| * Copyright (c) 2022, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. Oracle designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Oracle in the LICENSE file that accompanied this code. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| #include "MsiCA.h" |
| #include "MsiDb.h" |
| #include "MsiUtils.h" |
| #include "FileUtils.h" |
| #include "ErrorHandling.h" |
| #include "Toolbox.h" |
| |
| |
| // Code in this file requires linking with msi.lib |
| |
| |
| namespace msi { |
| |
| tstring CAImpl::getProperty(const tstring& name) const { |
| return getPropertyFromCustomAction(handle, name); |
| } |
| |
| |
| void CAImpl::setProperty(const tstring& name, const tstring& value) { |
| if (value.empty()) { |
| JP_THROW(tstrings::any() << "Attempt to assign empty value to '" |
| << name << "' MSI property"); |
| } |
| |
| LOG_TRACE(tstrings::any() << "Setting MSI property '" << name << |
| "' to '" << value << "'"); |
| const UINT status = MsiSetProperty(handle, name.c_str(), value.c_str()); |
| if (status != ERROR_SUCCESS) { |
| JP_THROW(msi::Error(tstrings::any() << "MsiSetProperty(" << name |
| << ", " << value << ") failed", status)); |
| } |
| } |
| |
| |
| void CAImpl::removeProperty(const tstring& name) { |
| LOG_TRACE(tstrings::any() << "Removing MSI property '" << name << "'"); |
| const UINT status = MsiSetProperty(handle, name.c_str(), NULL); |
| if (status != ERROR_SUCCESS) { |
| JP_THROW(msi::Error(tstrings::any() << "MsiSetProperty(" << name |
| << ", NULL) failed", status)); |
| } |
| } |
| |
| |
| Guid CAFacade::getProductCode() const { |
| return impl.getProperty(_T("ProductCode")); |
| } |
| |
| |
| bool CAFacade::isInMode(MSIRUNMODE v) const { |
| return MsiGetMode(impl.getHandle(), v) != FALSE; |
| } |
| |
| |
| tstring CAFacade::getModes() const { |
| tstring modes; |
| // Iterate all modes in the range [MSIRUNMODE_ADMIN, MSIRUNMODE_COMMIT] |
| for (int mode = MSIRUNMODE_ADMIN; mode != MSIRUNMODE_COMMIT + 1; ++mode) { |
| modes.insert(modes.end(), isInMode(MSIRUNMODE(mode)) ? |
| _T('1') : _T('0')); |
| } |
| return modes; |
| } |
| |
| |
| void CAFacade::doAction(const tstring& name) const { |
| const UINT status = MsiDoAction(impl.getHandle(), name.c_str()); |
| if (status != ERROR_SUCCESS) { |
| JP_THROW(msi::Error(tstrings::any() << "MsiDoAction(" << name |
| << ") failed", status)); |
| } |
| } |
| |
| |
| tstring CAFacade::normalizeDirectoryPath(tstring v) { |
| if (v.empty()) { |
| return v; |
| } |
| std::replace(v.begin(), v.end(), '/', '\\'); |
| return FileUtils::removeTrailingSlash(v) + _T("\\"); |
| } |
| |
| |
| CA& CA::setPropertyIfEmpty(const tstring& name, const tstring& v) { |
| if (getProperty(name).empty()) { |
| setProperty(name, v); |
| } |
| return *this; |
| } |
| |
| |
| tstring DeferredCA::getArg() const { |
| if (isInMode(MSIRUNMODE_SCHEDULED) || caArgPropertyName.empty()) { |
| // Details on accessing MSI properties from deferred custom actions: |
| // http://blogs.technet.com/b/alexshev/archive/2008/03/25/property-does-not-exist-or-empty-when-accessed-from-deferred-custom-action.aspx |
| // http://stackoverflow.com/questions/17988392/unable-to-fetch-the-install-location-property-in-a-deferred-custom-action |
| // http://stackoverflow.com/questions/11233267/how-to-pass-customactiondata-to-a-customaction-using-wix |
| return impl.getProperty(_T("CustomActionData")); |
| } |
| |
| return impl.getProperty(caArgPropertyName); |
| } |
| |
| |
| tstring DeferredCA::getParsedArg(const tstring& name) const { |
| const auto entry = theParsedArgs.find(name); |
| if (entry == theParsedArgs.end()) { |
| JP_THROW(tstrings::any() << "Argument << '" << name |
| << "' not found."); |
| } |
| return entry->second; |
| } |
| |
| |
| namespace { |
| std::pair<tstring, tstring> parseArg(const tstring& v) { |
| const auto pos = v.find(_T('=')); |
| if (pos == tstring::npos) { |
| JP_THROW(tstrings::any() << "Missing expected '=' character in [" |
| << v << "] string."); |
| } |
| return std::pair<tstring, tstring>(v.substr(0, pos), v.substr(pos + 1)); |
| } |
| |
| void parseArgsImpl(DeferredCA::ArgsCtnr& dst, const tstring& src) { |
| const tstring_array pairs = tstrings::split(src, _T("*")); |
| for(auto it = pairs.begin(), end = pairs.end(); it != end; ++it) { |
| const auto pair = parseArg(*it); |
| dst[pair.first] = pair.second; |
| } |
| } |
| } // namespace |
| void DeferredCA::parseArgs(ArgsCtnr& dst, const tstring& src) { |
| DeferredCA::ArgsCtnr tmp; |
| |
| const auto end = src.find(_T("**")); |
| if (end != tstring::npos) { |
| parseArgsImpl(tmp, src.substr(0, end)); |
| tmp[tstring()] = src.substr(end + 2); |
| } else { |
| parseArgsImpl(tmp, src); |
| } |
| |
| tmp.insert(dst.begin(), dst.end()); |
| tmp.swap(dst); |
| } |
| |
| |
| MsiLogAppender::MsiLogAppender(MSIHANDLE h): handle(h), |
| ctorThread(GetCurrentThreadId()) { |
| |
| } |
| |
| void MsiLogAppender::append(const LogEvent& v) { |
| const LPCTSTR format = _T("[%02u:%02u:%02u.%03u%s%s:%u (%s)] %s: %s"); |
| |
| tstring ctxInfo = _T(" "); |
| if (v.tid != ctorThread) { |
| ctxInfo = (tstrings::any() << " (TID: " << v.tid << ") ").tstr(); |
| } |
| |
| const tstring buf = tstrings::unsafe_format(format, |
| unsigned(v.ts.wHour), unsigned(v.ts.wMinute), unsigned(v.ts.wSecond), unsigned(v.ts.wMilliseconds), // time |
| ctxInfo.c_str(), |
| v.fileName.c_str(), v.lineNum, v.funcName.c_str(), |
| v.logLevel.c_str(), |
| v.message.c_str()); |
| |
| DatabaseRecord r(1); |
| r.setString(0, _T("Java [1]")); |
| r.setString(1, buf); |
| |
| MsiProcessMessage(handle, INSTALLMESSAGE_INFO, r.getHandle()); |
| } |
| |
| |
| MsiLogTrigger::MsiLogTrigger(MSIHANDLE h): |
| msiLogAppender(h), |
| oldLogAppender(Logger::defaultLogger().getAppender()), |
| teeLogAppender(&msiLogAppender, &oldLogAppender) { |
| Logger::defaultLogger().setAppender(teeLogAppender); |
| } |
| |
| |
| MsiLogTrigger::~MsiLogTrigger() { |
| Logger::defaultLogger().setAppender(oldLogAppender); |
| } |
| |
| |
| |
| |
| namespace { |
| MSIHANDLE openDatabase(const CA& ca) { |
| MSIHANDLE h = MsiGetActiveDatabase(ca.getHandle()); |
| if (h == NULL) { |
| JP_THROW(Error(std::string("MsiGetActiveDatabase() failed"), |
| ERROR_FUNCTION_FAILED)); |
| } |
| return h; |
| } |
| |
| } // namespace |
| |
| Database::Database(const CA& ca): msiPath(_T("*CA*")), |
| dbHandle(openDatabase(ca)) { |
| } |
| |
| } // namespace msi |
