mqtt/src/mqtt_subs.c - diff

Return to mqtt_subs.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / mqtt / src

Diff for /mqtt/src/mqtt_subs.c between versions 1.2.2.3 and 1.2.2.12

version 1.2.2.3, 2012/04/27 10:07:51	version 1.2.2.12, 2012/06/19 08:32:17
Line 8 io_enableDEBUG;	Line 8 io_enableDEBUG;

extern char compiled[], compiledby[], compilehost[];	extern char compiled[], compiledby[], compilehost[];
volatile intptr_t Kill;	volatile intptr_t Kill;
	sched_root_task_t *root;

struct tagArgs *args;	struct tagArgs *args;

Line 19 Usage(void)	Line 20 Usage(void)
"=== %s@%s === Compiled: %s ===\n\n"	"=== %s@%s === Compiled: %s ===\n\n"
" Syntax: mqtt_subs [options] <connect_to_broker[:port]> <ConnectID> [exec_script <value>]\n\n"	" Syntax: mqtt_subs [options] <connect_to_broker[:port]> <ConnectID> [exec_script <value>]\n\n"
"\t-l <value2file>\t\tSave received values to file\n"	"\t-l <value2file>\t\tSave received values to file\n"
	"\t-u\t\t\tUnsubscribe given topic(s)\n"
