2276402292

Committed 21 Jan 2026 11:56AM UTC coverage: 79.855%. First build

Build # 2276402292

Build Type

push

gitlab-ci

Committed by

michalkopczan

Commit Message

feat: Handle HTTP 429 Too Many Requests in deployment polling

Ticket: MEN-8850
Changelog: Title

Signed-off-by: Michal Kopczan <michal.kopczan@northern.tech>

Run Details

30 of 35 new or added lines in 3 files covered. (85.71%)

7932 of 9933 relevant lines covered (79.86%)

13852.16 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

79.17

/src/mender-update/daemon/states.cpp

// Copyright 2023 Northern.tech AS
//
//    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 <mender-update/daemon/states.hpp>

#include <client_shared/conf.hpp>
#include <common/events_io.hpp>
#include <common/log.hpp>
#include <common/path.hpp>

#include <mender-update/daemon/context.hpp>
#include <mender-update/inventory.hpp>

namespace mender {
namespace update {
namespace daemon {

namespace conf = mender::client_shared::conf;
namespace error = mender::common::error;
namespace events = mender::common::events;
namespace kv_db = mender::common::key_value_database;
namespace path = mender::common::path;
namespace log = mender::common::log;

namespace main_context = mender::update::context;
namespace inventory = mender::update::inventory;

class DefaultStateHandler {
public:
        void operator()(const error::Error &err) {
                if (err != error::NoError) {
                        log::Error(err.String());
                        poster.PostEvent(StateEvent::Failure);
                        return;
                }
                poster.PostEvent(StateEvent::Success);
        }

        sm::EventPoster<StateEvent> &poster;
};

static void DefaultAsyncErrorHandler(sm::EventPoster<StateEvent> &poster, const error::Error &err) {
        if (err != error::NoError) {
                log::Error(err.String());
                poster.PostEvent(StateEvent::Failure);
        }
}

void EmptyState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        // Keep this state truly empty.
}

void InitState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        // I will never run - just a placeholder to start the state-machine at
        poster.PostEvent(StateEvent::Started); // Start the state machine
}

void StateScriptState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        string state_name {script_executor::Name(this->state_, this->action_)};
        log::Debug("Executing the  " + state_name + " State Scripts...");
        auto err = this->script_.AsyncRunScripts(
                this->state_,
                this->action_,
                [state_name, &poster](error::Error err) {
                        if (err != error::NoError) {
                                log::Error(
                                        "Received error: (" + err.String() + ") when running the State Script scripts "
                                        + state_name);
                                poster.PostEvent(StateEvent::Failure);
                                return;
                        }
                        log::Debug("Successfully ran the " + state_name + " State Scripts...");
                        poster.PostEvent(StateEvent::Success);
                },
                this->on_error_);

        if (err != error::NoError) {
                log::Error(
                        "Failed to schedule the state script execution for: " + state_name
                        + " got error: " + err.String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }
}


void SaveStateScriptState::OnEnterSaveState(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        return state_script_state_.OnEnter(ctx, poster);
}

void IdleState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering Idle state");
}

ScheduleNextPollState::ScheduleNextPollState(
        events::Timer &timer, const string &poll_action, const StateEvent event, int interval) :
        timer_ {timer},
        poll_action_ {poll_action},
        event_ {event},
        interval_ {interval} {
}

void ScheduleNextPollState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Scheduling the next " + poll_action_ + " in: " + to_string(interval_) + " seconds");
        timer_.AsyncWait(chrono::seconds(interval_), [this, &poster](error::Error err) {
                if (err != error::NoError) {
                        if (err.code != make_error_condition(errc::operation_canceled)) {
                                log::Error("Timer caused error: " + err.String());
                        }
                } else {
                        poster.PostEvent(event_);
                }
        });

        poster.PostEvent(StateEvent::Success);
}

SubmitInventoryState::SubmitInventoryState(int retry_interval_seconds, int retry_count) :
        backoff_ {chrono::seconds(retry_interval_seconds), retry_count} {
}

