1 /*
2  * Copyright (c) 2021-2023 Huawei Device Co., Ltd.
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 #include "sdcard_update.h"
16 #include <chrono>
17 #include <dirent.h>
18 #include <fcntl.h>
19 #include <string>
20 #include <sys/mount.h>
21 #include <sys/stat.h>
22 #include <thread>
23 #include <unistd.h>
24 #include <vector>
25 #include "language/language_ui.h"
26 #include "log/dump.h"
27 #include "log/log.h"
28 #include "fs_manager/mount.h"
29 #include "securec.h"
30 #include "ui/updater_ui_stub.h"
31 #include "updater/updater_const.h"
32 #include "utils.h"
33 
34 namespace Updater {
GetSdcardPkgsPath(UpdaterParams & upParams)35 __attribute__((weak)) UpdaterStatus GetSdcardPkgsPath(UpdaterParams &upParams)
36 {
37     if (upParams.updatePackage.size() != 0) {
38         LOG(INFO) << "get sdcard packages from misc";
39         return UPDATE_SUCCESS;
40     }
41     LOG(INFO) << "get sdcard packages from sdcard path";
42     std::vector<std::string> sdcardPkgs = Utils::SplitString(SDCARD_CARD_PKG_PATH, ", ");
43     for (auto pkgPath : sdcardPkgs) {
44         if (access(pkgPath.c_str(), 0) == 0) {
45             LOG(INFO) << "find sdcard package : " << pkgPath;
46             upParams.updatePackage.push_back(pkgPath);
47         }
48     }
49     if (upParams.updatePackage.size() == 0) {
50         return UPDATE_ERROR;
51     }
52     return UPDATE_SUCCESS;
53 }
54 
GetSdcardPkgsFromDev(UpdaterParams & upParams)55 __attribute__((weak)) UpdaterStatus GetSdcardPkgsFromDev(UpdaterParams &upParams)
56 {
57     LOG(INFO) << "not implemented get sdcard pkgs from dev";
58     return UPDATE_ERROR;
59 }
60 
CheckPathNeedMountSD(UpdaterParams & upParams)61 bool CheckPathNeedMountSD(UpdaterParams &upParams)
62 {
63     for (auto pkgPath : upParams.updatePackage) {
64         if (pkgPath.find("/sdcard") != 0) {
65             return false;
66         }
67     }
68     return true;
69 }
70 
DoMountSdcard(std::vector<std::string> & sdcardStr,std::string & mountPoint)71 bool DoMountSdcard(std::vector<std::string> &sdcardStr, std::string &mountPoint)
72 {
73     bool mountSuccess = false;
74     unsigned int retryTimes = 20;
75     for (unsigned int retryCount = 1; retryCount <= retryTimes; retryCount++) {
76         LOG(INFO) << "the retry time is: " << retryCount;
77         for (auto item : sdcardStr) {
78             if (MountSdcard(item, mountPoint) == 0) {
79                 mountSuccess = true;
80                 LOG(INFO) << "mount " << item << " sdcard success!";
81                 break;
82             }
83         }
84         if (mountSuccess) {
85             break;
86         }
87         sleep(1); // sleep 1 second to wait for sd card recognition
88     }
89     return mountSuccess;
90 }
91 
CheckSdcardPkgs(UpdaterParams & upParams)92 UpdaterStatus CheckSdcardPkgs(UpdaterParams &upParams)
93 {
94 #ifndef UPDATER_UT
95     auto sdParam = "updater.data.configs";
96     Utils::SetParameter(sdParam, "1");
97     if (upParams.sdExtMode == SDCARD_UPDATE_FROM_DEV && GetSdcardPkgsFromDev(upParams) == UPDATE_SUCCESS) {
98         LOG(INFO) << "get sd card from dev succeed, skip get package from sd card";
99         return UPDATE_SUCCESS;
100     }
101 
102     if (GetSdcardInternalPkgs(upParams) == UPDATE_SUCCESS) {
103         LOG(INFO) << "get sdcard internal pkgs succeed";
104         return UPDATE_SUCCESS;
105     }
106 
107     std::string mountPoint = std::string(SDCARD_PATH);
108     std::vector<std::string> sdcardStr = GetBlockDevicesByMountPoint(mountPoint);
109     if (sdcardStr.empty()) {
110         UPDATER_UI_INSTANCE.ShowLog(
111             (errno == ENOENT) ? TR(LOG_SDCARD_NOTFIND) : TR(LOG_SDCARD_ABNORMAL), true);
112         return UPDATE_ERROR;
113     }
114     if (Utils::CheckUpdateMode(Updater::SDCARD_INTRAL_MODE)) {
115         if (MountForPath("/data") != 0) {
116             LOG(ERROR) << "data partition mount fail";
117             return UPDATE_ERROR;
118         }
119     }
120     if ((Utils::CheckUpdateMode(Updater::SDCARD_MODE) && !Utils::CheckUpdateMode(Updater::SDCARD_INTRAL_MODE)) ||
121         (Utils::CheckUpdateMode(Updater::SDCARD_INTRAL_MODE) && CheckPathNeedMountSD(upParams))) {
122             if (!DoMountSdcard(sdcardStr, mountPoint)) {
123                 LOG(ERROR) << "mount sdcard fail!";
124                 return UPDATE_ERROR;
125             }
126         }
127 #endif
128     if (GetSdcardPkgsPath(upParams) != UPDATE_SUCCESS) {
129         LOG(ERROR) << "there is no package in sdcard/updater, please check";
130         return UPDATE_ERROR;
131     }
132     return UPDATE_SUCCESS;
133 }
134 
GetSdcardInternalPkgs(UpdaterParams & upParams)135 __attribute__((weak)) UpdaterStatus GetSdcardInternalPkgs(UpdaterParams &upParams)
136 {
137     LOG(INFO) << "not implemented get normal update sdcard pkgs";
138     return UPDATE_ERROR;
139 }
140 } // Updater
141