"\t-s <topic[\|QoS]>\tSubscribe for this topic, if wish add different \|QoS to topic\n"	"\t-s <topic[\|QoS]>\tSubscribe for this topic, if wish add different \|QoS to topic\n"
"\t-q <QoS>\t\tQoS level (0-at most 1, 1-at least 1, 2-exactly 1)\n"
"\t-d\t\t\tSend duplicate message\n\n"	"\t-d\t\t\tSend duplicate message\n\n"
"\t-C\t\t\tNot clear before connect!\n"	"\t-C\t\t\tNot clear before connect!\n"
"\t-p <port>\t\tDifferent port for connect (default: 1883)\n"	"\t-p <port>\t\tDifferent port for connect (default: 1883)\n"
Line 52 cleanArgs(struct tagArgs * __restrict args)	Line 53 cleanArgs(struct tagArgs * __restrict args)
static int	static int
Subscribe(int sock, FILE *lf)	Subscribe(int sock, FILE *lf)
{	{
int siz;	u_char qoses, qos;
u_char *qoses;
u_short mid;	u_short mid;
	mqtt_subscr_t *sub;

siz = mqtt_msgSUBSCRIBE(args->msg, args->subscr, AIT_GET_U16(&args->ConnID), args->Dup, args->QoS);	#ifdef __NetBSD__
if (siz == -1) {	srandom(getpid() ^ time(NULL));
printf("Error:: in msgSUBSCRIBE #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());	#else
	srandomdev();
	#endif
	mid = random() % USHRT_MAX;

	printf(" > Execute SUBSCRIBE request #%d ... ", mid);
	qoses = mqtt_cli_Subscribe(args->cli, args->subscr, mid, args->Dup, MQTT_QOS_ACK);
	if (!qoses) {
	printf("Error:: Subscribe #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
return -1;	return -1;
}	} else
siz = SendTo(sock, siz);	printf("OK\n");
if (siz == -1)
return -1;

siz = RecvFrom(sock);	for (sub = args->subscr, qos = qoses; sub->sub_topic.msg_base; sub++, qos++)
if (siz == -1)	printf(" + Topic %s with QoS %d subscribed %s\n", (char*)
	sub->sub_topic.msg_base, sub->sub_ret, *qos ? "done" : "failed");

	free(qoses);
	return 0;
	}

	static int
	Unsubscribe(int sock)
	{
	u_short mid;

	#ifdef __NetBSD__
	srandom(getpid() ^ time(NULL));
	#else
	srandomdev();
	#endif
	mid = random() % USHRT_MAX;

	printf(" > Execute UNSUBSCRIBE request #%d ... ", mid);
	if (mqtt_cli_Unsubscribe(args->cli, args->subscr, mid, args->Dup, MQTT_QOS_ACK)) {
	printf("Error:: Unsubscribe #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
return -1;	return -1;
siz = mqtt_readSUBACK(args->msg, &mid, &qoses);	} else
if (siz == -1) {	printf("OK\n");
printf("Error:: in readSUBACK #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
return -1;
}
if (mid != AIT_GET_U16(&args->ConnID)) {
printf("Error:: received different connection ID %d != %d\n",
mid, AIT_GET_U16(&args->ConnID));
return -1;
}

return 0;	return 0;
}	}


	static void *
	pubRX(sched_task_t *task)
	{
	int siz, rlen;
	char szTopic[BUFSIZ] = { 0 };
	void *data = NULL;
	u_short mid;

	rlen = RecvFrom(TASK_FD(task));
	if (siz == -1)
	goto end;
	siz = mqtt_readPUBLISH(args->msg->msg_base, szTopic, sizeof szTopic, &mid, &data);
	if (siz == -1)
	goto end;

	fprintf(TASK_ARG(task), "\nMessage ID: 0x%04hu\n", mid);

	if (data) {
	fputs((const char*) data, TASK_ARG(task));
	free(data);
	}
	end:
	schedReadSelf(task);
	return NULL;
	}


int	int
main(int argc, char **argv)	main(int argc, char **argv)
{	{
char ch, idx = 0, batch = 1;	char ch, un = 0, idx = 0, batch = 1;
ait_val_t val;	ait_val_t val;
u_short port = atoi(MQTT_PORT);	u_short port = atoi(MQTT_PORT);
mqtt_subscr_t *sub;	mqtt_subscr_t *sub;
int sock, ret = 0;	int ret = 0;
char *str, szStr[STRSIZ], szLogName[MAXPATHLEN] = { 0 };	char *str, szStr[STRSIZ], szLogName[MAXPATHLEN] = { 0 };
FILE *lf;	FILE *lf;

if (!(args = malloc(sizeof(struct tagArgs)))) {	if (!(args = io_malloc(sizeof(struct tagArgs)))) {
printf("Error:: in arguments #%d - %s\n", errno, strerror(errno));	printf("Error:: in arguments #%d - %s\n", errno, strerror(errno));
return 1;	return 1;
} else	} else
memset(args, 0, sizeof(struct tagArgs));	memset(args, 0, sizeof(struct tagArgs));
if (!(args->subscr = mqtt_subAlloc(idx))) {	if (!(args->subscr = mqtt_subAlloc(idx))) {
printf("Error:: in subscribes array #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());	printf("Error:: in subscribes array #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
free(args);	io_free(args);
return 1;	return 1;
} else	} else
args->free = cleanArgs;	args->free = cleanArgs;
Line 109 main(int argc, char **argv)	Line 156 main(int argc, char **argv)
if (!(args->msg = mqtt_msgAlloc(USHRT_MAX))) {	if (!(args->msg = mqtt_msgAlloc(USHRT_MAX))) {
printf("Error:: in mqtt buffer #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());	printf("Error:: in mqtt buffer #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
args->free(args);	args->free(args);
free(args);	io_free(args);
return 1;	return 1;
}	}

	AIT_SET_STR(&args->ConnID, "");
AIT_SET_STR(&args->User, "");	AIT_SET_STR(&args->User, "");
AIT_SET_STR(&args->Pass, "");	AIT_SET_STR(&args->Pass, "");

args->ka = MQTT_KEEPALIVE;	args->ka = MQTT_KEEPALIVE;
while ((ch = getopt(argc, argv, "T:U:P:p:s:q:dl:W:M:CDvh")) != -1)	while ((ch = getopt(argc, argv, "T:U:P:p:s:q:dl:W:M:CDvuh")) != -1)
switch (ch) {	switch (ch) {
case 'T':	case 'T':
args->ka = (u_short) strtol(optarg, NULL, 0);	args->ka = (u_short) strtol(optarg, NULL, 0);
Line 146 main(int argc, char **argv)	Line 194 main(int argc, char **argv)
if (!sub) {	if (!sub) {
printf("Error:: #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());	printf("Error:: #%d - %s\n", mqtt_GetErrno(), mqtt_GetError());
args->free(args);	args->free(args);
free(args);	io_free(args);
return 1;	return 1;
} else	} else
sub += idx++;	sub += idx++;
Line 169 main(int argc, char **argv)	Line 217 main(int argc, char **argv)
if (args->QoS > MQTT_QOS_EXACTLY) {	if (args->QoS > MQTT_QOS_EXACTLY) {
printf("Error:: invalid QoS level %d\n", args->QoS);	printf("Error:: invalid QoS level %d\n", args->QoS);
args->free(args);	args->free(args);
free(args);	io_free(args);
return 1;	return 1;
}	}
break;	break;
Line 188 main(int argc, char **argv)	Line 236 main(int argc, char **argv)
case 'v':	case 'v':
io_incDebug;	io_incDebug;
break;	break;
	case 'u':
	un = 1;
	break;
case 'h':	case 'h':
default:	default:
args->free(args);	args->free(args);
free(args);	io_free(args);
Usage();	Usage();
return 1;	return 1;
}	}
Line 200 main(int argc, char **argv)	Line 251 main(int argc, char **argv)
if (argc < 2) {	if (argc < 2) {
printf("Error:: host for connect not found, connection id or topic not supplied!\n\n");	printf("Error:: host for connect not found, connection id or topic not supplied!\n\n");
args->free(args);	args->free(args);
free(args);	io_free(args);
Usage();	Usage();
return 1;	return 1;
} else	} else {
AIT_SET_U16(&args->ConnID, strtol(argv[1], NULL, 0));	AIT_FREE_VAL(&args->ConnID);
	AIT_SET_STR(&args->ConnID, argv[1]);
	}
if (argc > 2) {	if (argc > 2) {
AIT_FREE_VAL(&args->Value);	AIT_FREE_VAL(&args->Value);
AIT_SET_STR(&args->Value, argv[2]);	AIT_SET_STR(&args->Value, argv[2]);
Line 212 main(int argc, char **argv)	Line 265 main(int argc, char **argv)
if (!io_gethostbyname(*argv, port, &args->addr)) {	if (!io_gethostbyname(*argv, port, &args->addr)) {
printf("Error:: host not valid #%d - %s\n", io_GetErrno(), io_GetError());	printf("Error:: host not valid #%d - %s\n", io_GetErrno(), io_GetError());
args->free(args);	args->free(args);
free(args);	io_free(args);
Usage();	Usage();
return 1;	return 1;
}	}
printf("Connecting to %s:%d ... ", io_n2addr(&args->addr, &val), io_n2port(&args->addr));	printf("Connecting to %s:%d ... ", io_n2addr(&args->addr, &val), io_n2port(&args->addr));
AIT_FREE_VAL(&val);	AIT_FREE_VAL(&val);

if ((sock = InitClient()) == -1) {	if (!(args->cli = mqtt_cli_Open(&args->addr.sa, args->ka))) {
args->free(args);	args->free(args);
free(args);	io_free(args);
return 2;	return 2;
}	}

switch ((ret = ConnectClient(sock))) {	switch ((ret = ConnectClient(args->cli->sock))) {
case -1:	case -1:
printf(">> FAILED!\n");	printf(">> FAILED!\n");
break;	break;
Line 255 main(int argc, char **argv)	Line 308 main(int argc, char **argv)
else	else
lf = stdout;	lf = stdout;
if (lf) {	if (lf) {
ret = Subscribe(sock, lf);	root = schedBegin();

	ret = Subscribe(args->cli->sock, lf);

	schedRead(root, pubRX, lf, args->cli->sock, NULL, 0);
	schedRun(root, &Kill);

	if (un)
	Unsubscribe(args->cli->sock);
fclose(lf);	fclose(lf);

	schedEnd(&root);
} else	} else
printf("Error:: in subscribe file #%d - %s\n", errno, strerror(errno));	printf("Error:: in subscribe file #%d - %s\n", errno, strerror(errno));
CloseClient(sock);	} else
} else {
close(sock);
ret = 3;	ret = 3;
}

	mqtt_cli_Close(&args->cli);

args->free(args);	args->free(args);
free(args);	io_free(args);
return ret;	return ret;
}	}

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>

Removed from v.1.2.2.3
changed lines
	Added in v.1.2.2.12