void SubmitInventoryState::HandlePollingError(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        // When using short polling intervals, we should adjust the backoff to ensure
        // that the intervals do not exceed the maximum retry polling interval, which
        // converts the backoff to a fixed interval.
        chrono::milliseconds max_interval =
                chrono::seconds(ctx.mender_context.GetConfig().retry_poll_interval_seconds);
        if (max_interval < backoff_.SmallestInterval()) {
                backoff_.SetSmallestInterval(max_interval);
                backoff_.SetMaxInterval(max_interval);
        }
        auto exp_interval = backoff_.NextInterval();
        if (!exp_interval) {
                log::Debug(
                        "Not retrying with backoff, retrying InventoryPollIntervalSeconds: "
                        + exp_interval.error().String());
                return;
        }
        log::Info(
                "Retrying inventory polling in "
                + to_string(chrono::duration_cast<chrono::seconds>(*exp_interval).count()) + " seconds");

        ctx.inventory_timer.Cancel();
        ctx.inventory_timer.AsyncWait(*exp_interval, [&poster](error::Error err) {
                if (err != error::NoError) {
                        if (err.code != make_error_condition(errc::operation_canceled)) {
                                log::Error("Retry poll timer caused error: " + err.String());
                        }
                } else {
                        poster.PostEvent(StateEvent::InventoryPollingTriggered);
                }
        });
}

void SubmitInventoryState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Submitting inventory");

        auto handler = [this, &ctx, &poster](error::Error err) {
                if (err != error::NoError) {
                        log::Error("Failed to submit inventory: " + err.String());
                        // Replace the inventory poll timer with a backoff
                        HandlePollingError(ctx, poster);
                        poster.PostEvent(StateEvent::Failure);
                        return;
                }
                backoff_.Reset();
                ctx.inventory_client->has_submitted_inventory = true;
                poster.PostEvent(StateEvent::Success);
        };

        auto err = ctx.inventory_client->PushData(
                ctx.mender_context.GetConfig().paths.GetInventoryScriptsDir(),
                ctx.event_loop,
                ctx.http_client,
                handler);

        if (err != error::NoError) {
                // This is the only case the handler won't be called for us by
                // PushData() (see inventory::PushInventoryData()).
                handler(err);
        }
}

PollForDeploymentState::PollForDeploymentState(int retry_interval_seconds, int retry_count) :
        backoff_ {chrono::seconds(retry_interval_seconds), retry_count} {
}

void PollForDeploymentState::HandlePollingError(
        Context &ctx,
        sm::EventPoster<StateEvent> &poster,
        mender::update::deployments::CheckUpdatesAPIResponseError response) {
        // When using short polling intervals, we should adjust the backoff to ensure
        // that the intervals do not exceed the maximum retry polling interval, which
        // converts the backoff to a fixed interval.
        chrono::milliseconds max_interval =
                chrono::seconds(ctx.mender_context.GetConfig().retry_poll_interval_seconds);
        if (max_interval < backoff_.SmallestInterval()) {
                backoff_.SetSmallestInterval(max_interval);
                backoff_.SetMaxInterval(max_interval);
        }

        chrono::milliseconds interval;
        bool retry_after_defined {false};

        if (response.code.has_value() && response.code.value() == http::StatusTooManyRequests
                && response.headers.has_value()) {
                auto retry_after_header = response.headers.value().find("Retry-After");
                if (retry_after_header != response.headers.value().end()) {
                        auto exp_interval = http::HTTPDateToUnixTimestampFromNow(retry_after_header->second);
                        if (exp_interval.has_value()) {
                                interval = exp_interval.value();
                                retry_after_defined = true;
                        } else {
                                log::Debug("Could not get the Retry-After value from HTTP response");
                        }
                }
        }

        if (!retry_after_defined) {
                auto exp_interval = backoff_.NextInterval();
                if (!exp_interval) {
                        log::Debug(
                                "Not retrying with backoff, retrying with UpdatePollIntervalSeconds: "
                                + exp_interval.error().String());
                        return;
                }
                interval = exp_interval.value();
        }
        log::Info(
                "Retrying deployment polling in "
                + to_string(chrono::duration_cast<chrono::seconds>(interval).count()) + " seconds");

        ctx.deployment_timer.Cancel();
        ctx.deployment_timer.AsyncWait(interval, [&poster](error::Error err) {
                if (err != error::NoError) {
                        if (err.code != make_error_condition(errc::operation_canceled)) {
                                log::Error("Retry poll timer caused error: " + err.String());
                        }
                } else {
                        poster.PostEvent(StateEvent::DeploymentPollingTriggered);
                }
        });
}

