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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
#ifndef MQTT_MOCK_SERVER_HANDLER_H
#define MQTT_MOCK_SERVER_HANDLER_H
/**
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
* SPDX-License-Identifier: Apache-2.0.
*/
#include <aws/mqtt/private/packets.h>
struct aws_channel_handler;
struct aws_channel_slot;
static const int MOCK_LOG_SUBJECT = 60000;
struct mqtt_decoded_packet {
enum aws_mqtt_packet_type type;
struct aws_allocator *alloc;
/* CONNECT */
bool clean_session;
bool has_will;
bool will_retain;
bool has_password;
bool has_username;
uint16_t keep_alive_timeout;
enum aws_mqtt_qos will_qos;
struct aws_byte_cursor client_identifier; /* These cursors live with the received_message */
struct aws_byte_cursor will_topic;
struct aws_byte_cursor will_message;
struct aws_byte_cursor username;
struct aws_byte_cursor password;
/* PUBLISH SUBSCRIBE UNSUBSCRIBE */
uint16_t packet_identifier;
struct aws_byte_cursor topic_name; /* PUBLISH topic */
struct aws_byte_cursor publish_payload; /* PUBLISH payload */
struct aws_array_list sub_topic_filters; /* list of aws_mqtt_subscription for SUBSCRIBE */
struct aws_array_list unsub_topic_filters; /* list of aws_byte_cursor for UNSUBSCRIBE */
bool duplicate; /* PUBLISH only */
/* index of the received packet, indicating when it's received by the server */
size_t index;
};
struct aws_channel_handler *new_mqtt_mock_server(struct aws_allocator *allocator);
void destroy_mqtt_mock_server(struct aws_channel_handler *handler);
void mqtt_mock_server_handler_update_slot(struct aws_channel_handler *handler, struct aws_channel_slot *slot);
/**
* Mock server sends a publish packet back to client
*/
int mqtt_mock_server_send_publish(
struct aws_channel_handler *handler,
struct aws_byte_cursor *topic,
struct aws_byte_cursor *payload,
bool dup,
enum aws_mqtt_qos qos,
bool retain);
/**
* Mock server sends a publish packet back to client with user-controlled packet id
*/
int mqtt_mock_server_send_publish_by_id(
struct aws_channel_handler *handler,
uint16_t packet_id,
struct aws_byte_cursor *topic,
struct aws_byte_cursor *payload,
bool dup,
enum aws_mqtt_qos qos,
bool retain);
/**
* Set max number of PINGRESP that mock server will send back to client
*/
void mqtt_mock_server_set_max_ping_resp(struct aws_channel_handler *handler, size_t max_ping);
/**
* Set max number of CONACK that mock server will send back to client
*/
void mqtt_mock_server_set_max_connack(struct aws_channel_handler *handler, size_t connack_avail);
/**
* Disable the automatically response (suback/unsuback/puback) to the client
*/
void mqtt_mock_server_disable_auto_ack(struct aws_channel_handler *handler);
/**
* Enable the automatically response (suback/unsuback/puback) to the client
*/
void mqtt_mock_server_enable_auto_ack(struct aws_channel_handler *handler);
/**
* Send response back the client given the packet ID
*/
int mqtt_mock_server_send_unsuback(struct aws_channel_handler *handler, uint16_t packet_id);
int mqtt_mock_server_send_puback(struct aws_channel_handler *handler, uint16_t packet_id);
int mqtt_mock_server_send_single_suback(
struct aws_channel_handler *handler,
uint16_t packet_id,
enum aws_mqtt_qos return_code);
/**
* Wait for puback_count PUBACK packages from client
*/
void mqtt_mock_server_wait_for_pubacks(struct aws_channel_handler *handler, size_t puback_count);
/**
* Getters for decoded packets, call mqtt_mock_server_decode_packets first.
*/
size_t mqtt_mock_server_decoded_packets_count(struct aws_channel_handler *handler);
/**
* Get the decoded packet by index
*/
struct mqtt_decoded_packet *mqtt_mock_server_get_decoded_packet_by_index(struct aws_channel_handler *handler, size_t i);
/**
* Get the latest received packet by index
*/
struct mqtt_decoded_packet *mqtt_mock_server_get_latest_decoded_packet(struct aws_channel_handler *handler);
/**
* Get the decoded packet by packet_id started from search_start_idx (included), Note: it may have multiple packets with
* the same ID, this will return the earliest received on with the packet_id. If out_idx is not NULL, the index of found
* packet will be stored at there, and if failed to find the packet, it will be set to SIZE_MAX, and the return value
* will be NULL.
*/
struct mqtt_decoded_packet *mqtt_mock_server_find_decoded_packet_by_id(
struct aws_channel_handler *handler,
size_t search_start_idx,
uint16_t packet_id,
size_t *out_idx);
/**
* Get the decoded packet by type started from search_start_idx (included), Note: it may have multiple packets with
* the same type, this will return the earliest received on with the packet_id. If out_idx is not NULL, the index of
* found packet will be stored at there, and if failed to find the packet, it will be set to SIZE_MAX, and the return
* value will be NULL.
*/
struct mqtt_decoded_packet *mqtt_mock_server_find_decoded_packet_by_type(
struct aws_channel_handler *handler,
size_t search_start_idx,
enum aws_mqtt_packet_type type,
size_t *out_idx);
/**
* Run all received messages through, and decode the messages.
*/
int mqtt_mock_server_decode_packets(struct aws_channel_handler *handler);
/**
* Returns the number of PINGs the server has gotten
*/
size_t mqtt_mock_server_get_ping_count(struct aws_channel_handler *handler);
#endif /* MQTT_MOCK_SERVER_HANDLER_H */
|
