init_reboot.c 6.9 KB
Newer Older
Z
zhong_ning 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
/*
 * Copyright (c) 2020 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "init_reboot.h"

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/mount.h>
#include <sys/prctl.h>
#include <sys/reboot.h>
S
sun_fan 已提交
26
#include "securec.h"
Z
zhong_ning 已提交
27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
#include "init_service.h"
#include "init_service_manager.h"
#include "init_log.h"

#define MAX_VALUE_LENGTH 500
#define MAX_COMMAND_SIZE 20
#define MAX_UPDATE_SIZE 100

struct RBMiscUpdateMessage {
    char command[MAX_COMMAND_SIZE];
    char update[MAX_UPDATE_SIZE];
};

static bool RBMiscWriteUpdaterMessage(const char *path, struct RBMiscUpdateMessage *boot)
{
S
sun_fan 已提交
42 43 44 45
    if (path == NULL || boot == NULL) {
        INIT_LOGE("path or boot is NULL.");
        return false;
    }
S
sun_fan 已提交
46 47 48 49 50
    char *realPath = realpath(path, NULL);
    if (realPath == NULL) {
        return false;
    }
    FILE* fp = fopen(realPath, "rb+");
Z
zhong_ning 已提交
51 52
    if (fp == NULL) {
        INIT_LOGE("open %s failed", path);
S
sun_fan 已提交
53 54
        free(realPath);
        realPath = NULL;
Z
zhong_ning 已提交
55 56 57 58 59
        return false;
    }

    size_t ret = fwrite(boot, sizeof(struct RBMiscUpdateMessage), 1, fp);
    if (ret < 0) {
Z
zhong_ning 已提交
60
        INIT_LOGE("write to misc failed");
S
sun_fan 已提交
61 62
        free(realPath);
        realPath = NULL;
Z
zhong_ning 已提交
63 64 65
        fclose(fp);
        return false;
    }
S
sun_fan 已提交
66 67
    free(realPath);
    realPath = NULL;
Z
zhong_ning 已提交
68 69 70 71 72 73
    fclose(fp);
    return true;
}

static bool RBMiscReadUpdaterMessage(const char *path, struct RBMiscUpdateMessage *boot)
{
S
sun_fan 已提交
74 75 76 77
    if (path == NULL || boot == NULL) {
        INIT_LOGE("path or boot is NULL.");
        return false;
    }
S
sun_fan 已提交
78 79 80 81 82
    char *realPath = realpath(path, NULL);
    if (realPath == NULL) {
        return false;
    }
    FILE* fp = fopen(realPath, "rb");
Z
zhong_ning 已提交
83 84
    if (fp == NULL) {
        INIT_LOGE("open %s failed", path);
S
sun_fan 已提交
85 86
        free(realPath);
        realPath = NULL;
Z
zhong_ning 已提交
87 88 89 90 91 92
        return false;
    }

    size_t ret = fread(boot, 1, sizeof(struct RBMiscUpdateMessage), fp);
    if (ret <= 0) {
        INIT_LOGE("read to misc failed");
S
sun_fan 已提交
93 94
        free(realPath);
        realPath = NULL;
Z
zhong_ning 已提交
95 96 97
        fclose(fp);
        return false;
    }
S
sun_fan 已提交
98 99
    free(realPath);
    realPath = NULL;
Z
zhong_ning 已提交
100 101 102 103 104 105
    fclose(fp);
    return true;
}

static int GetMountStatusForMountPoint(const char *mountPoint)
{
S
sun_fan 已提交
106 107
    const int bufferMaxSize = 512;
    char buffer[bufferMaxSize];
Z
zhong_ning 已提交
108 109 110 111
    size_t n;
    const char *mountFile = "/proc/mounts";
    FILE *fp = fopen(mountFile, "r");
    if (fp == NULL) {
Z
zhong_ning 已提交
112
        INIT_LOGE("DoReboot %s can't open.", mountPoint);
Z
zhong_ning 已提交
113 114 115
        return 1;
    }

S
sun_fan 已提交
116
    while (fgets(buffer, sizeof(buffer) - 1, fp) != NULL) {
Z
zhong_ning 已提交
117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
        n = strlen(buffer);
        if (buffer[n - 1] == '\n') {
            buffer[n - 1] = '\0';
        }
        if (strstr(buffer, mountPoint) != NULL) {
            fclose(fp);
            return 1;
        }
    }

    // Cannot find it from system.
    fclose(fp);
    return 0;
}

132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
static int UpdateUpdaterStatus(const char *valueData)
{
    const char *miscFile = "/dev/block/platform/soc/10100000.himci.eMMC/by-name/misc";
    struct RBMiscUpdateMessage msg;
    bool ret = RBMiscReadUpdaterMessage(miscFile, &msg);
    if (!ret) {
        INIT_LOGE("RBMiscReadUpdaterMessage error.");
        return -1;
    }
    if (snprintf_s(msg.command, MAX_COMMAND_SIZE, MAX_COMMAND_SIZE - 1, "%s", "boot_updater") == -1) {
        INIT_LOGE("RBMiscWriteUpdaterMessage error");
        return -1;
    }
    if (strlen(valueData) > strlen("updater:") && strncmp(valueData, "updater:", strlen("updater:")) == 0) {
        if (snprintf_s(msg.update, MAX_UPDATE_SIZE, MAX_UPDATE_SIZE - 1, "%s", valueData + strlen("updater:")) == -1) {
            INIT_LOGE("RBMiscWriteUpdaterMessage error");
            return -1;
        }
        ret = RBMiscWriteUpdaterMessage(miscFile, &msg);
Z
zhong_ning 已提交
151
        if (ret != true) {
152 153 154 155 156
            INIT_LOGE("RBMiscWriteUpdaterMessage error");
            return -1;
        }
    } else if (strlen(valueData) == strlen("updater") && strncmp(valueData, "updater", strlen("updater")) == 0) {
        ret = RBMiscWriteUpdaterMessage(miscFile, &msg);
Z
zhong_ning 已提交
157
        if (ret != true) {
158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186
            INIT_LOGE("RBMiscWriteUpdaterMessage error");
            return -1;
        }
    } else {
        return -1;
    }
    return 0;
}

static int DoRebootCore(const char *valueData)
{
    if (valueData == NULL) {
        reboot(RB_AUTOBOOT);
        return 0;
    } else if (strncmp(valueData, "shutdown", strlen("shutdown")) == 0) {
        reboot(RB_POWER_OFF);
        return 0;
    } else if (strncmp(valueData, "updater", strlen("updater")) == 0) {
        int ret = UpdateUpdaterStatus(valueData);
        if (ret == 0) {
            reboot(RB_AUTOBOOT);
            return 0;
        }
    } else {
        return -1;
    }
    return 0;
}

Z
zhong_ning 已提交
187 188 189
void DoReboot(const char *value)
{
    if (value == NULL) {
Z
zhong_ning 已提交
190
        INIT_LOGE("DoReboot value = NULL");
Z
zhong_ning 已提交
191 192
        return;
    }
Z
zhong_ning 已提交
193
    INIT_LOGI("DoReboot value = %s", value);
Z
zhong_ning 已提交
194

195
    if (strlen(value) > MAX_VALUE_LENGTH || strlen(value) < strlen("reboot") || strlen(value) == strlen("reboot,")) {
Z
zhong_ning 已提交
196
        INIT_LOGE("DoReboot reboot value error, value = %s.", value);
Z
zhong_ning 已提交
197 198
        return;
    }
S
sun_fan 已提交
199
    const char *valueData = NULL;
200 201 202 203 204
    if (strncmp(value, "reboot,", strlen("reboot,")) == 0) {
        valueData = value + strlen("reboot,");
    } else if (strlen(value) < strlen("reboot,") && strncmp(value, "reboot", strlen("reboot")) == 0) {
        valueData = NULL;
    } else {
Z
zhong_ning 已提交
205
        INIT_LOGE("DoReboot reboot value = %s, must started with reboot ,error.", value);
S
sun_fan 已提交
206 207
        return;
    }
Z
zhong_ning 已提交
208 209 210
    if (valueData != NULL && strncmp(valueData, "shutdown", strlen("shutdown")) != 0 &&
        strncmp(valueData, "updater:", strlen("updater:")) != 0 &&
        strncmp(valueData, "updater", strlen("updater")) != 0) {
Z
zhong_ning 已提交
211
        INIT_LOGE("DoReboot value = %s, parameters error.", value);
Z
fix bug  
zhong_ning 已提交
212 213 214
        return;
    }
    StopAllServicesBeforeReboot();
S
sun_fan 已提交
215
    if (GetMountStatusForMountPoint("/vendor") != 0) {
Z
zhong_ning 已提交
216
        if (umount("/vendor") != 0) {
Z
zhong_ning 已提交
217
            INIT_LOGE("DoReboot umount vendor failed! errno = %d.", errno);
Z
zhong_ning 已提交
218 219
        }
    }
S
sun_fan 已提交
220
    if (GetMountStatusForMountPoint("/data") != 0) {
Z
zhong_ning 已提交
221
        if (umount("/data") != 0) {
Z
zhong_ning 已提交
222
            INIT_LOGE("DoReboot umount data failed! errno = %d.", errno);
Z
zhong_ning 已提交
223 224
        }
    }
225 226 227
    int ret = DoRebootCore(valueData);
    if (ret != 0) {
        INIT_LOGE("DoReboot value = %s, error.", value);
Z
zhong_ning 已提交
228 229 230 231 232
    }
    return;

}