GRPCClientStateMachine.swift 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672
  1. /*
  2. * Copyright 2019, gRPC Authors All rights reserved.
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. import Foundation
  17. import NIO
  18. import NIOHTTP1
  19. import NIOHPACK
  20. import Logging
  21. import SwiftProtobuf
  22. enum ReceiveResponseHeadError: Error, Equatable {
  23. /// The 'content-type' header was missing or the value is not supported by this implementation.
  24. case invalidContentType(String?)
  25. /// The HTTP response status from the server was not 200 OK.
  26. case invalidHTTPStatus(String?)
  27. /// The encoding used by the server is not supported.
  28. case unsupportedMessageEncoding(String)
  29. /// An invalid state was encountered. This is a serious implementation error.
  30. case invalidState
  31. }
  32. enum ReceiveEndOfResponseStreamError: Error {
  33. /// The 'content-type' header was missing or the value is not supported by this implementation.
  34. case invalidContentType(String?)
  35. /// The HTTP response status from the server was not 200 OK.
  36. case invalidHTTPStatus(String?)
  37. /// The HTTP response status from the server was not 200 OK but the "grpc-status" header contained
  38. /// a valid value.
  39. case invalidHTTPStatusWithGRPCStatus(GRPCStatus)
  40. /// An invalid state was encountered. This is a serious implementation error.
  41. case invalidState
  42. }
  43. enum SendRequestHeadersError: Error {
  44. /// An invalid state was encountered. This is a serious implementation error.
  45. case invalidState
  46. }
  47. enum SendEndOfRequestStreamError: Error {
  48. /// The request stream has already been closed. This may happen if the RPC was cancelled, timed
  49. /// out, the server terminated the RPC, or the user explicitly closed the stream multiple times.
  50. case alreadyClosed
  51. /// An invalid state was encountered. This is a serious implementation error.
  52. case invalidState
  53. }
  54. /// A state machine for a single gRPC call from the perspective of a client.
  55. ///
  56. /// See: https://github.com/grpc/grpc/blob/master/doc/PROTOCOL-HTTP2.md
  57. struct GRPCClientStateMachine<Request: GRPCPayload, Response: GRPCPayload> {
  58. /// The combined state of the request (client) and response (server) streams for an RPC call.
  59. ///
  60. /// The following states are not possible:
  61. /// - `.clientIdleServerActive`: The client must initiate the call before the server moves
  62. /// from the idle state.
  63. /// - `.clientIdleServerClosed`: The client must initiate the call before the server moves from
  64. /// the idle state.
  65. /// - `.clientActiveServerClosed`: The client may not stream if the server is closed.
  66. ///
  67. /// Note: when a peer (client or server) state is "active" it means that messages _may_ be sent or
  68. /// received. That is, the headers for the stream have been processed by the state machine and
  69. /// end-of-stream has not yet been processed. A stream may expect any number of messages (i.e. up
  70. /// to one for a unary call and many for a streaming call).
  71. enum State {
  72. /// Initial state. Neither request stream nor response stream have been initiated. Holds the
  73. /// pending write state for the request stream and arity for the response stream, respectively.
  74. ///
  75. /// Valid transitions:
  76. /// - `clientActiveServerIdle`: if the client initiates the RPC,
  77. /// - `clientClosedServerClosed`: if the client terminates the RPC.
  78. case clientIdleServerIdle(pendingWriteState: PendingWriteState, readArity: MessageArity)
  79. /// The client has initiated an RPC and has not received initial metadata from the server. Holds
  80. /// the writing state for request stream and arity for the response stream.
  81. ///
  82. /// Valid transitions:
  83. /// - `clientActiveServerActive`: if the server acknowledges the RPC initiation,
  84. /// - `clientClosedServerIdle`: if the client closes the request stream,
  85. /// - `clientClosedServerClosed`: if the client terminates the RPC or the server terminates the
  86. /// RPC with a "trailers-only" response.
  87. case clientActiveServerIdle(writeState: WriteState, pendingReadState: PendingReadState)
  88. /// The client has indicated to the server that it has finished sending requests. The server
  89. /// has not yet sent response headers for the RPC. Holds the response stream arity.
  90. ///
  91. /// Valid transitions:
  92. /// - `clientClosedServerActive`: if the server acknowledges the RPC initiation,
  93. /// - `clientClosedServerClosed`: if the client terminates the RPC or the server terminates the
  94. /// RPC with a "trailers-only" response.
  95. case clientClosedServerIdle(pendingReadState: PendingReadState)
  96. /// The client has initiated the RPC and the server has acknowledged it. Messages may have been
  97. /// sent and/or received. Holds the request stream write state and response stream read state.
  98. ///
  99. /// Valid transitions:
  100. /// - `clientClosedServerActive`: if the client closes the request stream,
  101. /// - `clientClosedServerClosed`: if the client or server terminates the RPC.
  102. case clientActiveServerActive(writeState: WriteState, readState: ReadState)
  103. /// The client has indicated to the server that it has finished sending requests. The server
  104. /// has acknowledged the RPC. Holds the response stream read state.
  105. ///
  106. /// Valid transitions:
  107. /// - `clientClosedServerClosed`: if the client or server terminate the RPC.
  108. case clientClosedServerActive(readState: ReadState)
  109. /// The RPC has terminated. There are no valid transitions from this state.
  110. case clientClosedServerClosed
  111. }
  112. /// The current state of the state machine.
  113. internal private(set) var state: State {
  114. didSet {
  115. switch (oldValue, self.state) {
  116. // All valid transitions:
  117. case (.clientIdleServerIdle, .clientActiveServerIdle),
  118. (.clientIdleServerIdle, .clientClosedServerClosed),
  119. (.clientActiveServerIdle, .clientActiveServerActive),
  120. (.clientActiveServerIdle, .clientClosedServerIdle),
  121. (.clientActiveServerIdle, .clientClosedServerClosed),
  122. (.clientClosedServerIdle, .clientClosedServerActive),
  123. (.clientClosedServerIdle, .clientClosedServerClosed),
  124. (.clientActiveServerActive, .clientClosedServerActive),
  125. (.clientActiveServerActive, .clientClosedServerClosed),
  126. (.clientClosedServerActive, .clientClosedServerClosed):
  127. break
  128. // Self transitions, also valid:
  129. case (.clientIdleServerIdle, .clientIdleServerIdle),
  130. (.clientActiveServerIdle, .clientActiveServerIdle),
  131. (.clientClosedServerIdle, .clientClosedServerIdle),
  132. (.clientActiveServerActive, .clientActiveServerActive),
  133. (.clientClosedServerActive, .clientClosedServerActive),
  134. (.clientClosedServerClosed, .clientClosedServerClosed):
  135. break
  136. default:
  137. preconditionFailure("invalid state transition from '\(oldValue)' to '\(self.state)'")
  138. }
  139. }
  140. }
  141. /// Creates a state machine representing a gRPC client's request and response stream state.
  142. ///
  143. /// - Parameter requestArity: The expected number of messages on the request stream.
  144. /// - Parameter responseArity: The expected number of messages on the response stream.
  145. init(requestArity: MessageArity, responseArity: MessageArity) {
  146. self.state = .clientIdleServerIdle(
  147. pendingWriteState: .init(arity: requestArity, contentType: .protobuf),
  148. readArity: responseArity
  149. )
  150. }
  151. /// Creates a state machine representing a gRPC client's request and response stream state.
  152. ///
  153. /// - Parameter state: The initial state of the state machine.
  154. init(state: State) {
  155. self.state = state
  156. }
  157. /// Initiates an RPC.
  158. ///
  159. /// The only valid state transition is:
  160. /// - `.clientIdleServerIdle` → `.clientActiveServerIdle`
  161. ///
  162. /// All other states will result in an `.invalidState` error.
  163. ///
  164. /// On success the state will transition to `.clientActiveServerIdle`.
  165. ///
  166. /// - Parameter requestHead: The client request head for the RPC.
  167. mutating func sendRequestHeaders(
  168. requestHead: _GRPCRequestHead
  169. ) -> Result<HPACKHeaders, SendRequestHeadersError> {
  170. return self.state.sendRequestHeaders(requestHead: requestHead)
  171. }
  172. /// Formats a request to send to the server.
  173. ///
  174. /// The client must be streaming in order for this to return successfully. Therefore the valid
  175. /// state transitions are:
  176. /// - `.clientActiveServerIdle` → `.clientActiveServerIdle`
  177. /// - `.clientActiveServerActive` → `.clientActiveServerActive`
  178. ///
  179. /// The client should not attempt to send requests once the request stream is closed, that is
  180. /// from one of the following states:
  181. /// - `.clientClosedServerIdle`
  182. /// - `.clientClosedServerActive`
  183. /// - `.clientClosedServerClosed`
  184. /// Doing so will result in a `.cardinalityViolation`.
  185. ///
  186. /// Sending a message when both peers are idle (in the `.clientIdleServerIdle` state) will result
  187. /// in a `.invalidState` error.
  188. ///
  189. /// - Parameter message: The `Request` to send to the server.
  190. /// - Parameter allocator: A `ByteBufferAllocator` to allocate the buffer into which the encoded
  191. /// request will be written.
  192. mutating func sendRequest(
  193. _ message: Request,
  194. compressed: Bool,
  195. allocator: ByteBufferAllocator
  196. ) -> Result<ByteBuffer, MessageWriteError> {
  197. return self.state.sendRequest(message, compressed: compressed, allocator: allocator)
  198. }
  199. /// Closes the request stream.
  200. ///
  201. /// The client must be streaming requests in order to terminate the request stream. Valid
  202. /// states transitions are:
  203. /// - `.clientActiveServerIdle` → `.clientClosedServerIdle`
  204. /// - `.clientActiveServerActive` → `.clientClosedServerActive`
  205. ///
  206. /// The client should not attempt to close the request stream if it is already closed, that is
  207. /// from one of the following states:
  208. /// - `.clientClosedServerIdle`
  209. /// - `.clientClosedServerActive`
  210. /// - `.clientClosedServerClosed`
  211. /// Doing so will result in an `.alreadyClosed` error.
  212. ///
  213. /// Closing the request stream when both peers are idle (in the `.clientIdleServerIdle` state)
  214. /// will result in a `.invalidState` error.
  215. mutating func sendEndOfRequestStream() -> Result<Void, SendEndOfRequestStreamError> {
  216. return self.state.sendEndOfRequestStream()
  217. }
  218. /// Receive an acknowledgement of the RPC from the server. This **must not** be a "Trailers-Only"
  219. /// response.
  220. ///
  221. /// The server must be idle in order to receive response headers. The valid state transitions are:
  222. /// - `.clientActiveServerIdle` → `.clientActiveServerActive`
  223. /// - `.clientClosedServerIdle` → `.clientClosedServerActive`
  224. ///
  225. /// The response head will be parsed and validated against the gRPC specification. The following
  226. /// errors may be returned:
  227. /// - `.invalidHTTPStatus` if the status was not "200",
  228. /// - `.invalidContentType` if the "content-type" header does not start with "application/grpc",
  229. /// - `.unsupportedMessageEncoding` if the "grpc-encoding" header is not supported.
  230. ///
  231. /// It is not possible to receive response headers from the following states:
  232. /// - `.clientIdleServerIdle`
  233. /// - `.clientActiveServerActive`
  234. /// - `.clientClosedServerActive`
  235. /// - `.clientClosedServerClosed`
  236. /// Doing so will result in a `.invalidState` error.
  237. ///
  238. /// - Parameter headers: The headers received from the server.
  239. mutating func receiveResponseHeaders(
  240. _ headers: HPACKHeaders
  241. ) -> Result<Void, ReceiveResponseHeadError> {
  242. return self.state.receiveResponseHeaders(headers)
  243. }
  244. /// Read a response buffer from the server and return any decoded messages.
  245. ///
  246. /// If the response stream has an expected count of `.one` then this function is guaranteed to
  247. /// produce *at most* one `Response` in the `Result`.
  248. ///
  249. /// To receive a response buffer the server must be streaming. Valid states are:
  250. /// - `.clientClosedServerActive` → `.clientClosedServerActive`
  251. /// - `.clientActiveServerActive` → `.clientActiveServerActive`
  252. ///
  253. /// This function will read all of the bytes in the `buffer` and attempt to produce as many
  254. /// messages as possible. This may lead to a number of errors:
  255. /// - `.cardinalityViolation` if more than one message is received when the state reader is
  256. /// expects at most one.
  257. /// - `.leftOverBytes` if bytes remain in the buffer after reading one message when at most one
  258. /// message is expected.
  259. /// - `.deserializationFailed` if the message could not be deserialized.
  260. ///
  261. /// It is not possible to receive response headers from the following states:
  262. /// - `.clientIdleServerIdle`
  263. /// - `.clientClosedServerActive`
  264. /// - `.clientActiveServerActive`
  265. /// - `.clientClosedServerClosed`
  266. /// Doing so will result in a `.invalidState` error.
  267. ///
  268. /// - Parameter buffer: A buffer of bytes received from the server.
  269. mutating func receiveResponseBuffer(
  270. _ buffer: inout ByteBuffer
  271. ) -> Result<[Response], MessageReadError> {
  272. return self.state.receiveResponseBuffer(&buffer)
  273. }
  274. /// Receive the end of the response stream from the server and parse the results into
  275. /// a `GRPCStatus`.
  276. ///
  277. /// To close the response stream the server must be streaming or idle (since the server may choose
  278. /// to 'fast fail' the RPC). Valid states are:
  279. /// - `.clientActiveServerIdle` → `.clientClosedServerClosed`
  280. /// - `.clientActiveServerActive` → `.clientClosedServerClosed`
  281. /// - `.clientClosedServerIdle` → `.clientClosedServerClosed`
  282. /// - `.clientClosedServerActive` → `.clientClosedServerClosed`
  283. ///
  284. /// It is not possible to receive an end-of-stream if the RPC has not been initiated or has
  285. /// already been terminated. That is, in one of the following states:
  286. /// - `.clientIdleServerIdle`
  287. /// - `.clientClosedServerClosed`
  288. /// Doing so will result in a `.invalidState` error.
  289. ///
  290. /// - Parameter trailers: The trailers to parse.
  291. mutating func receiveEndOfResponseStream(
  292. _ trailers: HPACKHeaders
  293. ) -> Result<GRPCStatus, ReceiveEndOfResponseStreamError> {
  294. return self.state.receiveEndOfResponseStream(trailers)
  295. }
  296. }
  297. extension GRPCClientStateMachine.State {
  298. /// See `GRPCClientStateMachine.sendRequestHeaders(requestHead:)`.
  299. mutating func sendRequestHeaders(
  300. requestHead: _GRPCRequestHead
  301. ) -> Result<HPACKHeaders, SendRequestHeadersError> {
  302. let result: Result<HPACKHeaders, SendRequestHeadersError>
  303. switch self {
  304. case let .clientIdleServerIdle(pendingWriteState, responseArity):
  305. let headers = self.makeRequestHeaders(
  306. method: requestHead.method,
  307. scheme: requestHead.scheme,
  308. host: requestHead.host,
  309. path: requestHead.path,
  310. timeout: GRPCTimeout(deadline: requestHead.deadline),
  311. customMetadata: requestHead.customMetadata,
  312. compression: requestHead.encoding
  313. )
  314. result = .success(headers)
  315. self = .clientActiveServerIdle(
  316. writeState: pendingWriteState.makeWriteState(messageEncoding: requestHead.encoding),
  317. pendingReadState: .init(arity: responseArity, messageEncoding: requestHead.encoding)
  318. )
  319. case .clientActiveServerIdle,
  320. .clientClosedServerIdle,
  321. .clientClosedServerActive,
  322. .clientActiveServerActive,
  323. .clientClosedServerClosed:
  324. result = .failure(.invalidState)
  325. }
  326. return result
  327. }
  328. /// See `GRPCClientStateMachine.sendRequest(_:allocator:)`.
  329. mutating func sendRequest(
  330. _ message: Request,
  331. compressed: Bool,
  332. allocator: ByteBufferAllocator
  333. ) -> Result<ByteBuffer, MessageWriteError> {
  334. let result: Result<ByteBuffer, MessageWriteError>
  335. switch self {
  336. case .clientActiveServerIdle(var writeState, let pendingReadState):
  337. result = writeState.write(message, compressed: compressed, allocator: allocator)
  338. self = .clientActiveServerIdle(writeState: writeState, pendingReadState: pendingReadState)
  339. case .clientActiveServerActive(var writeState, let readState):
  340. result = writeState.write(message, compressed: compressed, allocator: allocator)
  341. self = .clientActiveServerActive(writeState: writeState, readState: readState)
  342. case .clientClosedServerIdle,
  343. .clientClosedServerActive,
  344. .clientClosedServerClosed:
  345. result = .failure(.cardinalityViolation)
  346. case .clientIdleServerIdle:
  347. result = .failure(.invalidState)
  348. }
  349. return result
  350. }
  351. /// See `GRPCClientStateMachine.sendEndOfRequestStream()`.
  352. mutating func sendEndOfRequestStream() -> Result<Void, SendEndOfRequestStreamError> {
  353. let result: Result<Void, SendEndOfRequestStreamError>
  354. switch self {
  355. case .clientActiveServerIdle(_, let pendingReadState):
  356. result = .success(())
  357. self = .clientClosedServerIdle(pendingReadState: pendingReadState)
  358. case .clientActiveServerActive(_, let readState):
  359. result = .success(())
  360. self = .clientClosedServerActive(readState: readState)
  361. case .clientClosedServerIdle,
  362. .clientClosedServerActive,
  363. .clientClosedServerClosed:
  364. result = .failure(.alreadyClosed)
  365. case .clientIdleServerIdle:
  366. result = .failure(.invalidState)
  367. }
  368. return result
  369. }
  370. /// See `GRPCClientStateMachine.receiveResponseHeaders(_:)`.
  371. mutating func receiveResponseHeaders(
  372. _ headers: HPACKHeaders
  373. ) -> Result<Void, ReceiveResponseHeadError> {
  374. let result: Result<Void, ReceiveResponseHeadError>
  375. switch self {
  376. case let .clientActiveServerIdle(writeState, pendingReadState):
  377. result = self.parseResponseHeaders(headers, pendingReadState: pendingReadState).map { readState in
  378. self = .clientActiveServerActive(writeState: writeState, readState: readState)
  379. }
  380. case let .clientClosedServerIdle(pendingReadState):
  381. result = self.parseResponseHeaders(headers, pendingReadState: pendingReadState).map { readState in
  382. self = .clientClosedServerActive(readState: readState)
  383. }
  384. case .clientIdleServerIdle,
  385. .clientClosedServerActive,
  386. .clientActiveServerActive,
  387. .clientClosedServerClosed:
  388. result = .failure(.invalidState)
  389. }
  390. return result
  391. }
  392. /// See `GRPCClientStateMachine.receiveResponseBuffer(_:)`.
  393. mutating func receiveResponseBuffer(
  394. _ buffer: inout ByteBuffer
  395. ) -> Result<[Response], MessageReadError> {
  396. let result: Result<[Response], MessageReadError>
  397. switch self {
  398. case .clientClosedServerActive(var readState):
  399. result = readState.readMessages(&buffer)
  400. self = .clientClosedServerActive(readState: readState)
  401. case .clientActiveServerActive(let writeState, var readState):
  402. result = readState.readMessages(&buffer)
  403. self = .clientActiveServerActive(writeState: writeState, readState: readState)
  404. case .clientIdleServerIdle,
  405. .clientActiveServerIdle,
  406. .clientClosedServerIdle,
  407. .clientClosedServerClosed:
  408. result = .failure(.invalidState)
  409. }
  410. return result
  411. }
  412. /// See `GRPCClientStateMachine.receiveEndOfResponseStream(_:)`.
  413. mutating func receiveEndOfResponseStream(
  414. _ trailers: HPACKHeaders
  415. ) -> Result<GRPCStatus, ReceiveEndOfResponseStreamError> {
  416. let result: Result<GRPCStatus, ReceiveEndOfResponseStreamError>
  417. switch self {
  418. case .clientActiveServerIdle,
  419. .clientClosedServerIdle:
  420. result = self.parseTrailersOnly(trailers).map { status in
  421. self = .clientClosedServerClosed
  422. return status
  423. }
  424. case .clientActiveServerActive,
  425. .clientClosedServerActive:
  426. result = .success(self.parseTrailers(trailers))
  427. self = .clientClosedServerClosed
  428. case .clientIdleServerIdle,
  429. .clientClosedServerClosed:
  430. result = .failure(.invalidState)
  431. }
  432. return result
  433. }
  434. /// Makes the request headers (`Request-Headers` in the specification) used to initiate an RPC
  435. /// call.
  436. ///
  437. /// See: https://github.com/grpc/grpc/blob/master/doc/PROTOCOL-HTTP2.md#requests
  438. ///
  439. /// - Parameter host: The host serving the RPC.
  440. /// - Parameter options: Any options related to the call.
  441. /// - Parameter requestID: A request ID associated with the call. An additional header will be
  442. /// added using this value if `options.requestIDHeader` is specified.
  443. private func makeRequestHeaders(
  444. method: String,
  445. scheme: String,
  446. host: String,
  447. path: String,
  448. timeout: GRPCTimeout,
  449. customMetadata: HPACKHeaders,
  450. compression: ClientMessageEncoding
  451. ) -> HPACKHeaders {
  452. // Note: we don't currently set the 'grpc-encoding' header, if we do we will need to feed that
  453. // encoded into the message writer.
  454. var headers: HPACKHeaders = [
  455. ":method": method,
  456. ":path": path,
  457. ":authority": host,
  458. ":scheme": scheme,
  459. "content-type": "application/grpc",
  460. "te": "trailers", // Used to detect incompatible proxies, part of the gRPC specification.
  461. ]
  462. switch compression {
  463. case .enabled(let configuration):
  464. // Request encoding.
  465. if let outbound = configuration.outbound {
  466. headers.add(name: GRPCHeaderName.encoding, value: outbound.name)
  467. }
  468. // Response encoding.
  469. if !configuration.inbound.isEmpty {
  470. headers.add(name: GRPCHeaderName.acceptEncoding, value: configuration.acceptEncodingHeader)
  471. }
  472. case .disabled:
  473. ()
  474. }
  475. // Add the timeout header, if a timeout was specified.
  476. if timeout != .infinite {
  477. headers.add(name: GRPCHeaderName.timeout, value: String(describing: timeout))
  478. }
  479. // Add user-defined custom metadata: this should come after the call definition headers.
  480. // TODO: make header normalization user-configurable.
  481. headers.add(contentsOf: customMetadata.map { (name, value, indexing) in
  482. return (name.lowercased(), value, indexing)
  483. })
  484. // Add default user-agent value, if `customMetadata` didn't contain user-agent
  485. if headers["user-agent"].isEmpty {
  486. headers.add(name: "user-agent", value: "grpc-swift-nio") // TODO: Add a more specific user-agent.
  487. }
  488. return headers
  489. }
  490. /// Parses the response headers ("Response-Headers" in the specification) from the server into
  491. /// a `ReadState`.
  492. ///
  493. /// See: https://github.com/grpc/grpc/blob/master/doc/PROTOCOL-HTTP2.md#responses
  494. ///
  495. /// - Parameter headers: The headers to parse.
  496. private func parseResponseHeaders(
  497. _ headers: HPACKHeaders,
  498. pendingReadState: PendingReadState
  499. ) -> Result<ReadState, ReceiveResponseHeadError> {
  500. // From: https://github.com/grpc/grpc/blob/master/doc/PROTOCOL-HTTP2.md#responses
  501. //
  502. // "Implementations should expect broken deployments to send non-200 HTTP status codes in
  503. // responses as well as a variety of non-GRPC content-types and to omit Status & Status-Message.
  504. // Implementations must synthesize a Status & Status-Message to propagate to the application
  505. // layer when this occurs."
  506. let statusHeader = headers.first(name: ":status")
  507. let responseStatus = statusHeader
  508. .flatMap(Int.init)
  509. .map { code in
  510. HTTPResponseStatus(statusCode: code)
  511. } ?? .preconditionFailed
  512. guard responseStatus == .ok else {
  513. return .failure(.invalidHTTPStatus(statusHeader))
  514. }
  515. let contentTypeHeader = headers.first(name: "content-type")
  516. guard contentTypeHeader.flatMap(ContentType.init) != nil else {
  517. return .failure(.invalidContentType(contentTypeHeader))
  518. }
  519. let result: Result<ReadState, ReceiveResponseHeadError>
  520. // What compression mechanism is the server using, if any?
  521. if let encodingHeader = headers.first(name: GRPCHeaderName.encoding) {
  522. // Note: the server is allowed to encode messages using an algorithm which wasn't included in
  523. // the 'grpc-accept-encoding' header. If the client still supports that algorithm (despite not
  524. // permitting the server to use it) then it must still decode that message. Ideally we should
  525. // log a message here if that was the case but we don't hold that information.
  526. if let compression = CompressionAlgorithm(rawValue: encodingHeader) {
  527. result = .success(pendingReadState.makeReadState(compression: compression))
  528. } else {
  529. // The algorithm isn't one we support.
  530. result = .failure(.unsupportedMessageEncoding(encodingHeader))
  531. }
  532. } else {
  533. // No compression was specified, this is fine.
  534. result = .success(pendingReadState.makeReadState(compression: nil))
  535. }
  536. return result
  537. }
  538. /// Parses the response trailers ("Trailers" in the specification) from the server into
  539. /// a `GRPCStatus`.
  540. ///
  541. /// See: https://github.com/grpc/grpc/blob/master/doc/PROTOCOL-HTTP2.md#responses
  542. ///
  543. /// - Parameter trailers: Trailers to parse.
  544. private func parseTrailers(_ trailers: HPACKHeaders) -> GRPCStatus {
  545. // Extract the "Status" and "Status-Message"
  546. let code = self.readStatusCode(from: trailers) ?? .unknown
  547. let message = self.readStatusMessage(from: trailers)
  548. return .init(code: code, message: message)
  549. }
  550. private func readStatusCode(from trailers: HPACKHeaders) -> GRPCStatus.Code? {
  551. return trailers.first(name: GRPCHeaderName.statusCode)
  552. .flatMap(Int.init)
  553. .flatMap(GRPCStatus.Code.init)
  554. }
  555. private func readStatusMessage(from trailers: HPACKHeaders) -> String? {
  556. return trailers.first(name: GRPCHeaderName.statusMessage)
  557. .map(GRPCStatusMessageMarshaller.unmarshall)
  558. }
  559. /// Parses a "Trailers-Only" response from the server into a `GRPCStatus`.
  560. ///
  561. /// See: https://github.com/grpc/grpc/blob/master/doc/PROTOCOL-HTTP2.md#responses
  562. ///
  563. /// - Parameter trailers: Trailers to parse.
  564. private func parseTrailersOnly(
  565. _ trailers: HPACKHeaders
  566. ) -> Result<GRPCStatus, ReceiveEndOfResponseStreamError> {
  567. // We need to check whether we have a valid HTTP status in the headers, if we don't then we also
  568. // need to check whether we have a gRPC status as it should take preference over a synthesising
  569. // one from the ":status".
  570. //
  571. // See: https://github.com/grpc/grpc/blob/master/doc/http-grpc-status-mapping.md
  572. let statusHeader = trailers.first(name: ":status")
  573. guard let status = statusHeader.flatMap(Int.init).map({ HTTPResponseStatus(statusCode: $0) }) else {
  574. return .failure(.invalidHTTPStatus(statusHeader))
  575. }
  576. guard status == .ok else {
  577. if let code = self.readStatusCode(from: trailers) {
  578. let message = self.readStatusMessage(from: trailers)
  579. return .failure(.invalidHTTPStatusWithGRPCStatus(.init(code: code, message: message)))
  580. } else {
  581. return .failure(.invalidHTTPStatus(statusHeader))
  582. }
  583. }
  584. let contentTypeHeader = trailers.first(name: "content-type")
  585. guard contentTypeHeader.map(ContentType.init) != nil else {
  586. return .failure(.invalidContentType(contentTypeHeader))
  587. }
  588. // We've verified the status and content type are okay: parse the trailers.
  589. return .success(self.parseTrailers(trailers))
  590. }
  591. }