void PollForDeploymentState::CheckNewDeploymentsHandler(
        Context &ctx,
        sm::EventPoster<StateEvent> &poster,
        mender::update::deployments::CheckUpdatesAPIResponse response) {
        if (!response) {
                log::Error("Error while polling for deployment: " + response.error().error.String());
                // Replace the update poll timer with a backoff
                HandlePollingError(ctx, poster, response.error());
                poster.PostEvent(StateEvent::Failure);
                return;
        } else if (!response.value()) {
                log::Info("No update available");
                poster.PostEvent(StateEvent::NothingToDo);
                if (not ctx.inventory_client->has_submitted_inventory) {
                        // If we have not submitted inventory successfully at least
                        // once, schedule this after receiving a successful response
                        // with no update. This enables inventory to be submitted
                        // immediately after the device has been accepted. If there
                        // is an update available, an inventory update will be
                        // scheduled at the end of it unconditionally.
                        poster.PostEvent(StateEvent::InventoryPollingTriggered);
                }

                backoff_.Reset();
                return;
        }
        backoff_.Reset();

        auto exp_data = ApiResponseJsonToStateData(response.value().value());
        if (!exp_data) {
                log::Error("Error in API response: " + exp_data.error().String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }

        // Make a new set of update data.
        ctx.deployment.state_data.reset(new StateData(std::move(exp_data.value())));

        ctx.BeginDeploymentLogging();

        // This is a duplicate message to one logged when mender-update
        // starts, but this one goes into the deployment log.
        log::Info("Running mender-update " + conf::kMenderVersion);
        log::Info("Deployment with ID " + ctx.deployment.state_data->update_info.id + " started.");

        poster.PostEvent(StateEvent::DeploymentStarted);
        poster.PostEvent(StateEvent::Success);
}

void PollForDeploymentState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Polling for update");

        auto err = ctx.deployment_client->CheckNewDeployments(
                ctx.mender_context,
                ctx.http_client,
                [this, &ctx, &poster](mender::update::deployments::CheckUpdatesAPIResponse response) {
                        this->CheckNewDeploymentsHandler(ctx, poster, response);
                });

        if (err != error::NoError) {
                log::Error("Error when trying to poll for deployment: " + err.String());
                poster.PostEvent(StateEvent::Failure);
        }
}

void SaveState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        assert(ctx.deployment.state_data);

        ctx.deployment.state_data->state = DatabaseStateString();

        log::Trace("Storing deployment state in the DB (database-string): " + DatabaseStateString());

        auto err = ctx.SaveDeploymentStateData(*ctx.deployment.state_data);
        if (err != error::NoError) {
                log::Error(err.String());
                if (err.code
                        == main_context::MakeError(main_context::StateDataStoreCountExceededError, "").code) {
                        poster.PostEvent(StateEvent::StateLoopDetected);
                        return;
                } else if (!IsFailureState()) {
                        // Non-failure states should be interrupted, but failure states should be
                        // allowed to do their work, even if a database error was detected.
                        poster.PostEvent(StateEvent::Failure);
                        return;
                }
        }

        OnEnterSaveState(ctx, poster);
}

