12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004 |
- #include <netlink-local.h>
- #include <netlink/netlink.h>
- #include <netlink/utils.h>
- #include <netlink/cache.h>
- #include <netlink/attr.h>
- #include <linux/socket.h>
- static size_t default_msg_size;
- static void __init init_msg_size(void)
- {
- default_msg_size = getpagesize();
- }
- int nlmsg_msg_size(int payload)
- {
- return NLMSG_HDRLEN + payload;
- }
- int nlmsg_total_size(int payload)
- {
- return NLMSG_ALIGN(nlmsg_msg_size(payload));
- }
- int nlmsg_padlen(int payload)
- {
- return nlmsg_total_size(payload) - nlmsg_msg_size(payload);
- }
- void *nlmsg_data(const struct nlmsghdr *nlh)
- {
- return (unsigned char *) nlh + NLMSG_HDRLEN;
- }
- void *nlmsg_tail(const struct nlmsghdr *nlh)
- {
- return (unsigned char *) nlh + NLMSG_ALIGN(nlh->nlmsg_len);
- }
- int nlmsg_len(const struct nlmsghdr *nlh)
- {
- return nlh->nlmsg_len - NLMSG_HDRLEN;
- }
- struct nlattr *nlmsg_attrdata(const struct nlmsghdr *nlh, int hdrlen)
- {
- unsigned char *data = nlmsg_data(nlh);
- return (struct nlattr *) (data + NLMSG_ALIGN(hdrlen));
- }
- int nlmsg_attrlen(const struct nlmsghdr *nlh, int hdrlen)
- {
- return nlmsg_len(nlh) - NLMSG_ALIGN(hdrlen);
- }
- int nlmsg_valid_hdr(const struct nlmsghdr *nlh, int hdrlen)
- {
- if (nlh->nlmsg_len < nlmsg_msg_size(hdrlen))
- return 0;
- return 1;
- }
- int nlmsg_ok(const struct nlmsghdr *nlh, int remaining)
- {
- return (remaining >= (int)sizeof(struct nlmsghdr) &&
- nlh->nlmsg_len >= sizeof(struct nlmsghdr) &&
- nlh->nlmsg_len <= remaining);
- }
- struct nlmsghdr *nlmsg_next(struct nlmsghdr *nlh, int *remaining)
- {
- int totlen = NLMSG_ALIGN(nlh->nlmsg_len);
- *remaining -= totlen;
- return (struct nlmsghdr *) ((unsigned char *) nlh + totlen);
- }
- int nlmsg_parse(struct nlmsghdr *nlh, int hdrlen, struct nlattr *tb[],
- int maxtype, struct nla_policy *policy)
- {
- if (!nlmsg_valid_hdr(nlh, hdrlen))
- return nl_errno(EINVAL);
- return nla_parse(tb, maxtype, nlmsg_attrdata(nlh, hdrlen),
- nlmsg_attrlen(nlh, hdrlen), policy);
- }
- struct nlattr *nlmsg_find_attr(struct nlmsghdr *nlh, int hdrlen, int attrtype)
- {
- return nla_find(nlmsg_attrdata(nlh, hdrlen),
- nlmsg_attrlen(nlh, hdrlen), attrtype);
- }
- int nlmsg_validate(struct nlmsghdr *nlh, int hdrlen, int maxtype,
- struct nla_policy *policy)
- {
- if (!nlmsg_valid_hdr(nlh, hdrlen))
- return nl_errno(EINVAL);
- return nla_validate(nlmsg_attrdata(nlh, hdrlen),
- nlmsg_attrlen(nlh, hdrlen), maxtype, policy);
- }
- static struct nl_msg *__nlmsg_alloc(size_t len)
- {
- struct nl_msg *nm;
- nm = calloc(1, sizeof(*nm));
- if (!nm)
- goto errout;
- nm->nm_nlh = calloc(1, len);
- if (!nm->nm_nlh)
- goto errout;
- nm->nm_protocol = -1;
- nm->nm_size = len;
- nm->nm_nlh->nlmsg_len = nlmsg_total_size(0);
- NL_DBG(2, "msg %p: Allocated new message, maxlen=%zu\n", nm, len);
- return nm;
- errout:
- free(nm);
- nl_errno(ENOMEM);
- return NULL;
- }
- struct nl_msg *nlmsg_alloc(void)
- {
- return __nlmsg_alloc(default_msg_size);
- }
- struct nl_msg *nlmsg_alloc_size(size_t max)
- {
- return __nlmsg_alloc(max);
- }
-
- struct nl_msg *nlmsg_inherit(struct nlmsghdr *hdr)
- {
- struct nl_msg *nm;
- nm = nlmsg_alloc();
- if (nm && hdr) {
- struct nlmsghdr *new = nm->nm_nlh;
- new->nlmsg_type = hdr->nlmsg_type;
- new->nlmsg_flags = hdr->nlmsg_flags;
- new->nlmsg_seq = hdr->nlmsg_seq;
- new->nlmsg_pid = hdr->nlmsg_pid;
- }
- return nm;
- }
- struct nl_msg *nlmsg_alloc_simple(int nlmsgtype, int flags)
- {
- struct nl_msg *msg;
- struct nlmsghdr nlh = {
- .nlmsg_type = nlmsgtype,
- .nlmsg_flags = flags,
- };
- msg = nlmsg_inherit(&nlh);
- if (msg)
- NL_DBG(2, "msg %p: Allocated new simple message\n", msg);
- return msg;
- }
- void nlmsg_set_default_size(size_t max)
- {
- if (max < nlmsg_total_size(0))
- max = nlmsg_total_size(0);
- default_msg_size = max;
- }
- struct nl_msg *nlmsg_convert(struct nlmsghdr *hdr)
- {
- struct nl_msg *nm;
- nm = __nlmsg_alloc(NLMSG_ALIGN(hdr->nlmsg_len));
- if (!nm)
- goto errout;
- memcpy(nm->nm_nlh, hdr, hdr->nlmsg_len);
- return nm;
- errout:
- nlmsg_free(nm);
- return NULL;
- }
- void *nlmsg_reserve(struct nl_msg *n, size_t len, int pad)
- {
- void *buf = n->nm_nlh;
- size_t nlmsg_len = n->nm_nlh->nlmsg_len;
- size_t tlen;
- tlen = pad ? ((len + (pad - 1)) & ~(pad - 1)) : len;
- if ((tlen + nlmsg_len) > n->nm_size) {
- nl_errno(ENOBUFS);
- return NULL;
- }
- buf += nlmsg_len;
- n->nm_nlh->nlmsg_len += tlen;
- if (tlen > len)
- memset(buf + len, 0, tlen - len);
- NL_DBG(2, "msg %p: Reserved %zu bytes, pad=%d, nlmsg_len=%d\n",
- n, len, pad, n->nm_nlh->nlmsg_len);
- return buf;
- }
- int nlmsg_append(struct nl_msg *n, void *data, size_t len, int pad)
- {
- void *tmp;
- tmp = nlmsg_reserve(n, len, pad);
- if (tmp == NULL)
- return nl_errno(ENOMEM);
- memcpy(tmp, data, len);
- NL_DBG(2, "msg %p: Appended %zu bytes with padding %d\n", n, len, pad);
- return 0;
- }
- int nlmsg_expand(struct nl_msg *n, size_t newlen)
- {
- void *tmp;
- if (newlen <= n->nm_size)
- return nl_errno(EINVAL);
- tmp = realloc(n->nm_nlh, newlen);
- if (tmp == NULL)
- return nl_errno(ENOMEM);
- n->nm_nlh = tmp;
- n->nm_size = newlen;
- return 0;
- }
- struct nlmsghdr *nlmsg_put(struct nl_msg *n, uint32_t pid, uint32_t seq,
- int type, int payload, int flags)
- {
- struct nlmsghdr *nlh;
- if (n->nm_nlh->nlmsg_len < NLMSG_HDRLEN)
- BUG();
- nlh = (struct nlmsghdr *) n->nm_nlh;
- nlh->nlmsg_type = type;
- nlh->nlmsg_flags = flags;
- nlh->nlmsg_pid = pid;
- nlh->nlmsg_seq = seq;
- NL_DBG(2, "msg %p: Added netlink header type=%d, flags=%d, pid=%d, "
- "seq=%d\n", n, type, flags, pid, seq);
- if (payload > 0 &&
- nlmsg_reserve(n, payload, NLMSG_ALIGNTO) == NULL)
- return NULL;
- return nlh;
- }
- struct nlmsghdr *nlmsg_hdr(struct nl_msg *n)
- {
- return n->nm_nlh;
- }
- void nlmsg_free(struct nl_msg *n)
- {
- if (!n)
- return;
- free(n->nm_nlh);
- free(n);
- NL_DBG(2, "msg %p: Freed\n", n);
- }
- void nlmsg_set_proto(struct nl_msg *msg, int protocol)
- {
- msg->nm_protocol = protocol;
- }
- int nlmsg_get_proto(struct nl_msg *msg)
- {
- return msg->nm_protocol;
- }
- size_t nlmsg_get_max_size(struct nl_msg *msg)
- {
- return msg->nm_size;
- }
- void nlmsg_set_src(struct nl_msg *msg, struct sockaddr_nl *addr)
- {
- memcpy(&msg->nm_src, addr, sizeof(*addr));
- }
- struct sockaddr_nl *nlmsg_get_src(struct nl_msg *msg)
- {
- return &msg->nm_src;
- }
- void nlmsg_set_dst(struct nl_msg *msg, struct sockaddr_nl *addr)
- {
- memcpy(&msg->nm_dst, addr, sizeof(*addr));
- }
- struct sockaddr_nl *nlmsg_get_dst(struct nl_msg *msg)
- {
- return &msg->nm_dst;
- }
- void nlmsg_set_creds(struct nl_msg *msg, struct ucred *creds)
- {
- memcpy(&msg->nm_creds, creds, sizeof(*creds));
- msg->nm_flags |= NL_MSG_CRED_PRESENT;
- }
- struct ucred *nlmsg_get_creds(struct nl_msg *msg)
- {
- if (msg->nm_flags & NL_MSG_CRED_PRESENT)
- return &msg->nm_creds;
- return NULL;
- }
- static struct trans_tbl nl_msgtypes[] = {
- __ADD(NLMSG_NOOP,NOOP)
- __ADD(NLMSG_ERROR,ERROR)
- __ADD(NLMSG_DONE,DONE)
- __ADD(NLMSG_OVERRUN,OVERRUN)
- };
- char *nl_nlmsgtype2str(int type, char *buf, size_t size)
- {
- return __type2str(type, buf, size, nl_msgtypes,
- ARRAY_SIZE(nl_msgtypes));
- }
- int nl_str2nlmsgtype(const char *name)
- {
- return __str2type(name, nl_msgtypes, ARRAY_SIZE(nl_msgtypes));
- }
- char *nl_nlmsg_flags2str(int flags, char *buf, size_t len)
- {
- memset(buf, 0, len);
- #define PRINT_FLAG(f) \
- if (flags & NLM_F_##f) { \
- flags &= ~NLM_F_##f; \
- strncat(buf, #f, len - strlen(buf) - 1); \
- if (flags) \
- strncat(buf, ",", len - strlen(buf) - 1); \
- }
-
- PRINT_FLAG(REQUEST);
- PRINT_FLAG(MULTI);
- PRINT_FLAG(ACK);
- PRINT_FLAG(ECHO);
- PRINT_FLAG(ROOT);
- PRINT_FLAG(MATCH);
- PRINT_FLAG(ATOMIC);
- PRINT_FLAG(REPLACE);
- PRINT_FLAG(EXCL);
- PRINT_FLAG(CREATE);
- PRINT_FLAG(APPEND);
- if (flags) {
- char s[32];
- snprintf(s, sizeof(s), "0x%x", flags);
- strncat(buf, s, len - strlen(buf) - 1);
- }
- #undef PRINT_FLAG
- return buf;
- }
- struct dp_xdata {
- void (*cb)(struct nl_object *, void *);
- void *arg;
- };
- static int parse_cb(struct nl_object *obj, struct nl_parser_param *p)
- {
- struct dp_xdata *x = p->pp_arg;
- x->cb(obj, x->arg);
- return 0;
- }
- int nl_msg_parse(struct nl_msg *msg, void (*cb)(struct nl_object *, void *),
- void *arg)
- {
- struct nl_cache_ops *ops;
- struct nl_parser_param p = {
- .pp_cb = parse_cb
- };
- struct dp_xdata x = {
- .cb = cb,
- .arg = arg,
- };
- ops = nl_cache_ops_associate(nlmsg_get_proto(msg),
- nlmsg_hdr(msg)->nlmsg_type);
- if (ops == NULL)
- return nl_error(ENOENT, "Unknown message type %d",
- nlmsg_hdr(msg)->nlmsg_type);
- p.pp_arg = &x;
- return nl_cache_parse(ops, NULL, nlmsg_hdr(msg), &p);
- }
- static void prefix_line(FILE *ofd, int prefix)
- {
- int i;
- for (i = 0; i < prefix; i++)
- fprintf(ofd, " ");
- }
- static inline void dump_hex(FILE *ofd, char *start, int len, int prefix)
- {
- int i, a, c, limit;
- char ascii[21] = {0};
- limit = 18 - (prefix * 2);
- prefix_line(ofd, prefix);
- fprintf(ofd, " ");
- for (i = 0, a = 0, c = 0; i < len; i++) {
- int v = *(uint8_t *) (start + i);
- fprintf(ofd, "%02x ", v);
- ascii[a++] = isprint(v) ? v : '.';
- if (c == limit-1) {
- fprintf(ofd, "%s\n", ascii);
- if (i < (len - 1)) {
- prefix_line(ofd, prefix);
- fprintf(ofd, " ");
- }
- a = c = 0;
- memset(ascii, 0, sizeof(ascii));
- } else
- c++;
- }
- if (c != 0) {
- for (i = 0; i < (limit - c); i++)
- fprintf(ofd, " ");
- fprintf(ofd, "%s\n", ascii);
- }
- }
- static void print_hdr(FILE *ofd, struct nl_msg *msg)
- {
- struct nlmsghdr *nlh = nlmsg_hdr(msg);
- struct nl_cache_ops *ops;
- struct nl_msgtype *mt;
- char buf[128];
- fprintf(ofd, " .nlmsg_len = %d\n", nlh->nlmsg_len);
- ops = nl_cache_ops_associate(nlmsg_get_proto(msg), nlh->nlmsg_type);
- if (ops) {
- mt = nl_msgtype_lookup(ops, nlh->nlmsg_type);
- if (!mt)
- BUG();
- snprintf(buf, sizeof(buf), "%s::%s", ops->co_name, mt->mt_name);
- } else
- nl_nlmsgtype2str(nlh->nlmsg_type, buf, sizeof(buf));
- fprintf(ofd, " .nlmsg_type = %d <%s>\n", nlh->nlmsg_type, buf);
- fprintf(ofd, " .nlmsg_flags = %d <%s>\n", nlh->nlmsg_flags,
- nl_nlmsg_flags2str(nlh->nlmsg_flags, buf, sizeof(buf)));
- fprintf(ofd, " .nlmsg_seq = %d\n", nlh->nlmsg_seq);
- fprintf(ofd, " .nlmsg_pid = %d\n", nlh->nlmsg_pid);
- }
- static void dump_attrs(FILE *ofd, struct nlattr *attrs, int attrlen,
- int prefix)
- {
- int rem;
- struct nlattr *nla;
- nla_for_each_attr(nla, attrs, attrlen, rem) {
- int padlen, alen = nla_len(nla);
- prefix_line(ofd, prefix);
- fprintf(ofd, " [ATTR %02d%s] %d octets\n", nla_type(nla),
- nla->nla_type & NLA_F_NESTED ? " NESTED" : "",
- alen);
- if (nla->nla_type & NLA_F_NESTED)
- dump_attrs(ofd, nla_data(nla), alen, prefix+1);
- else
- dump_hex(ofd, nla_data(nla), alen, prefix);
- padlen = nla_padlen(alen);
- if (padlen > 0) {
- prefix_line(ofd, prefix);
- fprintf(ofd, " [PADDING] %d octets\n",
- padlen);
- dump_hex(ofd, nla_data(nla) + alen,
- padlen, prefix);
- }
- }
- if (rem) {
- prefix_line(ofd, prefix);
- fprintf(ofd, " [LEFTOVER] %d octets\n", rem);
- }
- }
- void nl_msg_dump(struct nl_msg *msg, FILE *ofd)
- {
- struct nlmsghdr *hdr = nlmsg_hdr(msg);
-
- fprintf(ofd,
- "-------------------------- BEGIN NETLINK MESSAGE "
- "---------------------------\n");
- fprintf(ofd, " [HEADER] %Zu octets\n", sizeof(struct nlmsghdr));
- print_hdr(ofd, msg);
- if (hdr->nlmsg_type == NLMSG_ERROR &&
- hdr->nlmsg_len >= nlmsg_msg_size(sizeof(struct nlmsgerr))) {
- struct nl_msg *errmsg;
- struct nlmsgerr *err = nlmsg_data(hdr);
- char buf[256];
- fprintf(ofd, " [ERRORMSG] %Zu octets\n", sizeof(*err));
- fprintf(ofd, " .error = %d \"%s\"\n", err->error,
- strerror_r(-err->error, buf, sizeof(buf)));
- fprintf(ofd, " [ORIGINAL MESSAGE] %Zu octets\n", sizeof(*hdr));
- errmsg = nlmsg_inherit(&err->msg);
- print_hdr(ofd, errmsg);
- nlmsg_free(errmsg);
- } else if (nlmsg_len(hdr) > 0) {
- struct nl_cache_ops *ops;
- int payloadlen = nlmsg_len(hdr);
- int attrlen = 0;
- ops = nl_cache_ops_associate(nlmsg_get_proto(msg),
- hdr->nlmsg_type);
- if (ops) {
- attrlen = nlmsg_attrlen(hdr, ops->co_hdrsize);
- payloadlen -= attrlen;
- }
- fprintf(ofd, " [PAYLOAD] %d octets\n", payloadlen);
- dump_hex(ofd, nlmsg_data(hdr), payloadlen, 0);
- if (attrlen) {
- struct nlattr *attrs;
- int attrlen;
-
- attrs = nlmsg_attrdata(hdr, ops->co_hdrsize);
- attrlen = nlmsg_attrlen(hdr, ops->co_hdrsize);
- dump_attrs(ofd, attrs, attrlen, 0);
- }
- }
- fprintf(ofd,
- "--------------------------- END NETLINK MESSAGE "
- "---------------------------\n");
- }
|