Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 93065fba authored by Nikita Ioffe's avatar Nikita Ioffe Committed by Automerger Merge Worker
Browse files

Merge "Fix potential use-after-free bug in reboot" into rvc-dev am: 03ad5846 

Original change: https://googleplex-android-review.googlesource.com/c/platform/system/core/+/13144438

MUST ONLY BE SUBMITTED BY AUTOMERGER

Change-Id: Iec73de951c44463d72f9af3b236644c04c6de689
parents 4d4322a3 03ad5846

init/lmkd_service.cpp

+1 −1
Original line number Diff line number Diff line
@@ -79,7 +79,7 @@ static bool UnregisterProcess(pid_t pid) { 
}



static void RegisterServices(pid_t exclude_pid) {

    for (const auto& service : ServiceList::GetInstance().services()) {

    for (const auto& service : ServiceList::GetInstance()) {

        auto svc = service.get();

        if (svc->oom_score_adjust() != DEFAULT_OOM_SCORE_ADJUST) {

            // skip if process is excluded or not yet forked (pid==0)

init/reboot.cpp

+30 −24
Original line number Diff line number Diff line
@@ -85,12 +85,11 @@ static bool shutting_down = false; 


static const std::set<std::string> kDebuggingServices{"tombstoned", "logd", "adbd", "console"};



static std::vector<Service*> GetDebuggingServices(bool only_post_data) {

    std::vector<Service*> ret;

    ret.reserve(kDebuggingServices.size());

static std::set<std::string> GetPostDataDebuggingServices() {

    std::set<std::string> ret;

    for (const auto& s : ServiceList::GetInstance()) {

        if (kDebuggingServices.count(s->name()) && (!only_post_data || s->is_post_data())) {

            ret.push_back(s.get());

        if (kDebuggingServices.count(s->name()) && s->is_post_data()) {

            ret.insert(s->name());

        }

    }

    return ret;
@@ -503,13 +502,18 @@ static Result<void> KillZramBackingDevice() { 


// Stops given services, waits for them to be stopped for |timeout| ms.

// If terminate is true, then SIGTERM is sent to services, otherwise SIGKILL is sent.

static void StopServices(const std::vector<Service*>& services, std::chrono::milliseconds timeout,

// Note that services are stopped in order given by |ServiceList::services_in_shutdown_order|

// function.

static void StopServices(const std::set<std::string>& services, std::chrono::milliseconds timeout,

                         bool terminate) {

    LOG(INFO) << "Stopping " << services.size() << " services by sending "

              << (terminate ? "SIGTERM" : "SIGKILL");

    std::vector<pid_t> pids;

    pids.reserve(services.size());

    for (const auto& s : services) {

    for (const auto& s : ServiceList::GetInstance().services_in_shutdown_order()) {

        if (services.count(s->name()) == 0) {

            continue;

        }

        if (s->pid() > 0) {

            pids.push_back(s->pid());

        }
@@ -529,12 +533,12 @@ static void StopServices(const std::vector<Service*>& services, std::chrono::mil 


// Like StopServices, but also logs all the services that failed to stop after the provided timeout.

// Returns number of violators.

static int StopServicesAndLogViolations(const std::vector<Service*>& services,

static int StopServicesAndLogViolations(const std::set<std::string>& services,

                                        std::chrono::milliseconds timeout, bool terminate) {

    StopServices(services, timeout, terminate);

    int still_running = 0;

    for (const auto& s : services) {

        if (s->IsRunning()) {

    for (const auto& s : ServiceList::GetInstance()) {

        if (s->IsRunning() && services.count(s->name())) {

            LOG(ERROR) << "[service-misbehaving] : service '" << s->name() << "' is still running "

                       << timeout.count() << "ms after receiving "

                       << (terminate ? "SIGTERM" : "SIGKILL");
@@ -608,8 +612,7 @@ static void DoReboot(unsigned int cmd, const std::string& reason, const std::str 


    // watchdogd is a vendor specific component but should be alive to complete shutdown safely.

    const std::set<std::string> to_starts{"watchdogd"};

    std::vector<Service*> stop_first;

    stop_first.reserve(ServiceList::GetInstance().services().size());

    std::set<std::string> stop_first;

    for (const auto& s : ServiceList::GetInstance()) {

        if (kDebuggingServices.count(s->name())) {

            // keep debugging tools until non critical ones are all gone.
@@ -627,7 +630,7 @@ static void DoReboot(unsigned int cmd, const std::string& reason, const std::str 
                           << "': " << result.error();

            }

        } else {

            stop_first.push_back(s.get());

            stop_first.insert(s->name());

        }

    }
@@ -690,7 +693,7 @@ static void DoReboot(unsigned int cmd, const std::string& reason, const std::str 
        LOG(INFO) << "vold not running, skipping vold shutdown";

    }

    // logcat stopped here

    StopServices(GetDebuggingServices(false /* only_post_data */), 0ms, false /* SIGKILL */);

    StopServices(kDebuggingServices, 0ms, false /* SIGKILL */);

    // 4. sync, try umount, and optionally run fsck for user shutdown

    {

        Timer sync_timer;
@@ -779,17 +782,17 @@ static Result<void> DoUserspaceReboot() { 
        sub_reason = "resetprop";

        return Error() << "Failed to reset sys.powerctl property";

    }

    std::vector<Service*> stop_first;

    std::set<std::string> stop_first;

    // Remember the services that were enabled. We will need to manually enable them again otherwise

    // triggers like class_start won't restart them.

    std::vector<Service*> were_enabled;

    stop_first.reserve(ServiceList::GetInstance().services().size());

    std::set<std::string> were_enabled;

    for (const auto& s : ServiceList::GetInstance().services_in_shutdown_order()) {

        if (s->is_post_data() && !kDebuggingServices.count(s->name())) {

            stop_first.push_back(s);

            stop_first.insert(s->name());

        }

        // TODO(ioffe): we should also filter out temporary services here.

        if (s->is_post_data() && s->IsEnabled()) {

            were_enabled.push_back(s);

            were_enabled.insert(s->name());

        }

    }

    {
@@ -817,8 +820,9 @@ static Result<void> DoUserspaceReboot() { 
        sub_reason = "vold_reset";

        return result;

    }

    if (int r = StopServicesAndLogViolations(GetDebuggingServices(true /* only_post_data */),

                                             sigkill_timeout, false /* SIGKILL */);

    const auto& debugging_services = GetPostDataDebuggingServices();

    if (int r = StopServicesAndLogViolations(debugging_services, sigkill_timeout,

                                             false /* SIGKILL */);

        r > 0) {

        sub_reason = "sigkill_debug";

        // TODO(b/135984674): store information about offending services for debugging.
@@ -847,10 +851,12 @@ static Result<void> DoUserspaceReboot() { 
        return false;

    });

    // Re-enable services

    for (const auto& s : were_enabled) {

    for (const auto& s : ServiceList::GetInstance()) {

        if (were_enabled.count(s->name())) {

            LOG(INFO) << "Re-enabling service '" << s->name() << "'";

            s->Enable();

        }

    }

    ServiceList::GetInstance().ResetState();

    LeaveShutdown();

    ActionManager::GetInstance().QueueEventTrigger("userspace-reboot-resume");

init/service_list.h

+0 −1
Original line number Diff line number Diff line
@@ -66,7 +66,6 @@ class ServiceList { 


    auto begin() const { return services_.begin(); }

    auto end() const { return services_.end(); }

    const std::vector<std::unique_ptr<Service>>& services() const { return services_; }

    const std::vector<Service*> services_in_shutdown_order() const;



    void MarkPostData();

init/test_utils/service_utils.cpp

+1 −1
Original line number Diff line number Diff line
@@ -44,7 +44,7 @@ android::base::Result<ServiceInterfacesMap> GetOnDeviceServiceInterfacesMap() { 
    }



    ServiceInterfacesMap result;

    for (const auto& service : service_list.services()) {

    for (const auto& service : service_list) {

        // Create an entry for all services, including services that may not

        // have any declared interfaces.

        result[service->name()] = service->interfaces();