void UpdateDownloadState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering Download state");

        auto req = make_shared<http::OutgoingRequest>();
        req->SetMethod(http::Method::GET);
        auto err = req->SetAddress(ctx.deployment.state_data->update_info.artifact.source.uri);
        if (err != error::NoError) {
                log::Error(err.String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }

        err = ctx.download_client->AsyncCall(
                req,
                [&ctx, &poster](http::ExpectedIncomingResponsePtr exp_resp) {
                        if (!exp_resp) {
                                log::Error("Unexpected error during download: " + exp_resp.error().String());
                                poster.PostEvent(StateEvent::Failure);
                                return;
                        }

                        auto &resp = exp_resp.value();
                        if (resp->GetStatusCode() != http::StatusOK) {
                                log::Error(
                                        "Unexpected status code while fetching artifact: " + resp->GetStatusMessage());
                                poster.PostEvent(StateEvent::Failure);
                                return;
                        }

                        auto http_reader = resp->MakeBodyAsyncReader();
                        if (!http_reader) {
                                log::Error(http_reader.error().String());
                                poster.PostEvent(StateEvent::Failure);
                                return;
                        }
                        ctx.deployment.artifact_reader =
                                make_shared<events::io::ReaderFromAsyncReader>(ctx.event_loop, http_reader.value());
                        ParseArtifact(ctx, poster);
                },
                [](http::ExpectedIncomingResponsePtr exp_resp) {
                        if (!exp_resp) {
                                log::Error(exp_resp.error().String());
                                // Cannot handle error here, because this handler is called at the
                                // end of the download, when we have already left this state. So
                                // rely on this error being propagated through the BodyAsyncReader
                                // above instead.
                                return;
                        }
                });

        if (err != error::NoError) {
                log::Error(err.String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }
}

void UpdateDownloadState::ParseArtifact(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        string art_scripts_path = ctx.mender_context.GetConfig().paths.GetArtScriptsPath();

        // Clear the artifact scripts directory so we don't risk old scripts lingering.
        auto err = path::DeleteRecursively(art_scripts_path);
        if (err != error::NoError) {
                log::Error("When preparing to parse artifact: " + err.String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }

        artifact::config::ParserConfig config {
                .artifact_scripts_filesystem_path = art_scripts_path,
                .artifact_scripts_version = 3,
                .artifact_verify_keys = ctx.mender_context.GetConfig().artifact_verify_keys,
        };
        auto exp_parser = artifact::Parse(*ctx.deployment.artifact_reader, config);
        if (!exp_parser) {
                log::Error(exp_parser.error().String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }
        ctx.deployment.artifact_parser.reset(new artifact::Artifact(std::move(exp_parser.value())));

        auto exp_header = artifact::View(*ctx.deployment.artifact_parser, 0);
        if (!exp_header) {
                log::Error(exp_header.error().String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }
        auto &header = exp_header.value();

        auto exp_matches = ctx.mender_context.MatchesArtifactDepends(header.header);
        if (!exp_matches) {
                log::Error(exp_matches.error().String());
                poster.PostEvent(StateEvent::Failure);
                return;
        } else if (!exp_matches.value()) {
                // reasons already logged
                poster.PostEvent(StateEvent::Failure);
                return;
        }

        log::Info("Installing artifact...");

        ctx.deployment.state_data->FillUpdateDataFromArtifact(header);

        ctx.deployment.state_data->state = Context::kUpdateStateDownload;

        assert(ctx.deployment.state_data->update_info.artifact.payload_types.size() == 1);

        // Initial state data save, now that we have enough information from the artifact.
        err = ctx.SaveDeploymentStateData(*ctx.deployment.state_data);
        if (err != error::NoError) {
                log::Error(err.String());
                if (err.code
                        == main_context::MakeError(main_context::StateDataStoreCountExceededError, "").code) {
                        poster.PostEvent(StateEvent::StateLoopDetected);
                        return;
                } else {
                        poster.PostEvent(StateEvent::Failure);
                        return;
                }
        }

        if (header.header.payload_type == "") {
                // Empty-payload-artifact, aka "bootstrap artifact".
                poster.PostEvent(StateEvent::NothingToDo);
                return;
        }

        auto exp_update_module =
                update_module::UpdateModule::Create(ctx.mender_context, header.header.payload_type);
        if (!exp_update_module.has_value()) {
                log::Error(
                        "Error creating an Update Module when parsing artifact: "
                        + exp_update_module.error().String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }
        ctx.deployment.update_module = std::move(exp_update_module.value());

        err = ctx.deployment.update_module->CleanAndPrepareFileTree(
                ctx.deployment.update_module->GetUpdateModuleWorkDir(), header);
        if (err != error::NoError) {
                log::Error(err.String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }

        err = ctx.deployment.update_module->AsyncProvidePayloadFileSizes(
                ctx.event_loop, [&ctx, &poster](expected::ExpectedBool download_with_sizes) {
                        if (!download_with_sizes.has_value()) {
                                log::Error(download_with_sizes.error().String());
                                poster.PostEvent(StateEvent::Failure);
                                return;
                        }
                        ctx.deployment.download_with_sizes = download_with_sizes.value();
                        DoDownload(ctx, poster);
                });

        if (err != error::NoError) {
                log::Error(err.String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }
}

void UpdateDownloadState::DoDownload(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        auto exp_payload = ctx.deployment.artifact_parser->Next();
        if (!exp_payload) {
                log::Error(exp_payload.error().String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }
        ctx.deployment.artifact_payload.reset(new artifact::Payload(std::move(exp_payload.value())));

        auto handler = [&poster, &ctx](error::Error err) {
                if (err != error::NoError) {
                        log::Error(err.String());
                        poster.PostEvent(StateEvent::Failure);
                        return;
                }

                auto exp_payload = ctx.deployment.artifact_parser->Next();
                if (exp_payload) {
                        log::Error("Multiple payloads are not yet supported in daemon mode.");
                        poster.PostEvent(StateEvent::Failure);
                        return;
                } else if (
                        exp_payload.error().code
                        != artifact::parser_error::MakeError(artifact::parser_error::EOFError, "").code) {
                        log::Error(exp_payload.error().String());
                        poster.PostEvent(StateEvent::Failure);
                        return;
                }

                poster.PostEvent(StateEvent::Success);
        };

        if (ctx.deployment.download_with_sizes) {
                ctx.deployment.update_module->AsyncDownloadWithFileSizes(
                        ctx.event_loop, *ctx.deployment.artifact_payload, handler);
        } else {
                ctx.deployment.update_module->AsyncDownload(
                        ctx.event_loop, *ctx.deployment.artifact_payload, handler);
        }
}

void UpdateDownloadCancelState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering DownloadCancel state");
        ctx.download_client->Cancel();
        poster.PostEvent(StateEvent::Success);
}

SendStatusUpdateState::SendStatusUpdateState(optional<deployments::DeploymentStatus> status) :
        status_(status),
        mode_(FailureMode::Ignore) {
}

SendStatusUpdateState::SendStatusUpdateState(
        optional<deployments::DeploymentStatus> status,
        events::EventLoop &event_loop,
        int retry_interval_seconds,
        int retry_count) :
        status_(status),
        mode_(FailureMode::RetryThenFail),
        retry_(Retry {
                http::ExponentialBackoff(chrono::seconds(retry_interval_seconds), retry_count),
                event_loop}) {
}

void SendStatusUpdateState::SetSmallestWaitInterval(chrono::milliseconds interval) {
        if (retry_) {
                retry_->backoff.SetSmallestInterval(interval);
        }
}

void SendStatusUpdateState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        // Reset this every time we enter the state, which means a new round of retries.
        if (retry_) {
                retry_->backoff.Reset();
        }

        DoStatusUpdate(ctx, poster);
}

void SendStatusUpdateState::DoStatusUpdate(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        assert(ctx.deployment_client);
        assert(ctx.deployment.state_data);

        log::Info("Sending status update to server");

        auto result_handler = [this, &ctx, &poster](const error::Error &err) {
                if (err != error::NoError) {
                        log::Error("Could not send deployment status: " + err.String());

                        if (err.code == deployments::MakeError(deployments::DeploymentAbortedError, "").code) {
                                // If the deployment was aborted upstream it is an immediate
                                // failure, even if retry is enabled.
                                poster.PostEvent(StateEvent::DeploymentAborted);
                                return;
                        }

                        switch (mode_) {
                        case FailureMode::Ignore:
                                break;
                        case FailureMode::RetryThenFail:

                                auto exp_interval = retry_->backoff.NextInterval();
                                if (!exp_interval) {
                                        log::Error(
                                                "Giving up on sending status updates to server: "
                                                + exp_interval.error().String());
                                        poster.PostEvent(StateEvent::Failure);
                                        return;
                                }

                                log::Info(
                                        "Retrying status update after "
                                        + to_string(chrono::duration_cast<chrono::seconds>(*exp_interval).count())
                                        + " seconds");
                                retry_->wait_timer.AsyncWait(
                                        *exp_interval, [this, &ctx, &poster](error::Error err) {
                                                // Error here is quite unexpected (from a timer), so treat
                                                // this as an immediate error, despite Retry flag.
                                                if (err != error::NoError) {
                                                        log::Error(
                                                                "Unexpected error in SendStatusUpdateState wait timer: "
                                                                + err.String());
                                                        poster.PostEvent(StateEvent::Failure);
                                                        return;
                                                }

                                                // Try again. Since both status and logs are sent
                                                // from here, there's a chance this might resubmit
                                                // the status, but there's no harm in it, and it
                                                // won't happen often.
                                                DoStatusUpdate(ctx, poster);
                                        });
                                return;
                        }
                }

                poster.PostEvent(StateEvent::Success);
        };

        deployments::DeploymentStatus status;
        if (status_) {
                status = status_.value();
        } else {
                // If nothing is specified, grab success/failure status from the deployment status.
                if (ctx.deployment.failed) {
                        status = deployments::DeploymentStatus::Failure;
                } else {
                        status = deployments::DeploymentStatus::Success;
                }
        }

        // Push status.
        log::Debug("Pushing deployment status: " + DeploymentStatusString(status));
        auto err = ctx.deployment_client->PushStatus(
                ctx.deployment.state_data->update_info.id,
                status,
                "",
                ctx.http_client,
                [result_handler, &ctx](error::Error err) {
                        // If there is an error, we don't submit logs now, but call the handler,
                        // which may schedule a retry later. If there is no error, and the
                        // deployment as a whole was successful, then also call the handler here,
                        // since we don't need to submit logs at all then.
                        if (err != error::NoError || !ctx.deployment.failed) {
                                result_handler(err);
                                return;
                        }

                        // Push logs.
                        err = ctx.deployment_client->PushLogs(
                                ctx.deployment.state_data->update_info.id,
                                ctx.deployment.logger->LogFilePath(),
                                ctx.http_client,
                                result_handler);

                        if (err != error::NoError) {
                                result_handler(err);
                        }
                });

        if (err != error::NoError) {
                result_handler(err);
        }

        // No action, wait for reply from status endpoint.
}

void UpdateInstallState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering ArtifactInstall state");

        DefaultAsyncErrorHandler(
                poster,
                ctx.deployment.update_module->AsyncArtifactInstall(
                        ctx.event_loop, DefaultStateHandler {poster}));
}

void UpdateCheckRebootState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        DefaultAsyncErrorHandler(
                poster,
                ctx.deployment.update_module->AsyncNeedsReboot(
                        ctx.event_loop, [&ctx, &poster](update_module::ExpectedRebootAction reboot_action) {
                                if (!reboot_action.has_value()) {
                                        log::Error(reboot_action.error().String());
                                        poster.PostEvent(StateEvent::Failure);
                                        return;
                                }

                                ctx.deployment.state_data->update_info.reboot_requested.resize(1);
                                ctx.deployment.state_data->update_info.reboot_requested[0] =
                                        NeedsRebootToDbString(*reboot_action);
                                switch (*reboot_action) {
                                case update_module::RebootAction::No:
                                        poster.PostEvent(StateEvent::NothingToDo);
                                        break;
                                case update_module::RebootAction::Yes:
                                case update_module::RebootAction::Automatic:
                                        poster.PostEvent(StateEvent::Success);
                                        break;
                                }
                        }));
}

void UpdateRebootState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering ArtifactReboot state");

        assert(ctx.deployment.state_data->update_info.reboot_requested.size() == 1);
        auto exp_reboot_mode =
                DbStringToNeedsReboot(ctx.deployment.state_data->update_info.reboot_requested[0]);
        // Should always be true because we check it at load time.
        assert(exp_reboot_mode);

        switch (exp_reboot_mode.value()) {
        case update_module::RebootAction::No:
                // Should not happen because then we don't enter this state.
                assert(false);
                poster.PostEvent(StateEvent::Failure);
                break;
        case update_module::RebootAction::Yes:
                DefaultAsyncErrorHandler(
                        poster,
                        ctx.deployment.update_module->AsyncArtifactReboot(
                                ctx.event_loop, DefaultStateHandler {poster}));
                break;
        case update_module::RebootAction::Automatic:
                DefaultAsyncErrorHandler(
                        poster,
                        ctx.deployment.update_module->AsyncSystemReboot(
                                ctx.event_loop, DefaultStateHandler {poster}));
                break;
        }
}

void UpdateVerifyRebootState::OnEnterSaveState(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering ArtifactVerifyReboot state");

        ctx.deployment.update_module->EnsureRootfsImageFileTree(
                ctx.deployment.update_module->GetUpdateModuleWorkDir());

        DefaultAsyncErrorHandler(
                poster,
                ctx.deployment.update_module->AsyncArtifactVerifyReboot(
                        ctx.event_loop, DefaultStateHandler {poster}));
}

void UpdateBeforeCommitState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        // It's possible that the update we have done has changed our credentials. Therefore it's
        // important that we try to log in from scratch and do not use the token we already have.
        ctx.http_client.ExpireToken();

        poster.PostEvent(StateEvent::Success);
}

void UpdateCommitState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering ArtifactCommit state");

        // Explicitly check if state scripts version is supported
        auto err = script_executor::CheckScriptsCompatibility(
                ctx.mender_context.GetConfig().paths.GetRootfsScriptsPath());
        if (err != error::NoError) {
                log::Error("Failed script compatibility check: " + err.String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }

        DefaultAsyncErrorHandler(
                poster,
                ctx.deployment.update_module->AsyncArtifactCommit(
                        ctx.event_loop, DefaultStateHandler {poster}));
}

void UpdateAfterCommitState::OnEnterSaveState(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        // Now we have committed. If we had a schema update, re-save state data with the new schema.
        assert(ctx.deployment.state_data);
        auto &state_data = *ctx.deployment.state_data;
        if (state_data.update_info.has_db_schema_update) {
                state_data.update_info.has_db_schema_update = false;
                auto err = ctx.SaveDeploymentStateData(state_data);
                if (err != error::NoError) {
                        log::Error("Not able to commit schema update: " + err.String());
                        poster.PostEvent(StateEvent::Failure);
                        return;
                }
        }

        poster.PostEvent(StateEvent::Success);
}

void UpdateCheckRollbackState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        DefaultAsyncErrorHandler(
                poster,
                ctx.deployment.update_module->AsyncSupportsRollback(
                        ctx.event_loop, [&ctx, &poster](expected::ExpectedBool rollback_supported) {
                                if (!rollback_supported.has_value()) {
                                        log::Error(rollback_supported.error().String());
                                        poster.PostEvent(StateEvent::Failure);
                                        return;
                                }

                                ctx.deployment.state_data->update_info.supports_rollback =
                                        SupportsRollbackToDbString(*rollback_supported);
                                if (*rollback_supported) {
                                        poster.PostEvent(StateEvent::RollbackStarted);
                                        poster.PostEvent(StateEvent::Success);
                                } else {
                                        poster.PostEvent(StateEvent::NothingToDo);
                                }
                        }));
}

void UpdateRollbackState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering ArtifactRollback state");

        DefaultAsyncErrorHandler(
                poster,
                ctx.deployment.update_module->AsyncArtifactRollback(
                        ctx.event_loop, DefaultStateHandler {poster}));
}

