| OLD | NEW |
| (Empty) |
| 1 // Copyright 2009-2010 Google Inc. | |
| 2 // | |
| 3 // Licensed under the Apache License, Version 2.0 (the "License"); | |
| 4 // you may not use this file except in compliance with the License. | |
| 5 // You may obtain a copy of the License at | |
| 6 // | |
| 7 // http://www.apache.org/licenses/LICENSE-2.0 | |
| 8 // | |
| 9 // Unless required by applicable law or agreed to in writing, software | |
| 10 // distributed under the License is distributed on an "AS IS" BASIS, | |
| 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| 12 // See the License for the specific language governing permissions and | |
| 13 // limitations under the License. | |
| 14 // ======================================================================== | |
| 15 | |
| 16 #include "omaha/goopdate/app_version.h" | |
| 17 #include <atlsafe.h> | |
| 18 #include "omaha/base/debug.h" | |
| 19 #include "omaha/base/logging.h" | |
| 20 #include "omaha/base/scoped_ptr_address.h" | |
| 21 #include "omaha/base/synchronized.h" | |
| 22 #include "omaha/base/utils.h" | |
| 23 #include "omaha/goopdate/model.h" | |
| 24 | |
| 25 namespace omaha { | |
| 26 | |
| 27 AppVersion::AppVersion(App* app) | |
| 28 : ModelObject(app->model()), | |
| 29 app_(app) { | |
| 30 } | |
| 31 | |
| 32 // Destruction of App objects happens within the scope of their parent, | |
| 33 // which controls the locking. | |
| 34 AppVersion::~AppVersion() { | |
| 35 ASSERT1(model()->IsLockedByCaller()); | |
| 36 | |
| 37 for (size_t i = 0; i < packages_.size(); ++i) { | |
| 38 delete packages_[i]; | |
| 39 } | |
| 40 } | |
| 41 | |
| 42 CString AppVersion::version() const { | |
| 43 __mutexScope(model()->lock()); | |
| 44 return version_; | |
| 45 } | |
| 46 | |
| 47 void AppVersion::set_version(const CString& version) { | |
| 48 __mutexScope(model()->lock()); | |
| 49 version_ = version; | |
| 50 } | |
| 51 | |
| 52 App* AppVersion::app() { | |
| 53 __mutexScope(model()->lock()); | |
| 54 return app_; | |
| 55 } | |
| 56 | |
| 57 const App* AppVersion::app() const { | |
| 58 __mutexScope(model()->lock()); | |
| 59 return app_; | |
| 60 } | |
| 61 | |
| 62 // TODO(omaha3): It's unfortunate that the manifest can be empty. We need to | |
| 63 // make a copy anyway in UpdateResponse::BuildApp, so maybe this class | |
| 64 // should just expose a manifest object created in the constructor. On the | |
| 65 // other hand, current_version may not have a manifest, so this would be an | |
| 66 // empty object. Because the manifest must be created, AppData must friend | |
| 67 // InstallManager tests and other tests that need a manifest. This could | |
| 68 // probably be solved through mocking too. | |
| 69 const xml::InstallManifest* AppVersion::install_manifest() const { | |
| 70 __mutexScope(model()->lock()); | |
| 71 return install_manifest_.get(); | |
| 72 } | |
| 73 | |
| 74 void AppVersion::set_install_manifest(xml::InstallManifest* install_manifest) { | |
| 75 __mutexScope(model()->lock()); | |
| 76 ASSERT1(install_manifest); | |
| 77 install_manifest_.reset(install_manifest); | |
| 78 } | |
| 79 | |
| 80 size_t AppVersion::GetNumberOfPackages() const { | |
| 81 __mutexScope(model()->lock()); | |
| 82 return packages_.size(); | |
| 83 } | |
| 84 | |
| 85 HRESULT AppVersion::AddPackage(const CString& filename, | |
| 86 uint32 size, | |
| 87 const CString& hash) { | |
| 88 __mutexScope(model()->lock()); | |
| 89 Package* package = new Package(this); | |
| 90 package->SetFileInfo(filename, size, hash); | |
| 91 packages_.push_back(package); | |
| 92 return S_OK; | |
| 93 } | |
| 94 | |
| 95 Package* AppVersion::GetPackage(size_t index) { | |
| 96 __mutexScope(model()->lock()); | |
| 97 | |
| 98 if (index >= GetNumberOfPackages()) { | |
| 99 ASSERT1(false); | |
| 100 return NULL; | |
| 101 } | |
| 102 | |
| 103 return packages_[index]; | |
| 104 } | |
| 105 | |
| 106 const Package* AppVersion::GetPackage(size_t index) const { | |
| 107 __mutexScope(model()->lock()); | |
| 108 | |
| 109 if (index >= GetNumberOfPackages()) { | |
| 110 ASSERT1(false); | |
| 111 return NULL; | |
| 112 } | |
| 113 | |
| 114 return packages_[index]; | |
| 115 } | |
| 116 | |
| 117 const std::vector<CString>& AppVersion::download_base_urls() const { | |
| 118 __mutexScope(model()->lock()); | |
| 119 ASSERT1(!download_base_urls_.empty()); | |
| 120 return download_base_urls_; | |
| 121 } | |
| 122 | |
| 123 HRESULT AppVersion::AddDownloadBaseUrl(const CString& base_url) { | |
| 124 __mutexScope(model()->lock()); | |
| 125 ASSERT1(!base_url.IsEmpty()); | |
| 126 download_base_urls_.push_back(base_url); | |
| 127 return S_OK; | |
| 128 } | |
| 129 | |
| 130 // IAppVersion. | |
| 131 STDMETHODIMP AppVersion::get_version(BSTR* version) { | |
| 132 __mutexScope(model()->lock()); | |
| 133 ASSERT1(version); | |
| 134 *version = version_.AllocSysString(); | |
| 135 return S_OK; | |
| 136 } | |
| 137 | |
| 138 STDMETHODIMP AppVersion::get_packageCount(long* count) { // NOLINT | |
| 139 __mutexScope(model()->lock()); | |
| 140 | |
| 141 *count = GetNumberOfPackages(); | |
| 142 return S_OK; | |
| 143 } | |
| 144 | |
| 145 STDMETHODIMP AppVersion::get_package(long index, Package** package) { // NOLINT | |
| 146 __mutexScope(model()->lock()); | |
| 147 | |
| 148 if (index < 0 || static_cast<size_t>(index) >= GetNumberOfPackages()) { | |
| 149 return HRESULT_FROM_WIN32(ERROR_INVALID_INDEX); | |
| 150 } | |
| 151 | |
| 152 *package = GetPackage(index); | |
| 153 return S_OK; | |
| 154 } | |
| 155 | |
| 156 STDMETHODIMP AppVersionWrapper::get_version(BSTR* version) { | |
| 157 __mutexScope(model()->lock()); | |
| 158 return wrapped_obj()->get_version(version); | |
| 159 } | |
| 160 | |
| 161 STDMETHODIMP AppVersionWrapper::get_packageCount(long* count) { // NOLINT | |
| 162 __mutexScope(model()->lock()); | |
| 163 return wrapped_obj()->get_packageCount(count); | |
| 164 } | |
| 165 | |
| 166 STDMETHODIMP AppVersionWrapper::get_package(long index, // NOLINT | |
| 167 IDispatch** package) { | |
| 168 __mutexScope(model()->lock()); | |
| 169 | |
| 170 Package* p = NULL; | |
| 171 HRESULT hr = wrapped_obj()->get_package(index, &p); | |
| 172 if (FAILED(hr)) { | |
| 173 return hr; | |
| 174 } | |
| 175 | |
| 176 return PackageWrapper::Create(controlling_ptr(), p, package); | |
| 177 } | |
| 178 | |
| 179 HRESULT CopyAppVersionPackages(const AppVersion* app_version, | |
| 180 const CString& dir) { | |
| 181 ASSERT1(app_version); | |
| 182 | |
| 183 HRESULT hr(CreateDir(dir, NULL)); | |
| 184 if (FAILED(hr)) { | |
| 185 return hr; | |
| 186 } | |
| 187 | |
| 188 for (size_t i = 0; i != app_version->GetNumberOfPackages(); ++i) { | |
| 189 const Package* package = app_version->GetPackage(i); | |
| 190 ASSERT1(package); | |
| 191 if (package) { | |
| 192 hr = package->get(CComBSTR(dir)); | |
| 193 if (FAILED(hr)) { | |
| 194 CORE_LOG(LE, (_T("[Package::get failed][%s][%s][0x%x]"), | |
| 195 package->filename(), dir, hr)); | |
| 196 return hr; | |
| 197 } | |
| 198 } | |
| 199 } | |
| 200 | |
| 201 return S_OK; | |
| 202 } | |
| 203 | |
| 204 } // namespace omaha | |
| OLD | NEW |