~aleteoryx/muditaos

muditaos/module-services/service-desktop/WorkerDesktop.hpp -rw-r--r-- 2.1 KiB
a405cad6 — Aleteoryx trim readme 6 days ago 
                                                                                
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

// Copyright (c) 2017-2024, Mudita Sp. z.o.o. All rights reserved.
// For licensing, see https://github.com/mudita/MuditaOS/blob/master/LICENSE.md

#pragma once

#include <Service/Message.hpp>
#include <Service/Service.hpp>
#include <Service/Worker.hpp>
#include <SystemManager/CpuSentinel.hpp>
#include <service-desktop/parser/ParserFSM.hpp>
#include <bsp/usb/usb.hpp>
#include <service-desktop/USBSecurityModel.hpp>

#include <crc32.h>

#include <filesystem>
#include <atomic>

class WorkerDesktop : public sys::Worker
{
  public:
    enum class Signal
    {
        unlockMTP
    };

    WorkerDesktop(sys::Service *ownerServicePtr,
                  std::function<void()> messageProcessedCallback,
                  const sdesktop::USBSecurityModel &securityModel,
                  const std::string &serialNumber,
                  const std::string &caseColour,
                  const std::string &rootPath);

    virtual bool init(std::list<sys::WorkerQueueInfo> queues) override;
    void closeWorker();

    bool handleMessage(std::uint32_t queueID) override final;
    void notify(Signal command);

  private:
    void reset();

    bool handleReceiveQueueMessage(std::shared_ptr<sys::WorkerQueue> &queue);
    bool handleSendQueueMessage(std::shared_ptr<sys::WorkerQueue> &queue);
    bool handleServiceQueueMessage(std::shared_ptr<sys::WorkerQueue> &queue);
    bool handleIrqQueueMessage(std::shared_ptr<sys::WorkerQueue> &queue);
    bool handleSignallingQueueMessage(std::shared_ptr<sys::WorkerQueue> &queue);

    void handleUsbConnected();
    void handleUsbDisconnected();
    void handleUsbConfigured();

    std::atomic<bool> initialized = false;
    std::atomic<bool> configured  = false;

    xQueueHandle receiveQueue;
    xQueueHandle irqQueue;
    const sdesktop::USBSecurityModel &securityModel;
    const std::string serialNumber;
    const std::string caseColour;
    std::string rootPath;
    sys::Service *ownerService = nullptr;
    sdesktop::endpoints::StateMachine parser;

    static constexpr auto cpuSentinelName = "WorkerDesktop";
    std::shared_ptr<sys::CpuSentinel> cpuSentinel;

    std::function<void()> messageProcessedCallback;
};