void UpdateRollbackRebootState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering ArtifactRollbackReboot state");

        auto exp_reboot_mode =
                DbStringToNeedsReboot(ctx.deployment.state_data->update_info.reboot_requested[0]);
        // Should always be true because we check it at load time.
        assert(exp_reboot_mode);

        // We ignore errors in this state as long as the ArtifactVerifyRollbackReboot state
        // succeeds.
        auto handler = [&poster](error::Error err) {
                if (err != error::NoError) {
                        log::Error(err.String());
                }
                poster.PostEvent(StateEvent::Success);
        };

        error::Error err;
        switch (exp_reboot_mode.value()) {
        case update_module::RebootAction::No:
                // Should not happen because then we don't enter this state.
                assert(false);

                err = error::MakeError(
                        error::ProgrammingError, "Entered UpdateRollbackRebootState with RebootAction = No");
                break;

        case update_module::RebootAction::Yes:
                err = ctx.deployment.update_module->AsyncArtifactRollbackReboot(ctx.event_loop, handler);
                break;

        case update_module::RebootAction::Automatic:
                err = ctx.deployment.update_module->AsyncSystemReboot(ctx.event_loop, handler);
                break;
        }

        if (err != error::NoError) {
                log::Error(err.String());
                poster.PostEvent(StateEvent::Success);
        }
}

