tests-mx32/nlattr_smc_diag_msg.c - platform/external/strace - Git at Google

 /*
  * Copyright (c) 2017 JingPiao Chen <[email protected]>
  * Copyright (c) 2017-2018 The strace developers.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
  * 3. The name of the author may not be used to endorse or promote products
  *    derived from this software without specific prior written permission.
  *
  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */

 #include "tests.h"
 #include <sys/socket.h>

 #ifndef AF_SMC
 # define AF_SMC 43
 #endif

 #include <stdio.h>
 #include <string.h>
 #include <stdint.h>
 #include <arpa/inet.h>
 #include "test_nlattr.h"
 #include <linux/rtnetlink.h>
 #include <linux/smc_diag.h>
 #include <linux/sock_diag.h>

 #ifndef SMC_CLNT
 # define SMC_CLNT 0
 #endif
 #ifndef SMC_ACTIVE
 # define SMC_ACTIVE 1
 #endif

 static const char address[] = "12.34.56.78";

 static void
 init_smc_diag_msg(struct nlmsghdr *const nlh, const unsigned int msg_len)
 {
 	SET_STRUCT(struct nlmsghdr, nlh,
 		.nlmsg_len = msg_len,
 		.nlmsg_type = SOCK_DIAG_BY_FAMILY,
 		.nlmsg_flags = NLM_F_DUMP
 	);

 	struct smc_diag_msg *const msg = NLMSG_DATA(nlh);
 	SET_STRUCT(struct smc_diag_msg, msg,
 		.diag_family = AF_SMC,
 		.diag_state = SMC_ACTIVE
 	);

 	if (!inet_pton(AF_INET, address, msg->id.idiag_src) ||
 	    !inet_pton(AF_INET, address, msg->id.idiag_dst))
 		perror_msg_and_skip("inet_pton");
 }

 static void
 print_smc_diag_msg(const unsigned int msg_len)
 {
 	printf("{len=%u, type=SOCK_DIAG_BY_FAMILY"
 	       ", flags=NLM_F_DUMP, seq=0, pid=0}"
 	       ", {diag_family=AF_SMC, diag_state=SMC_ACTIVE"
 	       ", diag_fallback=SMC_DIAG_MODE_SMCR, diag_shutdown=0"
 	       ", id={idiag_sport=htons(0), idiag_dport=htons(0)"
 	       ", idiag_src=inet_addr(\"%s\")"
 	       ", idiag_dst=inet_addr(\"%s\")"
 	       ", idiag_if=0, idiag_cookie=[0, 0]}"
 	       ", diag_uid=0, diag_inode=0}",
 	       msg_len, address, address);
 }

 #define PRINT_FIELD_SMC_DIAG_CURSOR(prefix_, where_, field_)		\
 	do {								\
 		printf("%s%s=", (prefix_), #field_);			\
 		PRINT_FIELD_U("{", (where_).field_, reserved);		\
 		PRINT_FIELD_U(", ", (where_).field_, wrap);		\
 		PRINT_FIELD_U(", ", (where_).field_, count);		\
 		printf("}");						\
 	} while (0)

 int main(void)
 {
 	skip_if_unavailable("/proc/self/fd/");

 	static const struct smc_diag_conninfo cinfo = {
 		.token = 0xabcdefac,
 		.sndbuf_size = 0xbcdaefad,
 		.rmbe_size = 0xcdbaefab,
 		.peer_rmbe_size = 0xdbcdedaf,
 		.rx_prod = {
 			.reserved = 0xabc1,
 			.wrap = 0xbca1,
 			.count = 0xcdedbad1
 		},
 		.rx_cons = {
 			.reserved = 0xabc2,
 			.wrap = 0xbca2,
 			.count = 0xcdedbad2
 		},
 		.tx_prod = {
 			.reserved = 0xabc3,
 			.wrap = 0xbca3,
 			.count = 0xcdedbad3
 		},
 		.tx_cons = {
 			.reserved = 0xabc4,
 			.wrap = 0xbca4,
 			.count = 0xcdedbad4
 		},
 		.rx_prod_flags = 0xff,
 		.rx_conn_state_flags = 0xff,
 		.tx_prod_flags = 0xff,
 		.tx_conn_state_flags = 0xff,
 		.tx_prep = {
 			.reserved = 0xabc5,
 			.wrap = 0xbca5,
 			.count = 0xcdedbad5
 		},
 		.tx_sent = {
 			.reserved = 0xabc6,
 			.wrap = 0xbca6,
 			.count = 0xcdedbad6
 		},
 		.tx_fin = {
 			.reserved = 0xabc7,
 			.wrap = 0xbca7,
 			.count = 0xcdedbad7
 		}
 	};
 	static const struct smc_diag_lgrinfo linfo = {
 		.lnk[0] = {
 			.link_id = 0xaf,
 			.ibport = 0xfa,
 			.ibname = "123",
 			.gid = "456",
 			.peer_gid = "789"
 		},
 		.role = SMC_CLNT
 	};
 	static const struct smcd_diag_dmbinfo dinfo = {
 		.linkid     = 0xdeadc0de,
 		.peer_gid   = 0xbefeededbadc0dedULL,
 		.my_gid     = 0xdeec0dedfacebeefULL,
 		.token      = 0xcafedecaffeedeedULL,
 		.peer_token = 0xfeedfacebeeff00dULL,
 	};
 	static const struct smc_diag_fallback fb1 = {
 		.reason         = 0,
 		.peer_diagnosis = 0x03020000,
 	};
 	static const struct smc_diag_fallback fb2 = {
 		.reason         = 0x03060000,
 		.peer_diagnosis = 0x99999999,
 	};
 	static uint8_t sd1 = 0x23;
 	static uint8_t sd2 = 0x40;

 	int fd = create_nl_socket(NETLINK_SOCK_DIAG);
 	const unsigned int hdrlen = sizeof(struct smc_diag_msg);
 	void *const nlh0 = midtail_alloc(NLMSG_SPACE(hdrlen),
 					 NLA_HDRLEN +
 					 MAX(sizeof(cinfo), sizeof(linfo)));

 	static char pattern[4096];
 	fill_memory_ex(pattern, sizeof(pattern), 'a', 'z' - 'a' + 1);

 	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
 			   init_smc_diag_msg, print_smc_diag_msg,
 			   SMC_DIAG_SHUTDOWN, pattern, sd1,
 			   printf("RCV_SHUTDOWN|SEND_SHUTDOWN|0x20"));

 	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
 			   init_smc_diag_msg, print_smc_diag_msg,
 			   SMC_DIAG_SHUTDOWN, pattern, sd2,
 			   printf("0x40 /* ???_SHUTDOWN */"));

 	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
 			   init_smc_diag_msg, print_smc_diag_msg,
 			   SMC_DIAG_CONNINFO, pattern, cinfo,
 			   PRINT_FIELD_U("{", cinfo, token);
 			   PRINT_FIELD_U(", ", cinfo, sndbuf_size);
 			   PRINT_FIELD_U(", ", cinfo, rmbe_size);
 			   PRINT_FIELD_U(", ", cinfo, peer_rmbe_size);
 			   PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, rx_prod);
 			   PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, rx_cons);
 			   PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_prod);
 			   PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_cons);
 			   printf(", rx_prod_flags=0xff");
 			   printf(", rx_conn_state_flags=0xff");
 			   printf(", tx_prod_flags=0xff");
 			   printf(", tx_conn_state_flags=0xff");
 			   PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_prep);
 			   PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_sent);
 			   PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_fin);
 			   printf("}"));

 	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
 			   init_smc_diag_msg, print_smc_diag_msg,
 			   SMC_DIAG_LGRINFO, pattern, linfo,
 			   PRINT_FIELD_U("{lnk[0]={", linfo.lnk[0], link_id);
 			   printf(", ibname=\"%s\"", linfo.lnk[0].ibname);
 			   PRINT_FIELD_U(", ", linfo.lnk[0], ibport);
 			   printf(", gid=\"%s\"", linfo.lnk[0].gid);
 			   printf(", peer_gid=\"%s\"}", linfo.lnk[0].peer_gid);
 			   printf(", role=SMC_CLNT}"));

 	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
 			   init_smc_diag_msg, print_smc_diag_msg,
 			   SMC_DIAG_DMBINFO, pattern, dinfo,
 			   PRINT_FIELD_U("{", dinfo, linkid);
 			   PRINT_FIELD_X(", ", dinfo, peer_gid);
 			   PRINT_FIELD_X(", ", dinfo, my_gid);
 			   PRINT_FIELD_X(", ", dinfo, token);
 			   PRINT_FIELD_X(", ", dinfo, peer_token);
 			   printf("}"));

 	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
 			   init_smc_diag_msg, print_smc_diag_msg,
 			   SMC_DIAG_FALLBACK, pattern, fb1,
 			   printf("{reason=0 /* SMC_CLC_DECL_??? */");
 			   printf(", peer_diagnosis=0x3020000"
 			          " /* SMC_CLC_DECL_IPSEC */}"));

 	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
 			   init_smc_diag_msg, print_smc_diag_msg,
 			   SMC_DIAG_FALLBACK, pattern, fb2,
 			   printf("{reason=0x3060000"
 			          " /* SMC_CLC_DECL_OPTUNSUPP */");
 			   printf(", peer_diagnosis=0x99999999"
 			          " /* SMC_CLC_DECL_??? */}"));

 	printf("+++ exited with 0 +++\n");
 	return 0;
 }
	/*
	* Copyright (c) 2017 JingPiao Chen <[email protected]>
	* Copyright (c) 2017-2018 The strace developers.
	* All rights reserved.
	*
	* Redistribution and use in source and binary forms, with or without
	* modification, are permitted provided that the following conditions
	* are met:
	* 1. Redistributions of source code must retain the above copyright
	* notice, this list of conditions and the following disclaimer.
	* 2. Redistributions in binary form must reproduce the above copyright
	* notice, this list of conditions and the following disclaimer in the
	* documentation and/or other materials provided with the distribution.
	* 3. The name of the author may not be used to endorse or promote products
	* derived from this software without specific prior written permission.
	*
	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	*/

	#include "tests.h"
	#include <sys/socket.h>

	#ifndef AF_SMC
	# define AF_SMC 43
	#endif

	#include <stdio.h>
	#include <string.h>
	#include <stdint.h>
	#include <arpa/inet.h>
	#include "test_nlattr.h"
	#include <linux/rtnetlink.h>
	#include <linux/smc_diag.h>
	#include <linux/sock_diag.h>

	#ifndef SMC_CLNT
	# define SMC_CLNT 0
	#endif
	#ifndef SMC_ACTIVE
	# define SMC_ACTIVE 1
	#endif

	static const char address[] = "12.34.56.78";

	static void
	init_smc_diag_msg(struct nlmsghdr *const nlh, const unsigned int msg_len)
	{
	SET_STRUCT(struct nlmsghdr, nlh,
	.nlmsg_len = msg_len,
	.nlmsg_type = SOCK_DIAG_BY_FAMILY,
	.nlmsg_flags = NLM_F_DUMP
	);

	struct smc_diag_msg *const msg = NLMSG_DATA(nlh);
	SET_STRUCT(struct smc_diag_msg, msg,
	.diag_family = AF_SMC,
	.diag_state = SMC_ACTIVE
	);

	if (!inet_pton(AF_INET, address, msg->id.idiag_src) \|\|
	!inet_pton(AF_INET, address, msg->id.idiag_dst))
	perror_msg_and_skip("inet_pton");
	}

	static void
	print_smc_diag_msg(const unsigned int msg_len)
	{
	printf("{len=%u, type=SOCK_DIAG_BY_FAMILY"
	", flags=NLM_F_DUMP, seq=0, pid=0}"
	", {diag_family=AF_SMC, diag_state=SMC_ACTIVE"
	", diag_fallback=SMC_DIAG_MODE_SMCR, diag_shutdown=0"
	", id={idiag_sport=htons(0), idiag_dport=htons(0)"
	", idiag_src=inet_addr(\"%s\")"
	", idiag_dst=inet_addr(\"%s\")"
	", idiag_if=0, idiag_cookie=[0, 0]}"
	", diag_uid=0, diag_inode=0}",
	msg_len, address, address);
	}

	#define PRINT_FIELD_SMC_DIAG_CURSOR(prefix_, where_, field_) \
	do { \
	printf("%s%s=", (prefix_), #field_); \
	PRINT_FIELD_U("{", (where_).field_, reserved); \
	PRINT_FIELD_U(", ", (where_).field_, wrap); \
	PRINT_FIELD_U(", ", (where_).field_, count); \
	printf("}"); \
	} while (0)

	int main(void)
	{
	skip_if_unavailable("/proc/self/fd/");

	static const struct smc_diag_conninfo cinfo = {
	.token = 0xabcdefac,
	.sndbuf_size = 0xbcdaefad,
	.rmbe_size = 0xcdbaefab,
	.peer_rmbe_size = 0xdbcdedaf,
	.rx_prod = {
	.reserved = 0xabc1,
	.wrap = 0xbca1,
	.count = 0xcdedbad1
	},
	.rx_cons = {
	.reserved = 0xabc2,
	.wrap = 0xbca2,
	.count = 0xcdedbad2
	},
	.tx_prod = {
	.reserved = 0xabc3,
	.wrap = 0xbca3,
	.count = 0xcdedbad3
	},
	.tx_cons = {
	.reserved = 0xabc4,
	.wrap = 0xbca4,
	.count = 0xcdedbad4
	},
	.rx_prod_flags = 0xff,
	.rx_conn_state_flags = 0xff,
	.tx_prod_flags = 0xff,
	.tx_conn_state_flags = 0xff,
	.tx_prep = {
	.reserved = 0xabc5,
	.wrap = 0xbca5,
	.count = 0xcdedbad5
	},
	.tx_sent = {
	.reserved = 0xabc6,
	.wrap = 0xbca6,
	.count = 0xcdedbad6
	},
	.tx_fin = {
	.reserved = 0xabc7,
	.wrap = 0xbca7,
	.count = 0xcdedbad7
	}
	};
	static const struct smc_diag_lgrinfo linfo = {
	.lnk[0] = {
	.link_id = 0xaf,
	.ibport = 0xfa,
	.ibname = "123",
	.gid = "456",
	.peer_gid = "789"
	},
	.role = SMC_CLNT
	};
	static const struct smcd_diag_dmbinfo dinfo = {
	.linkid = 0xdeadc0de,
	.peer_gid = 0xbefeededbadc0dedULL,
	.my_gid = 0xdeec0dedfacebeefULL,
	.token = 0xcafedecaffeedeedULL,
	.peer_token = 0xfeedfacebeeff00dULL,
	};
	static const struct smc_diag_fallback fb1 = {
	.reason = 0,
	.peer_diagnosis = 0x03020000,
	};
	static const struct smc_diag_fallback fb2 = {
	.reason = 0x03060000,
	.peer_diagnosis = 0x99999999,
	};
	static uint8_t sd1 = 0x23;
	static uint8_t sd2 = 0x40;

	int fd = create_nl_socket(NETLINK_SOCK_DIAG);
	const unsigned int hdrlen = sizeof(struct smc_diag_msg);
	void *const nlh0 = midtail_alloc(NLMSG_SPACE(hdrlen),
	NLA_HDRLEN +
	MAX(sizeof(cinfo), sizeof(linfo)));

	static char pattern[4096];
	fill_memory_ex(pattern, sizeof(pattern), 'a', 'z' - 'a' + 1);

	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
	init_smc_diag_msg, print_smc_diag_msg,
	SMC_DIAG_SHUTDOWN, pattern, sd1,
	printf("RCV_SHUTDOWN\|SEND_SHUTDOWN\|0x20"));

	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
	init_smc_diag_msg, print_smc_diag_msg,
	SMC_DIAG_SHUTDOWN, pattern, sd2,
	printf("0x40 /* ???_SHUTDOWN */"));

	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
	init_smc_diag_msg, print_smc_diag_msg,
	SMC_DIAG_CONNINFO, pattern, cinfo,
	PRINT_FIELD_U("{", cinfo, token);
	PRINT_FIELD_U(", ", cinfo, sndbuf_size);
	PRINT_FIELD_U(", ", cinfo, rmbe_size);
	PRINT_FIELD_U(", ", cinfo, peer_rmbe_size);
	PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, rx_prod);
	PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, rx_cons);
	PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_prod);
	PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_cons);
	printf(", rx_prod_flags=0xff");
	printf(", rx_conn_state_flags=0xff");
	printf(", tx_prod_flags=0xff");
	printf(", tx_conn_state_flags=0xff");
	PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_prep);
	PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_sent);
	PRINT_FIELD_SMC_DIAG_CURSOR(", ", cinfo, tx_fin);
	printf("}"));

	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
	init_smc_diag_msg, print_smc_diag_msg,
	SMC_DIAG_LGRINFO, pattern, linfo,
	PRINT_FIELD_U("{lnk[0]={", linfo.lnk[0], link_id);
	printf(", ibname=\"%s\"", linfo.lnk[0].ibname);
	PRINT_FIELD_U(", ", linfo.lnk[0], ibport);
	printf(", gid=\"%s\"", linfo.lnk[0].gid);
	printf(", peer_gid=\"%s\"}", linfo.lnk[0].peer_gid);
	printf(", role=SMC_CLNT}"));

	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
	init_smc_diag_msg, print_smc_diag_msg,
	SMC_DIAG_DMBINFO, pattern, dinfo,
	PRINT_FIELD_U("{", dinfo, linkid);
	PRINT_FIELD_X(", ", dinfo, peer_gid);
	PRINT_FIELD_X(", ", dinfo, my_gid);
	PRINT_FIELD_X(", ", dinfo, token);
	PRINT_FIELD_X(", ", dinfo, peer_token);
	printf("}"));

	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
	init_smc_diag_msg, print_smc_diag_msg,
	SMC_DIAG_FALLBACK, pattern, fb1,
	printf("{reason=0 /* SMC_CLC_DECL_??? */");
	printf(", peer_diagnosis=0x3020000"
	" /* SMC_CLC_DECL_IPSEC */}"));

	TEST_NLATTR_OBJECT(fd, nlh0, hdrlen,
	init_smc_diag_msg, print_smc_diag_msg,
	SMC_DIAG_FALLBACK, pattern, fb2,
	printf("{reason=0x3060000"
	" /* SMC_CLC_DECL_OPTUNSUPP */");
	printf(", peer_diagnosis=0x99999999"
	" /* SMC_CLC_DECL_??? */}"));

	printf("+++ exited with 0 +++\n");
	return 0;
	}