xref: /openbmc/bmcweb/features/redfish/include/utils/query_param.hpp (revision 478b7adffab38c98d880ad2c39c8dc421433bd92)
1f4c99e70SEd Tanous #pragma once
2d5c80ad9SNan Zhou #include "bmcweb_config.h"
3d5c80ad9SNan Zhou 
4f4c99e70SEd Tanous #include "app.hpp"
5f4c99e70SEd Tanous #include "async_resp.hpp"
6f4c99e70SEd Tanous #include "error_messages.hpp"
725991f7dSEd Tanous #include "filter_expr_executor.hpp"
825991f7dSEd Tanous #include "filter_expr_printer.hpp"
9f4c99e70SEd Tanous #include "http_request.hpp"
1002cad96eSEd Tanous #include "http_response.hpp"
1195c6307aSEd Tanous #include "json_formatters.hpp"
12d5c80ad9SNan Zhou #include "logging.hpp"
1350ebd4afSEd Tanous #include "str_utility.hpp"
14f4c99e70SEd Tanous 
15d5c80ad9SNan Zhou #include <sys/types.h>
16d5c80ad9SNan Zhou 
17*478b7adfSEd Tanous #include <boost/beast/http/message.hpp>
18d5c80ad9SNan Zhou #include <boost/beast/http/status.hpp>
19d5c80ad9SNan Zhou #include <boost/beast/http/verb.hpp>
20d5c80ad9SNan Zhou #include <boost/url/params_view.hpp>
21d5c80ad9SNan Zhou #include <nlohmann/json.hpp>
22d5c80ad9SNan Zhou 
23d5c80ad9SNan Zhou #include <algorithm>
24e155ab54SNan Zhou #include <array>
25e155ab54SNan Zhou #include <cctype>
267cf436c9SEd Tanous #include <charconv>
27827c4902SNan Zhou #include <compare>
28d5c80ad9SNan Zhou #include <cstdint>
29d5c80ad9SNan Zhou #include <functional>
30e155ab54SNan Zhou #include <iterator>
31d5c80ad9SNan Zhou #include <limits>
32d5c80ad9SNan Zhou #include <map>
33d5c80ad9SNan Zhou #include <memory>
34d5c80ad9SNan Zhou #include <optional>
353544d2a7SEd Tanous #include <ranges>
36f4c99e70SEd Tanous #include <string>
37f4c99e70SEd Tanous #include <string_view>
38d5c80ad9SNan Zhou #include <system_error>
397cf436c9SEd Tanous #include <utility>
40f4c99e70SEd Tanous #include <vector>
41f4c99e70SEd Tanous 
42f4c99e70SEd Tanous namespace redfish
43f4c99e70SEd Tanous {
44f4c99e70SEd Tanous namespace query_param
45f4c99e70SEd Tanous {
46f4c99e70SEd Tanous 
477cf436c9SEd Tanous enum class ExpandType : uint8_t
487cf436c9SEd Tanous {
497cf436c9SEd Tanous     None,
507cf436c9SEd Tanous     Links,
517cf436c9SEd Tanous     NotLinks,
527cf436c9SEd Tanous     Both,
537cf436c9SEd Tanous };
547cf436c9SEd Tanous 
55827c4902SNan Zhou // A simple implementation of Trie to help |recursiveSelect|.
56827c4902SNan Zhou class SelectTrieNode
57827c4902SNan Zhou {
58827c4902SNan Zhou   public:
59827c4902SNan Zhou     SelectTrieNode() = default;
60827c4902SNan Zhou 
61827c4902SNan Zhou     const SelectTrieNode* find(const std::string& jsonKey) const
62827c4902SNan Zhou     {
63827c4902SNan Zhou         auto it = children.find(jsonKey);
64827c4902SNan Zhou         if (it == children.end())
65827c4902SNan Zhou         {
66827c4902SNan Zhou             return nullptr;
67827c4902SNan Zhou         }
68827c4902SNan Zhou         return &it->second;
69827c4902SNan Zhou     }
70827c4902SNan Zhou 
71827c4902SNan Zhou     // Creates a new node if the key doesn't exist, returns the reference to the
72827c4902SNan Zhou     // newly created node; otherwise, return the reference to the existing node
73827c4902SNan Zhou     SelectTrieNode* emplace(std::string_view jsonKey)
74827c4902SNan Zhou     {
75827c4902SNan Zhou         auto [it, _] = children.emplace(jsonKey, SelectTrieNode{});
76827c4902SNan Zhou         return &it->second;
77827c4902SNan Zhou     }
78827c4902SNan Zhou 
79827c4902SNan Zhou     bool empty() const
80827c4902SNan Zhou     {
81827c4902SNan Zhou         return children.empty();
82827c4902SNan Zhou     }
83827c4902SNan Zhou 
84827c4902SNan Zhou     void clear()
85827c4902SNan Zhou     {
86827c4902SNan Zhou         children.clear();
87827c4902SNan Zhou     }
88827c4902SNan Zhou 
89827c4902SNan Zhou     void setToSelected()
90827c4902SNan Zhou     {
91827c4902SNan Zhou         selected = true;
92827c4902SNan Zhou     }
93827c4902SNan Zhou 
94827c4902SNan Zhou     bool isSelected() const
95827c4902SNan Zhou     {
96827c4902SNan Zhou         return selected;
97827c4902SNan Zhou     }
98827c4902SNan Zhou 
99827c4902SNan Zhou   private:
100827c4902SNan Zhou     std::map<std::string, SelectTrieNode, std::less<>> children;
101827c4902SNan Zhou     bool selected = false;
102827c4902SNan Zhou };
103827c4902SNan Zhou 
104827c4902SNan Zhou // Validates the property in the $select parameter. Every character is among
105827c4902SNan Zhou // [a-zA-Z0-9#@_.] (taken from Redfish spec, section 9.6 Properties)
106827c4902SNan Zhou inline bool isSelectedPropertyAllowed(std::string_view property)
107827c4902SNan Zhou {
108827c4902SNan Zhou     // These a magic number, but with it it's less likely that this code
109827c4902SNan Zhou     // introduces CVE; e.g., too large properties crash the service.
110827c4902SNan Zhou     constexpr int maxPropertyLength = 60;
111827c4902SNan Zhou     if (property.empty() || property.size() > maxPropertyLength)
112827c4902SNan Zhou     {
113827c4902SNan Zhou         return false;
114827c4902SNan Zhou     }
115827c4902SNan Zhou     for (char ch : property)
116827c4902SNan Zhou     {
117827c4902SNan Zhou         if (std::isalnum(static_cast<unsigned char>(ch)) == 0 && ch != '#' &&
118827c4902SNan Zhou             ch != '@' && ch != '.')
119827c4902SNan Zhou         {
120827c4902SNan Zhou             return false;
121827c4902SNan Zhou         }
122827c4902SNan Zhou     }
123827c4902SNan Zhou     return true;
124827c4902SNan Zhou }
125827c4902SNan Zhou 
126827c4902SNan Zhou struct SelectTrie
127827c4902SNan Zhou {
128827c4902SNan Zhou     SelectTrie() = default;
129827c4902SNan Zhou 
130827c4902SNan Zhou     // Inserts a $select value; returns false if the nestedProperty is illegal.
131827c4902SNan Zhou     bool insertNode(std::string_view nestedProperty)
132827c4902SNan Zhou     {
133827c4902SNan Zhou         if (nestedProperty.empty())
134827c4902SNan Zhou         {
135827c4902SNan Zhou             return false;
136827c4902SNan Zhou         }
137827c4902SNan Zhou         SelectTrieNode* currNode = &root;
138827c4902SNan Zhou         size_t index = nestedProperty.find_first_of('/');
139827c4902SNan Zhou         while (!nestedProperty.empty())
140827c4902SNan Zhou         {
141827c4902SNan Zhou             std::string_view property = nestedProperty.substr(0, index);
142827c4902SNan Zhou             if (!isSelectedPropertyAllowed(property))
143827c4902SNan Zhou             {
144827c4902SNan Zhou                 return false;
145827c4902SNan Zhou             }
146827c4902SNan Zhou             currNode = currNode->emplace(property);
147827c4902SNan Zhou             if (index == std::string::npos)
148827c4902SNan Zhou             {
149827c4902SNan Zhou                 break;
150827c4902SNan Zhou             }
151827c4902SNan Zhou             nestedProperty.remove_prefix(index + 1);
152827c4902SNan Zhou             index = nestedProperty.find_first_of('/');
153827c4902SNan Zhou         }
154827c4902SNan Zhou         currNode->setToSelected();
155827c4902SNan Zhou         return true;
156827c4902SNan Zhou     }
157827c4902SNan Zhou 
158827c4902SNan Zhou     SelectTrieNode root;
159827c4902SNan Zhou };
160827c4902SNan Zhou 
161a6b9125fSNan Zhou // The struct stores the parsed query parameters of the default Redfish route.
162f4c99e70SEd Tanous struct Query
163f4c99e70SEd Tanous {
164a6b9125fSNan Zhou     // Only
165f4c99e70SEd Tanous     bool isOnly = false;
16625991f7dSEd Tanous 
167a6b9125fSNan Zhou     // Expand
168a6b9125fSNan Zhou     uint8_t expandLevel = 0;
1697cf436c9SEd Tanous     ExpandType expandType = ExpandType::None;
170c937d2bfSEd Tanous 
171c937d2bfSEd Tanous     // Skip
1723648c8beSEd Tanous     std::optional<size_t> skip = std::nullopt;
173c937d2bfSEd Tanous 
174c937d2bfSEd Tanous     // Top
1755143f7a5SJiaqing Zhao     static constexpr size_t maxTop = 1000; // Max entries a response contain
1763648c8beSEd Tanous     std::optional<size_t> top = std::nullopt;
177e155ab54SNan Zhou 
17825991f7dSEd Tanous     // Filter
17925991f7dSEd Tanous     std::optional<filter_ast::LogicalAnd> filter = std::nullopt;
18025991f7dSEd Tanous 
181e155ab54SNan Zhou     // Select
18247f2934cSEd Tanous     // Unclear how to make this use structured initialization without this.
18347f2934cSEd Tanous     // Might be a tidy bug?  Ignore for now
18447f2934cSEd Tanous     // NOLINTNEXTLINE(readability-redundant-member-init)
18547f2934cSEd Tanous     SelectTrie selectTrie{};
186f4c99e70SEd Tanous };
187f4c99e70SEd Tanous 
188a6b9125fSNan Zhou // The struct defines how resource handlers in redfish-core/lib/ can handle
189a6b9125fSNan Zhou // query parameters themselves, so that the default Redfish route will delegate
190a6b9125fSNan Zhou // the processing.
191a6b9125fSNan Zhou struct QueryCapabilities
192a6b9125fSNan Zhou {
193a6b9125fSNan Zhou     bool canDelegateOnly = false;
194c937d2bfSEd Tanous     bool canDelegateTop = false;
195c937d2bfSEd Tanous     bool canDelegateSkip = false;
196a6b9125fSNan Zhou     uint8_t canDelegateExpandLevel = 0;
197e155ab54SNan Zhou     bool canDelegateSelect = false;
198a6b9125fSNan Zhou };
199a6b9125fSNan Zhou 
200a6b9125fSNan Zhou // Delegates query parameters according to the given |queryCapabilities|
201a6b9125fSNan Zhou // This function doesn't check query parameter conflicts since the parse
202a6b9125fSNan Zhou // function will take care of it.
203a6b9125fSNan Zhou // Returns a delegated query object which can be used by individual resource
204a6b9125fSNan Zhou // handlers so that handlers don't need to query again.
205a6b9125fSNan Zhou inline Query delegate(const QueryCapabilities& queryCapabilities, Query& query)
206a6b9125fSNan Zhou {
207f1a1e3dcSEd Tanous     Query delegated{};
208a6b9125fSNan Zhou     // delegate only
209a6b9125fSNan Zhou     if (query.isOnly && queryCapabilities.canDelegateOnly)
210a6b9125fSNan Zhou     {
211a6b9125fSNan Zhou         delegated.isOnly = true;
212a6b9125fSNan Zhou         query.isOnly = false;
213a6b9125fSNan Zhou     }
214a6b9125fSNan Zhou     // delegate expand as much as we can
215a6b9125fSNan Zhou     if (query.expandType != ExpandType::None)
216a6b9125fSNan Zhou     {
217a6b9125fSNan Zhou         delegated.expandType = query.expandType;
218a6b9125fSNan Zhou         if (query.expandLevel <= queryCapabilities.canDelegateExpandLevel)
219a6b9125fSNan Zhou         {
220a6b9125fSNan Zhou             query.expandType = ExpandType::None;
221a6b9125fSNan Zhou             delegated.expandLevel = query.expandLevel;
222a6b9125fSNan Zhou             query.expandLevel = 0;
223a6b9125fSNan Zhou         }
224a6b9125fSNan Zhou         else
225a6b9125fSNan Zhou         {
226a6b9125fSNan Zhou             delegated.expandLevel = queryCapabilities.canDelegateExpandLevel;
227a6b9125fSNan Zhou         }
228a6b9125fSNan Zhou     }
229c937d2bfSEd Tanous 
230c937d2bfSEd Tanous     // delegate top
2313648c8beSEd Tanous     if (query.top && queryCapabilities.canDelegateTop)
232c937d2bfSEd Tanous     {
233c937d2bfSEd Tanous         delegated.top = query.top;
2343648c8beSEd Tanous         query.top = std::nullopt;
235c937d2bfSEd Tanous     }
236c937d2bfSEd Tanous 
237c937d2bfSEd Tanous     // delegate skip
2383648c8beSEd Tanous     if (query.skip && queryCapabilities.canDelegateSkip)
239c937d2bfSEd Tanous     {
240c937d2bfSEd Tanous         delegated.skip = query.skip;
241c937d2bfSEd Tanous         query.skip = 0;
242c937d2bfSEd Tanous     }
243e155ab54SNan Zhou 
244e155ab54SNan Zhou     // delegate select
245827c4902SNan Zhou     if (!query.selectTrie.root.empty() && queryCapabilities.canDelegateSelect)
246e155ab54SNan Zhou     {
247827c4902SNan Zhou         delegated.selectTrie = std::move(query.selectTrie);
248827c4902SNan Zhou         query.selectTrie.root.clear();
249e155ab54SNan Zhou     }
250a6b9125fSNan Zhou     return delegated;
251a6b9125fSNan Zhou }
252a6b9125fSNan Zhou 
2537cf436c9SEd Tanous inline bool getExpandType(std::string_view value, Query& query)
2547cf436c9SEd Tanous {
2557cf436c9SEd Tanous     if (value.empty())
2567cf436c9SEd Tanous     {
2577cf436c9SEd Tanous         return false;
2587cf436c9SEd Tanous     }
2597cf436c9SEd Tanous     switch (value[0])
2607cf436c9SEd Tanous     {
2617cf436c9SEd Tanous         case '*':
2627cf436c9SEd Tanous             query.expandType = ExpandType::Both;
2637cf436c9SEd Tanous             break;
2647cf436c9SEd Tanous         case '.':
2657cf436c9SEd Tanous             query.expandType = ExpandType::NotLinks;
2667cf436c9SEd Tanous             break;
2677cf436c9SEd Tanous         case '~':
2687cf436c9SEd Tanous             query.expandType = ExpandType::Links;
2697cf436c9SEd Tanous             break;
2707cf436c9SEd Tanous         default:
2717cf436c9SEd Tanous             return false;
2727cf436c9SEd Tanous     }
2737cf436c9SEd Tanous     value.remove_prefix(1);
2747cf436c9SEd Tanous     if (value.empty())
2757cf436c9SEd Tanous     {
2767cf436c9SEd Tanous         query.expandLevel = 1;
2777cf436c9SEd Tanous         return true;
2787cf436c9SEd Tanous     }
2797cf436c9SEd Tanous     constexpr std::string_view levels = "($levels=";
2807cf436c9SEd Tanous     if (!value.starts_with(levels))
2817cf436c9SEd Tanous     {
2827cf436c9SEd Tanous         return false;
2837cf436c9SEd Tanous     }
2847cf436c9SEd Tanous     value.remove_prefix(levels.size());
2857cf436c9SEd Tanous 
2862bd4ab43SPatrick Williams     auto it = std::from_chars(value.begin(), value.end(), query.expandLevel);
2877cf436c9SEd Tanous     if (it.ec != std::errc())
2887cf436c9SEd Tanous     {
2897cf436c9SEd Tanous         return false;
2907cf436c9SEd Tanous     }
2912bd4ab43SPatrick Williams     value.remove_prefix(
2922bd4ab43SPatrick Williams         static_cast<size_t>(std::distance(value.begin(), it.ptr)));
2937cf436c9SEd Tanous     return value == ")";
2947cf436c9SEd Tanous }
2957cf436c9SEd Tanous 
296c937d2bfSEd Tanous enum class QueryError
297c937d2bfSEd Tanous {
298c937d2bfSEd Tanous     Ok,
299c937d2bfSEd Tanous     OutOfRange,
300c937d2bfSEd Tanous     ValueFormat,
301c937d2bfSEd Tanous };
302c937d2bfSEd Tanous 
303c937d2bfSEd Tanous inline QueryError getNumericParam(std::string_view value, size_t& param)
304c937d2bfSEd Tanous {
305bd79bce8SPatrick Williams     std::from_chars_result r =
306bd79bce8SPatrick Williams         std::from_chars(value.begin(), value.end(), param);
307c937d2bfSEd Tanous 
308c937d2bfSEd Tanous     // If the number wasn't representable in the type, it's out of range
309c937d2bfSEd Tanous     if (r.ec == std::errc::result_out_of_range)
310c937d2bfSEd Tanous     {
311c937d2bfSEd Tanous         return QueryError::OutOfRange;
312c937d2bfSEd Tanous     }
313c937d2bfSEd Tanous     // All other errors are value format
314c937d2bfSEd Tanous     if (r.ec != std::errc())
315c937d2bfSEd Tanous     {
316c937d2bfSEd Tanous         return QueryError::ValueFormat;
317c937d2bfSEd Tanous     }
318c937d2bfSEd Tanous     return QueryError::Ok;
319c937d2bfSEd Tanous }
320c937d2bfSEd Tanous 
321c937d2bfSEd Tanous inline QueryError getSkipParam(std::string_view value, Query& query)
322c937d2bfSEd Tanous {
3233648c8beSEd Tanous     return getNumericParam(value, query.skip.emplace());
324c937d2bfSEd Tanous }
325c937d2bfSEd Tanous 
326c937d2bfSEd Tanous inline QueryError getTopParam(std::string_view value, Query& query)
327c937d2bfSEd Tanous {
3283648c8beSEd Tanous     QueryError ret = getNumericParam(value, query.top.emplace());
329c937d2bfSEd Tanous     if (ret != QueryError::Ok)
330c937d2bfSEd Tanous     {
331c937d2bfSEd Tanous         return ret;
332c937d2bfSEd Tanous     }
333c937d2bfSEd Tanous 
334c937d2bfSEd Tanous     // Range check for sanity.
3355143f7a5SJiaqing Zhao     if (query.top > Query::maxTop)
336c937d2bfSEd Tanous     {
337c937d2bfSEd Tanous         return QueryError::OutOfRange;
338c937d2bfSEd Tanous     }
339c937d2bfSEd Tanous 
340c937d2bfSEd Tanous     return QueryError::Ok;
341c937d2bfSEd Tanous }
342c937d2bfSEd Tanous 
343e155ab54SNan Zhou // Parses and validates the $select parameter.
344e155ab54SNan Zhou // As per OData URL Conventions and Redfish Spec, the $select values shall be
345e155ab54SNan Zhou // comma separated Resource Path
346e155ab54SNan Zhou // Ref:
347e155ab54SNan Zhou // 1. https://datatracker.ietf.org/doc/html/rfc3986#section-3.3
348e155ab54SNan Zhou // 2.
349e155ab54SNan Zhou // https://docs.oasis-open.org/odata/odata/v4.01/os/abnf/odata-abnf-construction-rules.txt
350e155ab54SNan Zhou inline bool getSelectParam(std::string_view value, Query& query)
351e155ab54SNan Zhou {
352e155ab54SNan Zhou     std::vector<std::string> properties;
35350ebd4afSEd Tanous     bmcweb::split(properties, value, ',');
354e155ab54SNan Zhou     if (properties.empty())
355e155ab54SNan Zhou     {
356e155ab54SNan Zhou         return false;
357e155ab54SNan Zhou     }
358e155ab54SNan Zhou     // These a magic number, but with it it's less likely that this code
359e155ab54SNan Zhou     // introduces CVE; e.g., too large properties crash the service.
360e155ab54SNan Zhou     constexpr int maxNumProperties = 10;
361e155ab54SNan Zhou     if (properties.size() > maxNumProperties)
362e155ab54SNan Zhou     {
363e155ab54SNan Zhou         return false;
364e155ab54SNan Zhou     }
365827c4902SNan Zhou     for (const auto& property : properties)
366e155ab54SNan Zhou     {
367827c4902SNan Zhou         if (!query.selectTrie.insertNode(property))
368e155ab54SNan Zhou         {
369e155ab54SNan Zhou             return false;
370e155ab54SNan Zhou         }
371e155ab54SNan Zhou     }
372e155ab54SNan Zhou     return true;
373e155ab54SNan Zhou }
374e155ab54SNan Zhou 
37525991f7dSEd Tanous // Parses and validates the $filter parameter.
37625991f7dSEd Tanous inline bool getFilterParam(std::string_view value, Query& query)
37725991f7dSEd Tanous {
37825991f7dSEd Tanous     query.filter = parseFilter(value);
37925991f7dSEd Tanous     return query.filter.has_value();
38025991f7dSEd Tanous }
38125991f7dSEd Tanous 
382bd79bce8SPatrick Williams inline std::optional<Query>
383bd79bce8SPatrick Williams     parseParameters(boost::urls::params_view urlParams, crow::Response& res)
384f4c99e70SEd Tanous {
385f1a1e3dcSEd Tanous     Query ret{};
386f4c99e70SEd Tanous     for (const boost::urls::params_view::value_type& it : urlParams)
387f4c99e70SEd Tanous     {
388079360aeSEd Tanous         if (it.key == "only")
389f4c99e70SEd Tanous         {
390f4c99e70SEd Tanous             if (!it.value.empty())
391f4c99e70SEd Tanous             {
392079360aeSEd Tanous                 messages::queryParameterValueFormatError(res, it.value, it.key);
393f4c99e70SEd Tanous                 return std::nullopt;
394f4c99e70SEd Tanous             }
395f4c99e70SEd Tanous             ret.isOnly = true;
396f4c99e70SEd Tanous         }
39725b54dbaSEd Tanous         else if (it.key == "$expand" && BMCWEB_INSECURE_ENABLE_REDFISH_QUERY)
3987cf436c9SEd Tanous         {
399079360aeSEd Tanous             if (!getExpandType(it.value, ret))
4007cf436c9SEd Tanous             {
401079360aeSEd Tanous                 messages::queryParameterValueFormatError(res, it.value, it.key);
4027cf436c9SEd Tanous                 return std::nullopt;
403f4c99e70SEd Tanous             }
4047cf436c9SEd Tanous         }
405079360aeSEd Tanous         else if (it.key == "$top")
406c937d2bfSEd Tanous         {
407079360aeSEd Tanous             QueryError topRet = getTopParam(it.value, ret);
408c937d2bfSEd Tanous             if (topRet == QueryError::ValueFormat)
409c937d2bfSEd Tanous             {
410079360aeSEd Tanous                 messages::queryParameterValueFormatError(res, it.value, it.key);
411c937d2bfSEd Tanous                 return std::nullopt;
412c937d2bfSEd Tanous             }
413c937d2bfSEd Tanous             if (topRet == QueryError::OutOfRange)
414c937d2bfSEd Tanous             {
415c937d2bfSEd Tanous                 messages::queryParameterOutOfRange(
416079360aeSEd Tanous                     res, it.value, "$top",
417079360aeSEd Tanous                     "0-" + std::to_string(Query::maxTop));
418c937d2bfSEd Tanous                 return std::nullopt;
419c937d2bfSEd Tanous             }
420c937d2bfSEd Tanous         }
421079360aeSEd Tanous         else if (it.key == "$skip")
422c937d2bfSEd Tanous         {
423079360aeSEd Tanous             QueryError topRet = getSkipParam(it.value, ret);
424c937d2bfSEd Tanous             if (topRet == QueryError::ValueFormat)
425c937d2bfSEd Tanous             {
426079360aeSEd Tanous                 messages::queryParameterValueFormatError(res, it.value, it.key);
427c937d2bfSEd Tanous                 return std::nullopt;
428c937d2bfSEd Tanous             }
429c937d2bfSEd Tanous             if (topRet == QueryError::OutOfRange)
430c937d2bfSEd Tanous             {
431c937d2bfSEd Tanous                 messages::queryParameterOutOfRange(
432079360aeSEd Tanous                     res, it.value, it.key,
433a926c53eSJiaqing Zhao                     "0-" + std::to_string(std::numeric_limits<size_t>::max()));
434c937d2bfSEd Tanous                 return std::nullopt;
435c937d2bfSEd Tanous             }
436c937d2bfSEd Tanous         }
437079360aeSEd Tanous         else if (it.key == "$select")
438e155ab54SNan Zhou         {
439079360aeSEd Tanous             if (!getSelectParam(it.value, ret))
440e155ab54SNan Zhou             {
441079360aeSEd Tanous                 messages::queryParameterValueFormatError(res, it.value, it.key);
442e155ab54SNan Zhou                 return std::nullopt;
443e155ab54SNan Zhou             }
444e155ab54SNan Zhou         }
44525991f7dSEd Tanous         else if (it.key == "$filter" && BMCWEB_INSECURE_ENABLE_REDFISH_QUERY)
44625991f7dSEd Tanous         {
44725991f7dSEd Tanous             if (!getFilterParam(it.value, ret))
44825991f7dSEd Tanous             {
44925991f7dSEd Tanous                 messages::queryParameterValueFormatError(res, it.value, it.key);
45025991f7dSEd Tanous                 return std::nullopt;
45125991f7dSEd Tanous             }
45225991f7dSEd Tanous         }
4537cf436c9SEd Tanous         else
4547cf436c9SEd Tanous         {
4557cf436c9SEd Tanous             // Intentionally ignore other errors Redfish spec, 7.3.1
456079360aeSEd Tanous             if (it.key.starts_with("$"))
4577cf436c9SEd Tanous             {
4587cf436c9SEd Tanous                 // Services shall return... The HTTP 501 Not Implemented
4597cf436c9SEd Tanous                 // status code for any unsupported query parameters that
4607cf436c9SEd Tanous                 // start with $ .
461079360aeSEd Tanous                 messages::queryParameterValueFormatError(res, it.value, it.key);
4627cf436c9SEd Tanous                 res.result(boost::beast::http::status::not_implemented);
4637cf436c9SEd Tanous                 return std::nullopt;
4647cf436c9SEd Tanous             }
4657cf436c9SEd Tanous             // "Shall ignore unknown or unsupported query parameters that do
4667cf436c9SEd Tanous             // not begin with $ ."
4677cf436c9SEd Tanous         }
4687cf436c9SEd Tanous     }
4697cf436c9SEd Tanous 
470827c4902SNan Zhou     if (ret.expandType != ExpandType::None && !ret.selectTrie.root.empty())
471e155ab54SNan Zhou     {
472e155ab54SNan Zhou         messages::queryCombinationInvalid(res);
473e155ab54SNan Zhou         return std::nullopt;
474e155ab54SNan Zhou     }
475e155ab54SNan Zhou 
476f4c99e70SEd Tanous     return ret;
477f4c99e70SEd Tanous }
478f4c99e70SEd Tanous 
479f4c99e70SEd Tanous inline bool processOnly(crow::App& app, crow::Response& res,
480f4c99e70SEd Tanous                         std::function<void(crow::Response&)>& completionHandler)
481f4c99e70SEd Tanous {
48262598e31SEd Tanous     BMCWEB_LOG_DEBUG("Processing only query param");
483f4c99e70SEd Tanous     auto itMembers = res.jsonValue.find("Members");
484f4c99e70SEd Tanous     if (itMembers == res.jsonValue.end())
485f4c99e70SEd Tanous     {
486f4c99e70SEd Tanous         messages::queryNotSupportedOnResource(res);
487f4c99e70SEd Tanous         completionHandler(res);
488f4c99e70SEd Tanous         return false;
489f4c99e70SEd Tanous     }
490f4c99e70SEd Tanous     auto itMemBegin = itMembers->begin();
491f4c99e70SEd Tanous     if (itMemBegin == itMembers->end() || itMembers->size() != 1)
492f4c99e70SEd Tanous     {
49362598e31SEd Tanous         BMCWEB_LOG_DEBUG(
49462598e31SEd Tanous             "Members contains {} element, returning full collection.",
49562598e31SEd Tanous             itMembers->size());
496f4c99e70SEd Tanous         completionHandler(res);
497f4c99e70SEd Tanous         return false;
498f4c99e70SEd Tanous     }
499f4c99e70SEd Tanous 
500f4c99e70SEd Tanous     auto itUrl = itMemBegin->find("@odata.id");
501f4c99e70SEd Tanous     if (itUrl == itMemBegin->end())
502f4c99e70SEd Tanous     {
50362598e31SEd Tanous         BMCWEB_LOG_DEBUG("No found odata.id");
504f4c99e70SEd Tanous         messages::internalError(res);
505f4c99e70SEd Tanous         completionHandler(res);
506f4c99e70SEd Tanous         return false;
507f4c99e70SEd Tanous     }
508f4c99e70SEd Tanous     const std::string* url = itUrl->get_ptr<const std::string*>();
509f4c99e70SEd Tanous     if (url == nullptr)
510f4c99e70SEd Tanous     {
51162598e31SEd Tanous         BMCWEB_LOG_DEBUG("@odata.id wasn't a string????");
512f4c99e70SEd Tanous         messages::internalError(res);
513f4c99e70SEd Tanous         completionHandler(res);
514f4c99e70SEd Tanous         return false;
515f4c99e70SEd Tanous     }
516f4c99e70SEd Tanous     // TODO(Ed) copy request headers?
517f4c99e70SEd Tanous     // newReq.session = req.session;
518f4c99e70SEd Tanous     std::error_code ec;
519102a4cdaSJonathan Doman     auto newReq = std::make_shared<crow::Request>(
520102a4cdaSJonathan Doman         crow::Request::Body{boost::beast::http::verb::get, *url, 11}, ec);
521f4c99e70SEd Tanous     if (ec)
522f4c99e70SEd Tanous     {
523f4c99e70SEd Tanous         messages::internalError(res);
524f4c99e70SEd Tanous         completionHandler(res);
525f4c99e70SEd Tanous         return false;
526f4c99e70SEd Tanous     }
527f4c99e70SEd Tanous 
528f4c99e70SEd Tanous     auto asyncResp = std::make_shared<bmcweb::AsyncResp>();
52962598e31SEd Tanous     BMCWEB_LOG_DEBUG("setting completion handler on {}",
53062598e31SEd Tanous                      logPtr(&asyncResp->res));
531f4c99e70SEd Tanous     asyncResp->res.setCompleteRequestHandler(std::move(completionHandler));
532f4c99e70SEd Tanous     app.handle(newReq, asyncResp);
533f4c99e70SEd Tanous     return true;
534f4c99e70SEd Tanous }
535f4c99e70SEd Tanous 
5367cf436c9SEd Tanous struct ExpandNode
5377cf436c9SEd Tanous {
5387cf436c9SEd Tanous     nlohmann::json::json_pointer location;
5397cf436c9SEd Tanous     std::string uri;
5407cf436c9SEd Tanous 
5419de65b34SEd Tanous     bool operator==(const ExpandNode& other) const
5427cf436c9SEd Tanous     {
5437cf436c9SEd Tanous         return location == other.location && uri == other.uri;
5447cf436c9SEd Tanous     }
5457cf436c9SEd Tanous };
5467cf436c9SEd Tanous 
54787788abfSEd Tanous inline void findNavigationReferencesInArrayRecursive(
548c59e338cSEd Tanous     ExpandType eType, nlohmann::json::array_t& array,
54937b1f7beSEd Tanous     const nlohmann::json::json_pointer& jsonPtr, int depth, int skipDepth,
55087788abfSEd Tanous     bool inLinks, std::vector<ExpandNode>& out);
55187788abfSEd Tanous 
55287788abfSEd Tanous inline void findNavigationReferencesInObjectRecursive(
553c59e338cSEd Tanous     ExpandType eType, nlohmann::json::object_t& obj,
55437b1f7beSEd Tanous     const nlohmann::json::json_pointer& jsonPtr, int depth, int skipDepth,
55587788abfSEd Tanous     bool inLinks, std::vector<ExpandNode>& out);
55687788abfSEd Tanous 
5577cf436c9SEd Tanous // Walks a json object looking for Redfish NavigationReference entries that
5587cf436c9SEd Tanous // might need resolved.  It recursively walks the jsonResponse object, looking
5597cf436c9SEd Tanous // for links at every level, and returns a list (out) of locations within the
5607cf436c9SEd Tanous // tree that need to be expanded.  The current json pointer location p is passed
5617cf436c9SEd Tanous // in to reference the current node that's being expanded, so it can be combined
5627cf436c9SEd Tanous // with the keys from the jsonResponse object
5637cf436c9SEd Tanous inline void findNavigationReferencesRecursive(
5647cf436c9SEd Tanous     ExpandType eType, nlohmann::json& jsonResponse,
56537b1f7beSEd Tanous     const nlohmann::json::json_pointer& jsonPtr, int depth, int skipDepth,
56632cdb4a7SWilly Tu     bool inLinks, std::vector<ExpandNode>& out)
5677cf436c9SEd Tanous {
5687cf436c9SEd Tanous     // If no expand is needed, return early
5697cf436c9SEd Tanous     if (eType == ExpandType::None)
5707cf436c9SEd Tanous     {
5717cf436c9SEd Tanous         return;
5727cf436c9SEd Tanous     }
573ad595fa6SEd Tanous 
5747cf436c9SEd Tanous     nlohmann::json::array_t* array =
5757cf436c9SEd Tanous         jsonResponse.get_ptr<nlohmann::json::array_t*>();
5767cf436c9SEd Tanous     if (array != nullptr)
5777cf436c9SEd Tanous     {
57837b1f7beSEd Tanous         findNavigationReferencesInArrayRecursive(eType, *array, jsonPtr, depth,
57987788abfSEd Tanous                                                  skipDepth, inLinks, out);
58087788abfSEd Tanous     }
58187788abfSEd Tanous     nlohmann::json::object_t* obj =
58287788abfSEd Tanous         jsonResponse.get_ptr<nlohmann::json::object_t*>();
58387788abfSEd Tanous     if (obj == nullptr)
58487788abfSEd Tanous     {
58587788abfSEd Tanous         return;
58687788abfSEd Tanous     }
58737b1f7beSEd Tanous     findNavigationReferencesInObjectRecursive(eType, *obj, jsonPtr, depth,
58837b1f7beSEd Tanous                                               skipDepth, inLinks, out);
58987788abfSEd Tanous }
59087788abfSEd Tanous 
59187788abfSEd Tanous inline void findNavigationReferencesInArrayRecursive(
592c59e338cSEd Tanous     ExpandType eType, nlohmann::json::array_t& array,
59337b1f7beSEd Tanous     const nlohmann::json::json_pointer& jsonPtr, int depth, int skipDepth,
59487788abfSEd Tanous     bool inLinks, std::vector<ExpandNode>& out)
59587788abfSEd Tanous {
5967cf436c9SEd Tanous     size_t index = 0;
5977cf436c9SEd Tanous     // For arrays, walk every element in the array
598c59e338cSEd Tanous     for (auto& element : array)
5997cf436c9SEd Tanous     {
60037b1f7beSEd Tanous         nlohmann::json::json_pointer newPtr = jsonPtr / index;
60162598e31SEd Tanous         BMCWEB_LOG_DEBUG("Traversing response at {}", newPtr.to_string());
602ad595fa6SEd Tanous         findNavigationReferencesRecursive(eType, element, newPtr, depth,
60332cdb4a7SWilly Tu                                           skipDepth, inLinks, out);
6047cf436c9SEd Tanous         index++;
6057cf436c9SEd Tanous     }
6067cf436c9SEd Tanous }
60787788abfSEd Tanous 
60887788abfSEd Tanous inline void findNavigationReferencesInObjectRecursive(
609c59e338cSEd Tanous     ExpandType eType, nlohmann::json::object_t& obj,
61037b1f7beSEd Tanous     const nlohmann::json::json_pointer& jsonPtr, int depth, int skipDepth,
61187788abfSEd Tanous     bool inLinks, std::vector<ExpandNode>& out)
6127cf436c9SEd Tanous {
6137cf436c9SEd Tanous     // Navigation References only ever have a single element
614c59e338cSEd Tanous     if (obj.size() == 1)
6157cf436c9SEd Tanous     {
616c59e338cSEd Tanous         if (obj.begin()->first == "@odata.id")
6177cf436c9SEd Tanous         {
6187cf436c9SEd Tanous             const std::string* uri =
619c59e338cSEd Tanous                 obj.begin()->second.get_ptr<const std::string*>();
6207cf436c9SEd Tanous             if (uri != nullptr)
6217cf436c9SEd Tanous             {
62262598e31SEd Tanous                 BMCWEB_LOG_DEBUG("Found {} at {}", *uri, jsonPtr.to_string());
62332cdb4a7SWilly Tu                 if (skipDepth == 0)
62432cdb4a7SWilly Tu                 {
62537b1f7beSEd Tanous                     out.push_back({jsonPtr, *uri});
62632cdb4a7SWilly Tu                 }
627ad595fa6SEd Tanous                 return;
6287cf436c9SEd Tanous             }
6297cf436c9SEd Tanous         }
6307cf436c9SEd Tanous     }
631ad595fa6SEd Tanous 
632ad595fa6SEd Tanous     int newDepth = depth;
633c59e338cSEd Tanous     auto odataId = obj.find("@odata.id");
634c59e338cSEd Tanous     if (odataId != obj.end())
635ad595fa6SEd Tanous     {
636ad595fa6SEd Tanous         // The Redfish spec requires all resources to include the resource
637ad595fa6SEd Tanous         // identifier.  If the object has multiple elements and one of them is
638ad595fa6SEd Tanous         // "@odata.id" then that means we have entered a new level / expanded
639ad595fa6SEd Tanous         // resource.  We need to stop traversing if we're already at the desired
640ad595fa6SEd Tanous         // depth
641c59e338cSEd Tanous         if (obj.size() > 1)
64232cdb4a7SWilly Tu         {
64332cdb4a7SWilly Tu             if (depth == 0)
644ad595fa6SEd Tanous             {
645ad595fa6SEd Tanous                 return;
646ad595fa6SEd Tanous             }
64732cdb4a7SWilly Tu             if (skipDepth > 0)
64832cdb4a7SWilly Tu             {
64932cdb4a7SWilly Tu                 skipDepth--;
65032cdb4a7SWilly Tu             }
65132cdb4a7SWilly Tu         }
65232cdb4a7SWilly Tu 
65332cdb4a7SWilly Tu         if (skipDepth == 0)
65432cdb4a7SWilly Tu         {
655ad595fa6SEd Tanous             newDepth--;
656ad595fa6SEd Tanous         }
65732cdb4a7SWilly Tu     }
658ad595fa6SEd Tanous 
6597cf436c9SEd Tanous     // Loop the object and look for links
660c59e338cSEd Tanous     for (auto& element : obj)
6617cf436c9SEd Tanous     {
662e479ad58SNan Zhou         bool localInLinks = inLinks;
663e479ad58SNan Zhou         if (!localInLinks)
6647cf436c9SEd Tanous         {
6657cf436c9SEd Tanous             // Check if this is a links node
666e479ad58SNan Zhou             localInLinks = element.first == "Links";
6677cf436c9SEd Tanous         }
6687cf436c9SEd Tanous         // Only traverse the parts of the tree the user asked for
6697cf436c9SEd Tanous         // Per section 7.3 of the redfish specification
670e479ad58SNan Zhou         if (localInLinks && eType == ExpandType::NotLinks)
6717cf436c9SEd Tanous         {
6727cf436c9SEd Tanous             continue;
6737cf436c9SEd Tanous         }
674e479ad58SNan Zhou         if (!localInLinks && eType == ExpandType::Links)
6757cf436c9SEd Tanous         {
6767cf436c9SEd Tanous             continue;
6777cf436c9SEd Tanous         }
67837b1f7beSEd Tanous         nlohmann::json::json_pointer newPtr = jsonPtr / element.first;
67962598e31SEd Tanous         BMCWEB_LOG_DEBUG("Traversing response at {}", newPtr);
6807cf436c9SEd Tanous 
6817cf436c9SEd Tanous         findNavigationReferencesRecursive(eType, element.second, newPtr,
68232cdb4a7SWilly Tu                                           newDepth, skipDepth, localInLinks,
68332cdb4a7SWilly Tu                                           out);
6847cf436c9SEd Tanous     }
6857cf436c9SEd Tanous }
6867cf436c9SEd Tanous 
687ad595fa6SEd Tanous // TODO: When aggregation is enabled and we receive a partially expanded
688ad595fa6SEd Tanous // response we may need need additional handling when the original URI was
689ad595fa6SEd Tanous // up tree from a top level collection.
690ad595fa6SEd Tanous // Isn't a concern until https://gerrit.openbmc.org/c/openbmc/bmcweb/+/60556
691ad595fa6SEd Tanous // lands.  May want to avoid forwarding query params when request is uptree from
692ad595fa6SEd Tanous // a top level collection.
693bd79bce8SPatrick Williams inline std::vector<ExpandNode> findNavigationReferences(
694bd79bce8SPatrick Williams     ExpandType eType, int depth, int skipDepth, nlohmann::json& jsonResponse)
6957cf436c9SEd Tanous {
6967cf436c9SEd Tanous     std::vector<ExpandNode> ret;
69772c3ae33SNan Zhou     const nlohmann::json::json_pointer root = nlohmann::json::json_pointer("");
69832cdb4a7SWilly Tu     // SkipDepth +1 since we are skipping the root by default.
69932cdb4a7SWilly Tu     findNavigationReferencesRecursive(eType, jsonResponse, root, depth,
70032cdb4a7SWilly Tu                                       skipDepth + 1, false, ret);
7017cf436c9SEd Tanous     return ret;
7027cf436c9SEd Tanous }
7037cf436c9SEd Tanous 
70472c3ae33SNan Zhou // Formats a query parameter string for the sub-query.
705b66cf2a2SNan Zhou // Returns std::nullopt on failures.
70672c3ae33SNan Zhou // This function shall handle $select when it is added.
7078ece0e45SEd Tanous // There is no need to handle parameters that's not compatible with $expand,
70872c3ae33SNan Zhou // e.g., $only, since this function will only be called in side $expand handlers
709b66cf2a2SNan Zhou inline std::optional<std::string> formatQueryForExpand(const Query& query)
71072c3ae33SNan Zhou {
71172c3ae33SNan Zhou     // query.expandLevel<=1: no need to do subqueries
71272c3ae33SNan Zhou     if (query.expandLevel <= 1)
71372c3ae33SNan Zhou     {
714b66cf2a2SNan Zhou         return "";
71572c3ae33SNan Zhou     }
71672c3ae33SNan Zhou     std::string str = "?$expand=";
71772c3ae33SNan Zhou     switch (query.expandType)
71872c3ae33SNan Zhou     {
71972c3ae33SNan Zhou         case ExpandType::Links:
72072c3ae33SNan Zhou             str += '~';
72172c3ae33SNan Zhou             break;
72272c3ae33SNan Zhou         case ExpandType::NotLinks:
72372c3ae33SNan Zhou             str += '.';
72472c3ae33SNan Zhou             break;
72572c3ae33SNan Zhou         case ExpandType::Both:
72672c3ae33SNan Zhou             str += '*';
72772c3ae33SNan Zhou             break;
728f1a1e3dcSEd Tanous         case ExpandType::None:
729f1a1e3dcSEd Tanous             return "";
7304da0490bSEd Tanous         default:
7314da0490bSEd Tanous             return std::nullopt;
732b66cf2a2SNan Zhou     }
73372c3ae33SNan Zhou     str += "($levels=";
73472c3ae33SNan Zhou     str += std::to_string(query.expandLevel - 1);
73572c3ae33SNan Zhou     str += ')';
73672c3ae33SNan Zhou     return str;
73772c3ae33SNan Zhou }
73872c3ae33SNan Zhou 
7398ece0e45SEd Tanous // Propagates the worst error code to the final response.
7403590bd1dSNan Zhou // The order of error code is (from high to low)
7413590bd1dSNan Zhou // 500 Internal Server Error
7423590bd1dSNan Zhou // 511 Network Authentication Required
7433590bd1dSNan Zhou // 510 Not Extended
7443590bd1dSNan Zhou // 508 Loop Detected
7453590bd1dSNan Zhou // 507 Insufficient Storage
7463590bd1dSNan Zhou // 506 Variant Also Negotiates
7473590bd1dSNan Zhou // 505 HTTP Version Not Supported
7483590bd1dSNan Zhou // 504 Gateway Timeout
7493590bd1dSNan Zhou // 503 Service Unavailable
7503590bd1dSNan Zhou // 502 Bad Gateway
7513590bd1dSNan Zhou // 501 Not Implemented
7523590bd1dSNan Zhou // 401 Unauthorized
7538ece0e45SEd Tanous // 451 - 409 Error codes (not listed explicitly)
7543590bd1dSNan Zhou // 408 Request Timeout
7553590bd1dSNan Zhou // 407 Proxy Authentication Required
7563590bd1dSNan Zhou // 406 Not Acceptable
7573590bd1dSNan Zhou // 405 Method Not Allowed
7583590bd1dSNan Zhou // 404 Not Found
7593590bd1dSNan Zhou // 403 Forbidden
7603590bd1dSNan Zhou // 402 Payment Required
7613590bd1dSNan Zhou // 400 Bad Request
7623590bd1dSNan Zhou inline unsigned propogateErrorCode(unsigned finalCode, unsigned subResponseCode)
7633590bd1dSNan Zhou {
7643590bd1dSNan Zhou     // We keep a explicit list for error codes that this project often uses
7658ece0e45SEd Tanous     // Higher priority codes are in lower indexes
7663590bd1dSNan Zhou     constexpr std::array<unsigned, 13> orderedCodes = {
7673590bd1dSNan Zhou         500, 507, 503, 502, 501, 401, 412, 409, 406, 405, 404, 403, 400};
7683590bd1dSNan Zhou     size_t finalCodeIndex = std::numeric_limits<size_t>::max();
7693590bd1dSNan Zhou     size_t subResponseCodeIndex = std::numeric_limits<size_t>::max();
7703590bd1dSNan Zhou     for (size_t i = 0; i < orderedCodes.size(); ++i)
7713590bd1dSNan Zhou     {
7723590bd1dSNan Zhou         if (orderedCodes[i] == finalCode)
7733590bd1dSNan Zhou         {
7743590bd1dSNan Zhou             finalCodeIndex = i;
7753590bd1dSNan Zhou         }
7763590bd1dSNan Zhou         if (orderedCodes[i] == subResponseCode)
7773590bd1dSNan Zhou         {
7783590bd1dSNan Zhou             subResponseCodeIndex = i;
7793590bd1dSNan Zhou         }
7803590bd1dSNan Zhou     }
7813590bd1dSNan Zhou     if (finalCodeIndex != std::numeric_limits<size_t>::max() &&
7823590bd1dSNan Zhou         subResponseCodeIndex != std::numeric_limits<size_t>::max())
7833590bd1dSNan Zhou     {
784bd79bce8SPatrick Williams         return finalCodeIndex <= subResponseCodeIndex
785bd79bce8SPatrick Williams                    ? finalCode
7863590bd1dSNan Zhou                    : subResponseCode;
7873590bd1dSNan Zhou     }
7883590bd1dSNan Zhou     if (subResponseCode == 500 || finalCode == 500)
7893590bd1dSNan Zhou     {
7903590bd1dSNan Zhou         return 500;
7913590bd1dSNan Zhou     }
7923590bd1dSNan Zhou     if (subResponseCode > 500 || finalCode > 500)
7933590bd1dSNan Zhou     {
7943590bd1dSNan Zhou         return std::max(finalCode, subResponseCode);
7953590bd1dSNan Zhou     }
7963590bd1dSNan Zhou     if (subResponseCode == 401)
7973590bd1dSNan Zhou     {
7983590bd1dSNan Zhou         return subResponseCode;
7993590bd1dSNan Zhou     }
8003590bd1dSNan Zhou     return std::max(finalCode, subResponseCode);
8013590bd1dSNan Zhou }
8023590bd1dSNan Zhou 
8038ece0e45SEd Tanous // Propagates all error messages into |finalResponse|
8043590bd1dSNan Zhou inline void propogateError(crow::Response& finalResponse,
8053590bd1dSNan Zhou                            crow::Response& subResponse)
8063590bd1dSNan Zhou {
8073590bd1dSNan Zhou     // no errors
8083590bd1dSNan Zhou     if (subResponse.resultInt() >= 200 && subResponse.resultInt() < 400)
8093590bd1dSNan Zhou     {
8103590bd1dSNan Zhou         return;
8113590bd1dSNan Zhou     }
8123590bd1dSNan Zhou     messages::moveErrorsToErrorJson(finalResponse.jsonValue,
8133590bd1dSNan Zhou                                     subResponse.jsonValue);
8143590bd1dSNan Zhou     finalResponse.result(
8153590bd1dSNan Zhou         propogateErrorCode(finalResponse.resultInt(), subResponse.resultInt()));
8163590bd1dSNan Zhou }
8173590bd1dSNan Zhou 
8187cf436c9SEd Tanous class MultiAsyncResp : public std::enable_shared_from_this<MultiAsyncResp>
8197cf436c9SEd Tanous {
8207cf436c9SEd Tanous   public:
8217cf436c9SEd Tanous     // This object takes a single asyncResp object as the "final" one, then
8227cf436c9SEd Tanous     // allows callers to attach sub-responses within the json tree that need
8237cf436c9SEd Tanous     // to be executed and filled into their appropriate locations.  This
8247cf436c9SEd Tanous     // class manages the final "merge" of the json resources.
8258a592810SEd Tanous     MultiAsyncResp(crow::App& appIn,
8267cf436c9SEd Tanous                    std::shared_ptr<bmcweb::AsyncResp> finalResIn) :
827bd79bce8SPatrick Williams         app(appIn), finalRes(std::move(finalResIn))
8287cf436c9SEd Tanous     {}
8297cf436c9SEd Tanous 
8307cf436c9SEd Tanous     void addAwaitingResponse(
83102cad96eSEd Tanous         const std::shared_ptr<bmcweb::AsyncResp>& res,
8327cf436c9SEd Tanous         const nlohmann::json::json_pointer& finalExpandLocation)
8337cf436c9SEd Tanous     {
8347cf436c9SEd Tanous         res->res.setCompleteRequestHandler(std::bind_front(
83572c3ae33SNan Zhou             placeResultStatic, shared_from_this(), finalExpandLocation));
8367cf436c9SEd Tanous     }
8377cf436c9SEd Tanous 
83872c3ae33SNan Zhou     void placeResult(const nlohmann::json::json_pointer& locationToPlace,
8397cf436c9SEd Tanous                      crow::Response& res)
8407cf436c9SEd Tanous     {
84162598e31SEd Tanous         BMCWEB_LOG_DEBUG("placeResult for {}", locationToPlace);
8423590bd1dSNan Zhou         propogateError(finalRes->res, res);
8433590bd1dSNan Zhou         if (!res.jsonValue.is_object() || res.jsonValue.empty())
8443590bd1dSNan Zhou         {
8453590bd1dSNan Zhou             return;
8463590bd1dSNan Zhou         }
8477cf436c9SEd Tanous         nlohmann::json& finalObj = finalRes->res.jsonValue[locationToPlace];
8487cf436c9SEd Tanous         finalObj = std::move(res.jsonValue);
8497cf436c9SEd Tanous     }
8507cf436c9SEd Tanous 
85172c3ae33SNan Zhou     // Handles the very first level of Expand, and starts a chain of sub-queries
85272c3ae33SNan Zhou     // for deeper levels.
85332cdb4a7SWilly Tu     void startQuery(const Query& query, const Query& delegated)
85472c3ae33SNan Zhou     {
855ad595fa6SEd Tanous         std::vector<ExpandNode> nodes = findNavigationReferences(
85632cdb4a7SWilly Tu             query.expandType, query.expandLevel, delegated.expandLevel,
85732cdb4a7SWilly Tu             finalRes->res.jsonValue);
85862598e31SEd Tanous         BMCWEB_LOG_DEBUG("{} nodes to traverse", nodes.size());
859b66cf2a2SNan Zhou         const std::optional<std::string> queryStr = formatQueryForExpand(query);
860b66cf2a2SNan Zhou         if (!queryStr)
861b66cf2a2SNan Zhou         {
862b66cf2a2SNan Zhou             messages::internalError(finalRes->res);
863b66cf2a2SNan Zhou             return;
864b66cf2a2SNan Zhou         }
8657cf436c9SEd Tanous         for (const ExpandNode& node : nodes)
8667cf436c9SEd Tanous         {
867b66cf2a2SNan Zhou             const std::string subQuery = node.uri + *queryStr;
86862598e31SEd Tanous             BMCWEB_LOG_DEBUG("URL of subquery:  {}", subQuery);
8697cf436c9SEd Tanous             std::error_code ec;
870102a4cdaSJonathan Doman             auto newReq = std::make_shared<crow::Request>(
871102a4cdaSJonathan Doman                 crow::Request::Body{boost::beast::http::verb::get, subQuery,
872102a4cdaSJonathan Doman                                     11},
8737cf436c9SEd Tanous                 ec);
8747cf436c9SEd Tanous             if (ec)
8757cf436c9SEd Tanous             {
87672c3ae33SNan Zhou                 messages::internalError(finalRes->res);
8777cf436c9SEd Tanous                 return;
8787cf436c9SEd Tanous             }
8797cf436c9SEd Tanous 
8807cf436c9SEd Tanous             auto asyncResp = std::make_shared<bmcweb::AsyncResp>();
88162598e31SEd Tanous             BMCWEB_LOG_DEBUG("setting completion handler on {}",
88262598e31SEd Tanous                              logPtr(&asyncResp->res));
88372c3ae33SNan Zhou 
88472c3ae33SNan Zhou             addAwaitingResponse(asyncResp, node.location);
8857cf436c9SEd Tanous             app.handle(newReq, asyncResp);
8867cf436c9SEd Tanous         }
8877cf436c9SEd Tanous     }
8887cf436c9SEd Tanous 
8897cf436c9SEd Tanous   private:
89072c3ae33SNan Zhou     static void
89172c3ae33SNan Zhou         placeResultStatic(const std::shared_ptr<MultiAsyncResp>& multi,
8927cf436c9SEd Tanous                           const nlohmann::json::json_pointer& locationToPlace,
8937cf436c9SEd Tanous                           crow::Response& res)
8947cf436c9SEd Tanous     {
89572c3ae33SNan Zhou         multi->placeResult(locationToPlace, res);
8967cf436c9SEd Tanous     }
8977cf436c9SEd Tanous 
8987cf436c9SEd Tanous     crow::App& app;
8997cf436c9SEd Tanous     std::shared_ptr<bmcweb::AsyncResp> finalRes;
9007cf436c9SEd Tanous };
9017cf436c9SEd Tanous 
9022a68dc80SEd Tanous inline void processTopAndSkip(const Query& query, crow::Response& res)
9032a68dc80SEd Tanous {
9043648c8beSEd Tanous     if (!query.skip && !query.top)
9053648c8beSEd Tanous     {
9063648c8beSEd Tanous         // No work to do.
9073648c8beSEd Tanous         return;
9083648c8beSEd Tanous     }
9092a68dc80SEd Tanous     nlohmann::json::object_t* obj =
9102a68dc80SEd Tanous         res.jsonValue.get_ptr<nlohmann::json::object_t*>();
9112a68dc80SEd Tanous     if (obj == nullptr)
9122a68dc80SEd Tanous     {
9132a68dc80SEd Tanous         // Shouldn't be possible.  All responses should be objects.
9142a68dc80SEd Tanous         messages::internalError(res);
9152a68dc80SEd Tanous         return;
9162a68dc80SEd Tanous     }
9172a68dc80SEd Tanous 
91862598e31SEd Tanous     BMCWEB_LOG_DEBUG("Handling top/skip");
9192a68dc80SEd Tanous     nlohmann::json::object_t::iterator members = obj->find("Members");
9202a68dc80SEd Tanous     if (members == obj->end())
9212a68dc80SEd Tanous     {
9222a68dc80SEd Tanous         // From the Redfish specification 7.3.1
9232a68dc80SEd Tanous         // ... the HTTP 400 Bad Request status code with the
9242a68dc80SEd Tanous         // QueryNotSupportedOnResource message from the Base Message Registry
9252a68dc80SEd Tanous         // for any supported query parameters that apply only to resource
9262a68dc80SEd Tanous         // collections but are used on singular resources.
9272a68dc80SEd Tanous         messages::queryNotSupportedOnResource(res);
9282a68dc80SEd Tanous         return;
9292a68dc80SEd Tanous     }
9302a68dc80SEd Tanous 
9312a68dc80SEd Tanous     nlohmann::json::array_t* arr =
9322a68dc80SEd Tanous         members->second.get_ptr<nlohmann::json::array_t*>();
9332a68dc80SEd Tanous     if (arr == nullptr)
9342a68dc80SEd Tanous     {
9352a68dc80SEd Tanous         messages::internalError(res);
9362a68dc80SEd Tanous         return;
9372a68dc80SEd Tanous     }
9382a68dc80SEd Tanous 
9393648c8beSEd Tanous     if (query.skip)
9403648c8beSEd Tanous     {
9413648c8beSEd Tanous         // Per section 7.3.1 of the Redfish specification, $skip is run before
9423648c8beSEd Tanous         // $top Can only skip as many values as we have
9433648c8beSEd Tanous         size_t skip = std::min(arr->size(), *query.skip);
9442a68dc80SEd Tanous         arr->erase(arr->begin(), arr->begin() + static_cast<ssize_t>(skip));
9453648c8beSEd Tanous     }
9463648c8beSEd Tanous     if (query.top)
9473648c8beSEd Tanous     {
9483648c8beSEd Tanous         size_t top = std::min(arr->size(), *query.top);
9492a68dc80SEd Tanous         arr->erase(arr->begin() + static_cast<ssize_t>(top), arr->end());
9502a68dc80SEd Tanous     }
9513648c8beSEd Tanous }
9522a68dc80SEd Tanous 
953827c4902SNan Zhou // Given a JSON subtree |currRoot|, this function erases leaves whose keys are
954827c4902SNan Zhou // not in the |currNode| Trie node.
955827c4902SNan Zhou inline void recursiveSelect(nlohmann::json& currRoot,
956827c4902SNan Zhou                             const SelectTrieNode& currNode)
957e155ab54SNan Zhou {
958e155ab54SNan Zhou     nlohmann::json::object_t* object =
959e155ab54SNan Zhou         currRoot.get_ptr<nlohmann::json::object_t*>();
960e155ab54SNan Zhou     if (object != nullptr)
961e155ab54SNan Zhou     {
96262598e31SEd Tanous         BMCWEB_LOG_DEBUG("Current JSON is an object");
963e155ab54SNan Zhou         auto it = currRoot.begin();
964e155ab54SNan Zhou         while (it != currRoot.end())
965e155ab54SNan Zhou         {
966e155ab54SNan Zhou             auto nextIt = std::next(it);
96762598e31SEd Tanous             BMCWEB_LOG_DEBUG("key={}", it.key());
968827c4902SNan Zhou             const SelectTrieNode* nextNode = currNode.find(it.key());
9695c9fb2d6SNan Zhou             // Per the Redfish spec section 7.3.3, the service shall select
9705c9fb2d6SNan Zhou             // certain properties as if $select was omitted. This applies to
9715c9fb2d6SNan Zhou             // every TrieNode that contains leaves and the root.
9725c9fb2d6SNan Zhou             constexpr std::array<std::string_view, 5> reservedProperties = {
9735c9fb2d6SNan Zhou                 "@odata.id", "@odata.type", "@odata.context", "@odata.etag",
9745c9fb2d6SNan Zhou                 "error"};
9753544d2a7SEd Tanous             bool reserved = std::ranges::find(reservedProperties, it.key()) !=
9763544d2a7SEd Tanous                             reservedProperties.end();
9775c9fb2d6SNan Zhou             if (reserved || (nextNode != nullptr && nextNode->isSelected()))
978e155ab54SNan Zhou             {
979e155ab54SNan Zhou                 it = nextIt;
980e155ab54SNan Zhou                 continue;
981e155ab54SNan Zhou             }
982827c4902SNan Zhou             if (nextNode != nullptr)
983e155ab54SNan Zhou             {
98462598e31SEd Tanous                 BMCWEB_LOG_DEBUG("Recursively select: {}", it.key());
985827c4902SNan Zhou                 recursiveSelect(*it, *nextNode);
986e155ab54SNan Zhou                 it = nextIt;
987e155ab54SNan Zhou                 continue;
988e155ab54SNan Zhou             }
98962598e31SEd Tanous             BMCWEB_LOG_DEBUG("{} is getting removed!", it.key());
990e155ab54SNan Zhou             it = currRoot.erase(it);
991e155ab54SNan Zhou         }
992e155ab54SNan Zhou     }
9935c9fb2d6SNan Zhou     nlohmann::json::array_t* array =
9945c9fb2d6SNan Zhou         currRoot.get_ptr<nlohmann::json::array_t*>();
9955c9fb2d6SNan Zhou     if (array != nullptr)
9965c9fb2d6SNan Zhou     {
99762598e31SEd Tanous         BMCWEB_LOG_DEBUG("Current JSON is an array");
9985c9fb2d6SNan Zhou         // Array index is omitted, so reuse the same Trie node
9995c9fb2d6SNan Zhou         for (nlohmann::json& nextRoot : *array)
10005c9fb2d6SNan Zhou         {
10015c9fb2d6SNan Zhou             recursiveSelect(nextRoot, currNode);
10025c9fb2d6SNan Zhou         }
10035c9fb2d6SNan Zhou     }
1004e155ab54SNan Zhou }
1005e155ab54SNan Zhou 
1006e155ab54SNan Zhou // The current implementation of $select still has the following TODOs due to
1007e155ab54SNan Zhou //  ambiguity and/or complexity.
10085c9fb2d6SNan Zhou // 1. combined with $expand; https://github.com/DMTF/Redfish/issues/5058 was
1009e155ab54SNan Zhou // created for clarification.
10105c9fb2d6SNan Zhou // 2. respect the full odata spec; e.g., deduplication, namespace, star (*),
1011e155ab54SNan Zhou // etc.
1012e155ab54SNan Zhou inline void processSelect(crow::Response& intermediateResponse,
1013827c4902SNan Zhou                           const SelectTrieNode& trieRoot)
1014e155ab54SNan Zhou {
101562598e31SEd Tanous     BMCWEB_LOG_DEBUG("Process $select quary parameter");
1016827c4902SNan Zhou     recursiveSelect(intermediateResponse.jsonValue, trieRoot);
1017e155ab54SNan Zhou }
1018e155ab54SNan Zhou 
10197cf436c9SEd Tanous inline void
102032cdb4a7SWilly Tu     processAllParams(crow::App& app, const Query& query, const Query& delegated,
10217cf436c9SEd Tanous                      std::function<void(crow::Response&)>& completionHandler,
10227cf436c9SEd Tanous                      crow::Response& intermediateResponse)
1023f4c99e70SEd Tanous {
1024f4c99e70SEd Tanous     if (!completionHandler)
1025f4c99e70SEd Tanous     {
102662598e31SEd Tanous         BMCWEB_LOG_DEBUG("Function was invalid?");
1027f4c99e70SEd Tanous         return;
1028f4c99e70SEd Tanous     }
1029f4c99e70SEd Tanous 
103062598e31SEd Tanous     BMCWEB_LOG_DEBUG("Processing query params");
1031f4c99e70SEd Tanous     // If the request failed, there's no reason to even try to run query
1032f4c99e70SEd Tanous     // params.
1033f4c99e70SEd Tanous     if (intermediateResponse.resultInt() < 200 ||
1034f4c99e70SEd Tanous         intermediateResponse.resultInt() >= 400)
1035f4c99e70SEd Tanous     {
1036f4c99e70SEd Tanous         completionHandler(intermediateResponse);
1037f4c99e70SEd Tanous         return;
1038f4c99e70SEd Tanous     }
1039f4c99e70SEd Tanous     if (query.isOnly)
1040f4c99e70SEd Tanous     {
1041f4c99e70SEd Tanous         processOnly(app, intermediateResponse, completionHandler);
1042f4c99e70SEd Tanous         return;
1043f4c99e70SEd Tanous     }
10442a68dc80SEd Tanous 
10453648c8beSEd Tanous     if (query.top || query.skip)
10462a68dc80SEd Tanous     {
10472a68dc80SEd Tanous         processTopAndSkip(query, intermediateResponse);
10482a68dc80SEd Tanous     }
10492a68dc80SEd Tanous 
10507cf436c9SEd Tanous     if (query.expandType != ExpandType::None)
10517cf436c9SEd Tanous     {
105262598e31SEd Tanous         BMCWEB_LOG_DEBUG("Executing expand query");
105313548d85SEd Tanous         auto asyncResp = std::make_shared<bmcweb::AsyncResp>(
105413548d85SEd Tanous             std::move(intermediateResponse));
10557cf436c9SEd Tanous 
105613548d85SEd Tanous         asyncResp->res.setCompleteRequestHandler(std::move(completionHandler));
105713548d85SEd Tanous         auto multi = std::make_shared<MultiAsyncResp>(app, asyncResp);
105832cdb4a7SWilly Tu         multi->startQuery(query, delegated);
10597cf436c9SEd Tanous         return;
10607cf436c9SEd Tanous     }
1061e155ab54SNan Zhou 
106225991f7dSEd Tanous     if (query.filter)
106325991f7dSEd Tanous     {
1064f80a87f2SEd Tanous         applyFilterToCollection(intermediateResponse.jsonValue, *query.filter);
106525991f7dSEd Tanous     }
106625991f7dSEd Tanous 
1067e155ab54SNan Zhou     // According to Redfish Spec Section 7.3.1, $select is the last parameter to
1068e155ab54SNan Zhou     // to process
1069827c4902SNan Zhou     if (!query.selectTrie.root.empty())
1070e155ab54SNan Zhou     {
1071827c4902SNan Zhou         processSelect(intermediateResponse, query.selectTrie.root);
1072e155ab54SNan Zhou     }
1073e155ab54SNan Zhou 
1074f4c99e70SEd Tanous     completionHandler(intermediateResponse);
1075f4c99e70SEd Tanous }
1076f4c99e70SEd Tanous 
1077f4c99e70SEd Tanous } // namespace query_param
1078f4c99e70SEd Tanous } // namespace redfish
1079