void UpdateVerifyRollbackRebootState::OnEnterSaveState(
        Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering ArtifactVerifyRollbackReboot state");

        // In this state we only retry, we don't fail. If this keeps on going forever, then the
        // state loop detection will eventually kick in.
        auto err = ctx.deployment.update_module->AsyncArtifactVerifyRollbackReboot(
                ctx.event_loop, [&poster](error::Error err) {
                        if (err != error::NoError) {
                                log::Error(err.String());
                                poster.PostEvent(StateEvent::Retry);
                                return;
                        }
                        poster.PostEvent(StateEvent::Success);
                });
        if (err != error::NoError) {
                log::Error(err.String());
                poster.PostEvent(StateEvent::Retry);
        }
}

void UpdateRollbackSuccessfulState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        ctx.deployment.state_data->update_info.all_rollbacks_successful = true;
        poster.PostEvent(StateEvent::Success);
}

void UpdateFailureState::OnEnterSaveState(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering ArtifactFailure state");

        DefaultAsyncErrorHandler(
                poster,
                ctx.deployment.update_module->AsyncArtifactFailure(
                        ctx.event_loop, DefaultStateHandler {poster}));
}

static string AddInconsistentSuffix(const string &str) {
        const auto &suffix = main_context::MenderContext::broken_artifact_name_suffix;
        // `string::ends_with` is C++20... grumble
        string ret {str};
        if (!common::EndsWith(ret, suffix)) {
                ret.append(suffix);
        }
        return ret;
}

