mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-15 01:10:52 +08:00
166 lines
3.4 KiB
C++
166 lines
3.4 KiB
C++
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <getopt.h>
|
|
#include "lib_acl.h"
|
|
#include "acl_cpp/lib_acl.hpp"
|
|
|
|
static bool __payload_binary = false;
|
|
|
|
static bool handle_connack(const acl::mqtt_message& message) {
|
|
const acl::mqtt_connack& connack = (const acl::mqtt_connack&) message;
|
|
printf("%s: connect code=%d\r\n", __FUNCTION__, connack.get_connack_code());
|
|
|
|
return true;
|
|
}
|
|
|
|
static bool handle_puback(acl::mqtt_client&, const acl::mqtt_message&) {
|
|
return true;
|
|
}
|
|
|
|
static bool test_publish(acl::mqtt_client& conn, unsigned short id, int length) {
|
|
acl::mqtt_publish publish;
|
|
|
|
publish.get_header().set_qos(acl::MQTT_QOS1);
|
|
publish.set_pkt_id(id);
|
|
|
|
const char* topic = "test/topic";
|
|
publish.set_topic(topic);
|
|
|
|
acl::string payload;
|
|
payload.format("payload-%d:", id);
|
|
|
|
char ch;
|
|
if (__payload_binary) {
|
|
ch = 0;
|
|
} else {
|
|
ch = 'c';
|
|
}
|
|
|
|
for (; payload.size() < (size_t) length;) {
|
|
payload.append(&ch, 1);
|
|
}
|
|
|
|
publish.set_payload((unsigned) payload.size(), payload);
|
|
|
|
if (!conn.send(publish)) {
|
|
printf("send publish error\r\n");
|
|
return false;
|
|
}
|
|
|
|
ACL_VSTREAM* vs = conn.sock_stream()->get_vstream();
|
|
assert(vs);
|
|
|
|
printf("payload->size=%zd, total sent size=%lld\n",
|
|
payload.size(), vs->total_write_cnt);
|
|
|
|
if (publish.get_header().get_qos() == acl::MQTT_QOS0) {
|
|
return true;
|
|
}
|
|
|
|
acl::mqtt_message* res = conn.get_message();
|
|
if (res == NULL) {
|
|
printf("get puback error\r\n");
|
|
return false;
|
|
}
|
|
|
|
if (res->get_header().get_type() != acl::MQTT_PUBACK) {
|
|
printf("not puback type, type=%d\r\n",
|
|
(int) res->get_header().get_type());
|
|
delete res;
|
|
return false;
|
|
}
|
|
|
|
if (!handle_puback(conn, *res)) {
|
|
printf("handle_puback error\r\n");
|
|
delete res;
|
|
return false;
|
|
}
|
|
|
|
acl::mqtt_suback* suback = (acl::mqtt_suback*) res;
|
|
printf("%s => puback ok, pkt id=%d\r\n",
|
|
__FUNCTION__, suback->get_pkt_id());
|
|
|
|
delete res;
|
|
return true;
|
|
}
|
|
|
|
static void usage(const char* procname) {
|
|
printf("usage: %s -h [help] -s addr -n max -c payload_length -B [if payload is binary data]\r\n", procname);
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
char ch;
|
|
int max = 1, length = 32;
|
|
acl::string addr("127.0.0.1|1883");
|
|
|
|
while ((ch = getopt(argc, argv, "hs:n:c:B")) > 0) {
|
|
switch (ch) {
|
|
case 'h':
|
|
usage(argv[0]);
|
|
return 0;
|
|
case 's':
|
|
addr = optarg;
|
|
break;
|
|
case 'n':
|
|
max = atoi(optarg);
|
|
break;
|
|
case 'c':
|
|
length = atoi(optarg);
|
|
break;
|
|
case 'B':
|
|
__payload_binary = true;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
acl::log::stdout_open(true);
|
|
|
|
acl::mqtt_connect message;
|
|
message.set_cid("client-id-zsx-pub");
|
|
message.set_username("user-zsx");
|
|
//message.set_passwd("pass");
|
|
#if 0
|
|
message.set_will_qos(acl::MQTT_QOS0);
|
|
message.set_will_topic("test/topic");
|
|
message.set_will_msg("msg-hello");
|
|
#endif
|
|
|
|
printf("-----------------------------------------------\r\n");
|
|
acl::mqtt_client conn(addr, 10, 0);
|
|
if (!conn.send(message)) {
|
|
printf("send message error\r\n");
|
|
return 1;
|
|
}
|
|
printf("send connect message ok\r\n");
|
|
|
|
acl::mqtt_message* res = conn.get_message();
|
|
if (res == NULL) {
|
|
printf("read CONNACK error\r\n");
|
|
return 1;
|
|
}
|
|
acl::mqtt_type_t type = res->get_header().get_type();
|
|
if (type != acl::MQTT_CONNACK) {
|
|
printf("invalid message type=%d\r\n", (int) type);
|
|
delete res;
|
|
return 1;
|
|
}
|
|
if (!handle_connack(*res)) {
|
|
delete res;
|
|
return 1;
|
|
}
|
|
|
|
unsigned short id = 1;
|
|
for (int i = 1; i <= max; i++) {
|
|
if (!test_publish(conn, id++, length)) {
|
|
break;
|
|
}
|
|
// id must be more than 0
|
|
if (id == 0) {
|
|
id = 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|