1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
/*
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://aws.amazon.com/apache2.0
*
* or in the "license" file accompanying this file. This file is distributed
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
#include "tls/s2n_async_offload.h"
#include "api/s2n.h"
#include "crypto/s2n_hash.h"
#include "crypto/s2n_signature.h"
#include "error/s2n_errno.h"
#include "tls/s2n_connection.h"
#include "tls/s2n_handshake.h"
#include "utils/s2n_blob.h"
#include "utils/s2n_mem.h"
#include "utils/s2n_result.h"
#include "utils/s2n_safety.h"
S2N_RESULT s2n_async_offload_cb_invoke(struct s2n_connection *conn, struct s2n_async_offload_op *op)
{
RESULT_ENSURE_REF(conn);
RESULT_ENSURE_REF(conn->config);
RESULT_ENSURE_REF(conn->config->async_offload_cb);
RESULT_ENSURE_REF(op);
RESULT_ENSURE_REF(op->perform);
RESULT_ENSURE_REF(op->op_data_free);
RESULT_ENSURE(op->async_state == S2N_ASYNC_NOT_INVOKED, S2N_ERR_ASYNC_MORE_THAN_ONE);
op->async_state = S2N_ASYNC_INVOKED;
RESULT_ENSURE(conn->config->async_offload_cb(conn, op, conn->config->async_offload_ctx) == S2N_SUCCESS,
S2N_ERR_CANCELLED);
/*
* If the callback already completed the operation, continue.
* Otherwise, we need to block s2n_negotiate and wait for the operation to complete.
*/
if (op->async_state == S2N_ASYNC_COMPLETE) {
return S2N_RESULT_OK;
}
RESULT_BAIL(S2N_ERR_ASYNC_BLOCKED);
}
int s2n_async_offload_op_perform(struct s2n_async_offload_op *op)
{
POSIX_ENSURE_REF(op);
POSIX_ENSURE(op->async_state == S2N_ASYNC_INVOKED, S2N_ERR_INVALID_STATE);
POSIX_ENSURE(op->type != 0, S2N_ERR_INVALID_STATE);
POSIX_ENSURE_REF(op->conn);
POSIX_ENSURE_REF(op->perform);
POSIX_GUARD_RESULT(op->perform(op));
op->async_state = S2N_ASYNC_COMPLETE;
return S2N_SUCCESS;
}
S2N_RESULT s2n_async_offload_op_wipe(struct s2n_async_offload_op *op)
{
RESULT_ENSURE_REF(op);
if (op->op_data_free == NULL) {
return S2N_RESULT_OK;
}
RESULT_GUARD(op->op_data_free(op));
RESULT_CHECKED_MEMSET(op, 0, sizeof(struct s2n_async_offload_op));
return S2N_RESULT_OK;
}
/**
* MUST be called at the end of each handshake state handler that may invoke async_offload_cb
* to clean up the op object for its next use.
*/
S2N_RESULT s2n_async_offload_op_reset(struct s2n_async_offload_op *op)
{
RESULT_ENSURE_REF(op);
/* Sync case without the callback: async_offload_cb not invoked in the current state */
if (op->async_state == S2N_ASYNC_NOT_INVOKED) {
return S2N_RESULT_OK;
}
RESULT_ENSURE(op->async_state == S2N_ASYNC_COMPLETE, S2N_ERR_INVALID_STATE);
RESULT_GUARD(s2n_async_offload_op_wipe(op));
return S2N_RESULT_OK;
}
bool s2n_async_offload_op_is_in_allow_list(struct s2n_config *config, s2n_async_offload_op_type op_type)
{
return config && (config->async_offload_allow_list & op_type);
}
|