void UpdateSaveProvidesState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        if (ctx.deployment.failed && !ctx.deployment.rollback_failed) {
                // If the update failed, but we rolled back successfully, then we don't need to do
                // anything, just keep the old data.
                poster.PostEvent(StateEvent::Success);
                return;
        }

        assert(ctx.deployment.state_data);
        // This state should never happen: rollback failed, but update not failed??
        assert(!(!ctx.deployment.failed && ctx.deployment.rollback_failed));

        // We expect Cleanup to be the next state after this.
        ctx.deployment.state_data->state = ctx.kUpdateStateCleanup;

        auto &artifact = ctx.deployment.state_data->update_info.artifact;

        string artifact_name;
        if (ctx.deployment.rollback_failed) {
                artifact_name = AddInconsistentSuffix(artifact.artifact_name);
        } else {
                artifact_name = artifact.artifact_name;
        }

        bool deploy_failed = ctx.deployment.failed;

        // Only the artifact_name and group should be committed in the case of a
        // failing update in order to make this consistent with the old client
        // behaviour.
        auto err = ctx.mender_context.CommitArtifactData(
                artifact_name,
                artifact.artifact_group,
                deploy_failed ? nullopt : optional<context::ProvidesData>(artifact.type_info_provides),
                /* Special case: Keep existing provides */
                deploy_failed ? context::ClearsProvidesData {}
                                          : optional<context::ClearsProvidesData>(artifact.clears_artifact_provides),
                [&ctx](kv_db::Transaction &txn) {
                        // Save the Cleanup state together with the artifact data, atomically.
                        return ctx.SaveDeploymentStateData(txn, *ctx.deployment.state_data);
                });
        if (err != error::NoError) {
                log::Error("Error saving artifact data: " + err.String());
                if (err.code
                        == main_context::MakeError(main_context::StateDataStoreCountExceededError, "").code) {
                        poster.PostEvent(StateEvent::StateLoopDetected);
                        return;
                }
                poster.PostEvent(StateEvent::Failure);
                return;
        }

        poster.PostEvent(StateEvent::Success);
}

