1 #pragma once
2 
3 #include "common/utils.hpp"
4 #include "libpldmresponder/pdr_utils.hpp"
5 #include "libpldmresponder/platform.hpp"
6 
7 #include <string>
8 
9 using namespace pldm::utils;
10 namespace pldm
11 {
12 namespace responder
13 {
14 
15 static constexpr uint8_t pSideNum = 1;
16 static constexpr uint8_t tSideNum = 2;
17 static constexpr auto Pside = "P";
18 static constexpr auto Tside = "T";
19 
20 static constexpr auto redundancyIntf =
21     "xyz.openbmc_project.Software.RedundancyPriority";
22 
23 /** @class CodeUpdate
24  *
25  *  @brief This class performs the necessary operation in pldm for
26  *         inband code update. That includes taking actions on the
27  *         setStateEffecterStates calls from Host and also sending
28  *         notification to phosphor-software-manager app
29  */
30 class CodeUpdate
31 {
32   public:
33     /** @brief Constructor to create an inband codeupdate object
34      *  @param[in] dBusIntf - D-Bus handler pointer
35      */
36     CodeUpdate(const pldm::utils::DBusHandler* dBusIntf) : dBusIntf(dBusIntf)
37     {
38         currBootSide = Tside;
39         nextBootSide = Tside;
40         markerLidSensorId = PLDM_INVALID_EFFECTER_ID;
41         firmwareUpdateSensorId = PLDM_INVALID_EFFECTER_ID;
42     }
43 
44     /* @brief Method to return the current boot side
45      */
46     std::string fetchCurrentBootSide();
47 
48     /* @brief Method to return the next boot side
49      */
50     std::string fetchNextBootSide();
51 
52     /* @brief Method to set the current boot side or
53      *        perform a rename operation on current boot side
54      * @param[in] currSide - current side to be set to
55      * @return PLDM_SUCCESS codes
56      */
57     int setCurrentBootSide(const std::string& currSide);
58 
59     /* @brief Method to set the next boot side
60      * @param[in] nextSide - next boot side to be set to
61      * @return PLDM_SUCCESS codes
62      */
63     int setNextBootSide(const std::string& nextSide);
64 
65     /* @brief Method to set the running and non-running
66      *        images
67      */
68     virtual void setVersions();
69 
70     /* @brief Method to return the newly upoaded image id in
71      *        /tmp
72      */
73     std::string fetchnewImageId()
74     {
75         return newImageId;
76     }
77 
78     /* @brief Method to set the oem platform handler in CodeUpdate class */
79     void setOemPlatformHandler(pldm::responder::oem_platform::Handler* handler);
80 
81     /* @brief Method to check whether code update is
82      *        going on
83      *  @return - bool
84      */
85     bool isCodeUpdateInProgress()
86     {
87         return codeUpdateInProgress;
88     }
89 
90     /* @brief Method to indicate whether code update
91      *        is going on
92      * @param[in] progress - yes/no
93      */
94     void setCodeUpdateProgress(bool progress)
95     {
96         codeUpdateInProgress = progress;
97     }
98 
99     /** @brief Method to clear contents the LID staging directory that contains
100      *  images such as host firmware and BMC.
101      *  @param[in] dirPath - directory system path that has to be cleared
102      *  @return none
103      */
104     void clearDirPath(const std::string& dirPath);
105 
106     /* @brief Method to set the RequestApplyTime D-Bus property
107      *        on start update to OnReset
108      * @return - Completion codes
109      */
110     int setRequestedApplyTime();
111 
112     /* @brief Method to set the RequestedActivation D-Bus property
113      *        on end update to Active by fetching the newImageID and
114      *        clearning it once RequestedActivation is set or on error
115      * @param[in] codeUpdate - codeUpdate pointer
116      * @return - Completion codes
117      */
118     int setRequestedActivation();
119 
120     /* @brief Method to fetch the sensor id for marker lid
121      * validation PDR
122      * @return - sensor id
123      */
124     uint16_t getMarkerLidSensor()
125     {
126         return markerLidSensorId;
127     }
128 
129     /* @brief Method to set the sensor id for marker lid
130      * validation
131      * @param[in] sensorId - sensor id for marker lid validation
132      */
133     void setMarkerLidSensor(uint16_t sensorId)
134     {
135         markerLidSensorId = sensorId;
136     }
137 
138     /* @brief Method to set the sensor id for firmware update state
139      * @param[in] sensorId - sensor id for firmware update state
140      */
141     void setFirmwareUpdateSensor(uint16_t sensorId)
142     {
143         firmwareUpdateSensorId = sensorId;
144     }
145 
146     /* @brief Method to fetch the sensor id for firmware update state
147      * @return - sensor id
148      */
149     uint16_t getFirmwareUpdateSensor()
150     {
151         return firmwareUpdateSensorId;
152     }
153 
154     /* @brief Method to send a state sensor event to Host from CodeUpdate class
155      * @param[in] sensorId - sensor id for the event
156      * @param[in] sensorEventClass - sensor event class wrt DSP0248
157      * @param[in] sensorOffset - sensor offset
158      * @param[in] eventState - new event state
159      * @param[in] prevEventState - previous state
160      */
161     void sendStateSensorEvent(uint16_t sensorId,
162                               enum sensor_event_class_states sensorEventClass,
163                               uint8_t sensorOffset, uint8_t eventState,
164                               uint8_t prevEventState);
165 
166     /* @brief Method to delete the image from non running side prior to
167      * an inband code update
168      */
169     void deleteImage();
170 
171     virtual ~CodeUpdate()
172     {}
173 
174   private:
175     std::string currBootSide;      //!< current boot side
176     std::string nextBootSide;      //!< next boot side
177     std::string runningVersion;    //!< currently running image
178     std::string nonRunningVersion; //!< alternate image
179     std::string newImageId;        //!< new image id
180     bool codeUpdateInProgress =
181         false; //!< indicates whether codeupdate is going on
182     const pldm::utils::DBusHandler* dBusIntf; //!< D-Bus handler
183     std::vector<std::unique_ptr<sdbusplus::bus::match::match>>
184         captureNextBootSideChange; //!< vector to catch the D-Bus property
185                                    //!< change for next boot side
186     std::unique_ptr<sdbusplus::bus::match::match>
187         fwUpdateMatcher; //!< pointer to capture the interface added signal for
188                          //!< new image
189     pldm::responder::oem_platform::Handler*
190         oemPlatformHandler; //!< oem platform handler
191     uint16_t markerLidSensorId;
192     uint16_t firmwareUpdateSensorId;
193 
194     /* @brief Method to take action when the subscribed D-Bus property is
195      *        changed
196      * @param[in] chProperties - list of properties which have changed
197      * @return - none
198      */
199     void
200         processPriorityChangeNotification(const DbusChangedProps& chProperties);
201 };
202 
203 /* @brief Method to fetch current or next boot side
204  * @param[in] entityInstance - entity instance for the sensor
205  * @param[in] codeUpdate - pointer to the CodeUpdate object
206  *
207  * @return - boot side
208  */
209 uint8_t fetchBootSide(uint16_t entityInstance, CodeUpdate* codeUpdate);
210 
211 /* @brief Method to set current or next  boot side
212  * @param[in] entityInstance - entity instance for the effecter
213  * @param[in] currState - state to be set
214  * @param[in] stateField - state field set as sent by Host
215  * @return - PLDM_SUCCESS codes
216  */
217 int setBootSide(uint16_t entityInstance, uint8_t currState,
218                 const std::vector<set_effecter_state_field>& stateField,
219                 CodeUpdate* codeUpdate);
220 
221 /* @brief Method to process LIDs during inband update, such as verifying and
222  *        removing the header to get them ready to be written to flash
223  * @param[in] filePath - Path to the LID file
224  * @return - PLDM_SUCCESS codes
225  */
226 int processCodeUpdateLid(const std::string& filePath);
227 
228 /** @brief Method to assemble the code update tarball and trigger the
229  *         phosphor software manager to create a version interface
230  *  @return - PLDM_SUCCESS codes
231  */
232 int assembleCodeUpdateImage();
233 
234 } // namespace responder
235 } // namespace pldm
236