RedfishRestExImpl.c 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164
  1. /** @file
  2. RestExDxe support functions implementation.
  3. Copyright (c) 2019, Intel Corporation. All rights reserved.<BR>
  4. (C) Copyright 2020 Hewlett Packard Enterprise Development LP<BR>
  5. SPDX-License-Identifier: BSD-2-Clause-Patent
  6. **/
  7. #include <Uefi.h>
  8. #include "RedfishRestExInternal.h"
  9. /**
  10. Create a new TLS session becuase the previous on is closed.
  11. status.
  12. @param[in] Instance Pointer to EFI_REST_EX_PROTOCOL instance for a particular
  13. REST service.
  14. @retval EFI_SUCCESS operation succeeded.
  15. @retval EFI_ERROR Other errors.
  16. **/
  17. EFI_STATUS
  18. ResetHttpTslSession (
  19. IN RESTEX_INSTANCE *Instance
  20. )
  21. {
  22. EFI_STATUS Status;
  23. DEBUG ((DEBUG_INFO, "%a: TCP connection is finished. Could be TSL session closure, reset HTTP instance for the new TLS session.\n", __FUNCTION__));
  24. Status = Instance->HttpIo.Http->Configure (Instance->HttpIo.Http, NULL);
  25. if (EFI_ERROR (Status)) {
  26. DEBUG ((DEBUG_ERROR, "%a: Error to reset HTTP instance.\n", __FUNCTION__));
  27. return Status;
  28. }
  29. Status = Instance->HttpIo.Http->Configure (Instance->HttpIo.Http, &((EFI_REST_EX_HTTP_CONFIG_DATA *)Instance->ConfigData)->HttpConfigData);
  30. if (EFI_ERROR (Status)) {
  31. DEBUG ((DEBUG_ERROR, "%a: Error to re-initiate HTTP instance.\n", __FUNCTION__));
  32. }
  33. return Status;
  34. }
  35. /**
  36. This function check
  37. @param[in] Instance Pointer to EFI_REST_EX_PROTOCOL instance for a particular
  38. REST service.
  39. @param[in] HttpIoReceiveStatus This is the status return from HttpIoRecvResponse
  40. @retval EFI_SUCCESS The payload receive from Redfish service in successfully.
  41. @retval EFI_NOT_READY May need to resend the HTTP request.
  42. @retval EFI_DEVICE_ERROR Something wrong and can't be resolved.
  43. @retval Others Other errors as indicated.
  44. **/
  45. EFI_STATUS
  46. RedfishCheckHttpReceiveStatus (
  47. IN RESTEX_INSTANCE *Instance,
  48. IN EFI_STATUS HttpIoReceiveStatus
  49. )
  50. {
  51. EFI_STATUS Status;
  52. EFI_STATUS ReturnStatus;
  53. if (!EFI_ERROR (HttpIoReceiveStatus)) {
  54. ReturnStatus = EFI_SUCCESS;
  55. } else if (EFI_ERROR (HttpIoReceiveStatus) && (HttpIoReceiveStatus != EFI_CONNECTION_FIN)) {
  56. if ((Instance->Flags & RESTEX_INSTANCE_FLAGS_TCP_ERROR_RETRY) == 0) {
  57. DEBUG ((DEBUG_ERROR, "%a: TCP error, reset HTTP session.\n", __FUNCTION__));
  58. Instance->Flags |= RESTEX_INSTANCE_FLAGS_TCP_ERROR_RETRY;
  59. gBS->Stall (500);
  60. Status = ResetHttpTslSession (Instance);
  61. if (EFI_ERROR (Status)) {
  62. DEBUG ((DEBUG_ERROR, "%a: Reset HTTP instance fail.\n", __FUNCTION__));
  63. ReturnStatus = EFI_DEVICE_ERROR;
  64. } else {
  65. return EFI_NOT_READY;
  66. }
  67. } else {
  68. ReturnStatus = EFI_DEVICE_ERROR;
  69. }
  70. } else {
  71. if (HttpIoReceiveStatus == EFI_CONNECTION_FIN) {
  72. if ((Instance->Flags & RESTEX_INSTANCE_FLAGS_TLS_RETRY) != 0) {
  73. DEBUG ((DEBUG_ERROR, "%a: REST_EX Send and receive fail even with a new TLS session.\n", __FUNCTION__));
  74. ReturnStatus = EFI_DEVICE_ERROR;
  75. }
  76. Instance->Flags |= RESTEX_INSTANCE_FLAGS_TLS_RETRY;
  77. Status = ResetHttpTslSession (Instance);
  78. if (EFI_ERROR (Status)) {
  79. DEBUG ((DEBUG_ERROR, "%a: Reset HTTP instance fail.\n", __FUNCTION__));
  80. ReturnStatus = EFI_DEVICE_ERROR;
  81. }
  82. return EFI_NOT_READY;
  83. }
  84. }
  85. //
  86. // Clean TLS new session retry and error try flags.
  87. //
  88. Instance->Flags &= ~(RESTEX_INSTANCE_FLAGS_TLS_RETRY | RESTEX_INSTANCE_FLAGS_TCP_ERROR_RETRY);
  89. return ReturnStatus;
  90. }
  91. /**
  92. This function send the HTTP request without body to see
  93. if the write to URL is permitted by Redfish service. This function
  94. checks if the HTTP request has Content-length in HTTP header. If yes,
  95. set HTTP body to NULL and then send to service. Check the HTTP status
  96. for the firther actions.
  97. @param[in] This Pointer to EFI_REST_EX_PROTOCOL instance for a particular
  98. REST service.
  99. @param[in] RequestMessage Pointer to the HTTP request data for this resource
  100. @param[in] PreservedRequestHeaders The pointer to save the request headers
  101. @param[in] ItsWrite This is write method to URL.
  102. @retval EFI_INVALID_PARAMETER Improper given parameters.
  103. @retval EFI_SUCCESS This HTTP request is free to send to Redfish service.
  104. @retval EFI_OUT_OF_RESOURCES NOt enough memory to process.
  105. @retval EFI_ACCESS_DENIED Not allowed to write to this URL.
  106. @retval Others Other errors as indicated.
  107. **/
  108. EFI_STATUS
  109. RedfishHttpAddExpectation (
  110. IN EFI_REST_EX_PROTOCOL *This,
  111. IN EFI_HTTP_MESSAGE *RequestMessage,
  112. IN EFI_HTTP_HEADER **PreservedRequestHeaders,
  113. IN BOOLEAN *ItsWrite
  114. )
  115. {
  116. EFI_HTTP_HEADER *NewHeaders;
  117. if ((This == NULL) || (RequestMessage == NULL)) {
  118. return EFI_INVALID_PARAMETER;
  119. }
  120. *ItsWrite = FALSE;
  121. if (PreservedRequestHeaders != NULL) {
  122. *PreservedRequestHeaders = RequestMessage->Headers;
  123. }
  124. if ((RequestMessage->Data.Request->Method != HttpMethodPut) && (RequestMessage->Data.Request->Method != HttpMethodPost) &&
  125. (RequestMessage->Data.Request->Method != HttpMethodPatch))
  126. {
  127. return EFI_SUCCESS;
  128. }
  129. *ItsWrite = TRUE;
  130. NewHeaders = AllocateZeroPool ((RequestMessage->HeaderCount + 1) * sizeof (EFI_HTTP_HEADER));
  131. CopyMem ((VOID *)NewHeaders, (VOID *)RequestMessage->Headers, RequestMessage->HeaderCount * sizeof (EFI_HTTP_HEADER));
  132. HttpSetFieldNameAndValue (NewHeaders + RequestMessage->HeaderCount, HTTP_HEADER_EXPECT, HTTP_EXPECT_100_CONTINUE);
  133. RequestMessage->HeaderCount++;
  134. RequestMessage->Headers = NewHeaders;
  135. return EFI_SUCCESS;
  136. }