void UpdateCleanupState::OnEnterSaveState(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Debug("Entering ArtifactCleanup state");

        // It's possible for there not to be an initialized update_module structure, if the
        // deployment failed before we could successfully parse the artifact. If so, cleanup is a
        // no-op.
        if (!ctx.deployment.update_module) {
                poster.PostEvent(StateEvent::Success);
                return;
        }

        DefaultAsyncErrorHandler(
                poster,
                ctx.deployment.update_module->AsyncCleanup(ctx.event_loop, DefaultStateHandler {poster}));
}

void ClearArtifactDataState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        auto err = ctx.mender_context.GetMenderStoreDB().WriteTransaction([](kv_db::Transaction &txn) {
                // Remove state data, since we're done now.
                auto err = txn.Remove(main_context::MenderContext::state_data_key);
                if (err != error::NoError) {
                        return err;
                }
                return txn.Remove(main_context::MenderContext::state_data_key_uncommitted);
        });
        if (err != error::NoError) {
                log::Error("Error removing artifact data: " + err.String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }

        poster.PostEvent(StateEvent::Success);
}

void StateLoopState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        assert(ctx.deployment.state_data);
        auto &artifact = ctx.deployment.state_data->update_info.artifact;

        // Mark update as inconsistent.
        string artifact_name = AddInconsistentSuffix(artifact.artifact_name);

        auto err = ctx.mender_context.CommitArtifactData(
                artifact_name,
                artifact.artifact_group,
                artifact.type_info_provides,
                artifact.clears_artifact_provides,
                [](kv_db::Transaction &txn) { return error::NoError; });
        if (err != error::NoError) {
                log::Error("Error saving inconsistent artifact data: " + err.String());
                poster.PostEvent(StateEvent::Failure);
                return;
        }

        poster.PostEvent(StateEvent::Success);
}

void EndOfDeploymentState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        log::Info(
                "Deployment with ID " + ctx.deployment.state_data->update_info.id
                + " finished with status: " + string(ctx.deployment.failed ? "Failure" : "Success"));

        ctx.FinishDeploymentLogging();

        ctx.deployment = {};
        poster.PostEvent(
                StateEvent::InventoryPollingTriggered); // Submit the inventory right after an update
        poster.PostEvent(StateEvent::DeploymentEnded);
        poster.PostEvent(StateEvent::Success);
}

ExitState::ExitState(events::EventLoop &event_loop) :
        event_loop_(event_loop) {
}

void ExitState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
#ifndef NDEBUG
        if (--iterations_left_ <= 0) {
                event_loop_.Stop();
        } else {
                poster.PostEvent(StateEvent::Success);
        }
#else
        event_loop_.Stop();
#endif
}

namespace deployment_tracking {

void NoFailuresState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        ctx.deployment.failed = false;
        ctx.deployment.rollback_failed = false;
}

void FailureState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        ctx.deployment.failed = true;
        ctx.deployment.rollback_failed = true;
}

void RollbackAttemptedState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        ctx.deployment.failed = true;
        ctx.deployment.rollback_failed = false;
}

void RollbackFailedState::OnEnter(Context &ctx, sm::EventPoster<StateEvent> &poster) {
        ctx.deployment.failed = true;
        ctx.deployment.rollback_failed = true;
}

} // namespace deployment_tracking

} // namespace daemon
} // namespace update
} // namespace mender

mendersoftware / mender / 2276402292

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous