int ipvs_zero_service(ipvs_service_t *svc)
{
ipvs_func = ipvs_zero_service;
#ifdef LIBIPVS_USE_NL
if (try_nl) {
struct nl_msg *msg = ipvs_nl_message(IPVS_CMD_ZERO, 0);
if (!msg) return -1;
if (svc->fwmark
|| memcmp(&in6addr_any, &svc->addr.in6, sizeof(struct in6_addr))
|| svc->port) {
if (ipvs_nl_fill_service_attr(msg, svc)) {
nlmsg_free(msg);
return -1;
}
}
return ipvs_nl_send_message(msg, ipvs_nl_noop_cb, NULL);
}
#endif
CHECK_COMPAT_SVC(svc, -1);
return setsockopt(sockfd, IPPROTO_IP, IP_VS_SO_SET_ZERO, (char *)svc,
sizeof(struct ip_vs_service_kern));
}
ipvs_service_entry_t *
ipvs_get_service(u_int32_t fwmark, u_int16_t af, u_int16_t protocol, union nf_inet_addr addr,
u_int16_t port)
{
ipvs_service_entry_t *svc;
socklen_t len;
ipvs_func = ipvs_get_service;
#ifdef LIBIPVS_USE_NL
if (try_nl) {
struct ip_vs_get_services *get;
struct nl_msg *msg;
ipvs_service_t tsvc;
svc = malloc(sizeof(*svc));
if (!svc)
return NULL;
tsvc.fwmark = fwmark;
tsvc.af = af;
tsvc.protocol= protocol;
tsvc.addr = addr;
tsvc.port = port;
if (!(get = malloc(sizeof(*get) + sizeof(ipvs_service_entry_t))))
goto ipvs_get_service_err2;
get->num_services = 0;
msg = ipvs_nl_message(IPVS_CMD_GET_SERVICE, 0);
if (!msg) goto ipvs_get_service_err;
if (ipvs_nl_fill_service_attr(msg, &tsvc))
goto nla_put_failure;
if (ipvs_nl_send_message(msg, ipvs_services_parse_cb, &get))
goto ipvs_get_service_err;
memcpy(svc, &(get->entrytable[0]), sizeof(*svc));
free(get);
return svc;
nla_put_failure:
nlmsg_free(msg);
ipvs_get_service_err:
free(get);
ipvs_get_service_err2:
free(svc);
return NULL;
}
#endif
len = sizeof(*svc);
svc = calloc(1, len);
if (!svc)
return NULL;
svc->fwmark = fwmark;
svc->af = af;
svc->protocol = protocol;
svc->addr = addr;
svc->port = port;
CHECK_COMPAT_SVC(svc, NULL);
if (getsockopt(sockfd, IPPROTO_IP, IP_VS_SO_GET_SERVICE,
(char *)svc, &len)) {
free(svc);
return NULL;
}
svc->af = AF_INET;
svc->addr.ip = svc->__addr_v4;
svc->pe_name[0] = '\0';
return svc;
out_err:
free(svc);
return NULL;
}
struct ip_vs_get_dests *ipvs_get_dests(ipvs_service_entry_t *svc)
{
struct ip_vs_get_dests *d;
struct ip_vs_get_dests_kern *dk;
socklen_t len;
int i;
len = sizeof(*d) + sizeof(ipvs_dest_entry_t) * svc->num_dests;
if (!(d = malloc(len)))
return NULL;
ipvs_func = ipvs_get_dests;
#ifdef LIBIPVS_USE_NL
if (try_nl) {
struct nl_msg *msg;
struct nlattr *nl_service;
if (svc->num_dests == 0)
d = realloc(d,sizeof(*d) + sizeof(ipvs_dest_entry_t));
d->fwmark = svc->fwmark;
d->protocol = svc->protocol;
d->addr = svc->addr;
d->port = svc->port;
d->num_dests = svc->num_dests;
d->af = svc->af;
msg = ipvs_nl_message(IPVS_CMD_GET_DEST, NLM_F_DUMP);
if (!msg)
goto ipvs_nl_dest_failure;
nl_service = nla_nest_start(msg, IPVS_CMD_ATTR_SERVICE);
if (!nl_service)
goto nla_put_failure;
NLA_PUT_U16(msg, IPVS_SVC_ATTR_AF, svc->af);
if (svc->fwmark) {
NLA_PUT_U32(msg, IPVS_SVC_ATTR_FWMARK, svc->fwmark);
} else {
NLA_PUT_U16(msg, IPVS_SVC_ATTR_PROTOCOL, svc->protocol);
NLA_PUT(msg, IPVS_SVC_ATTR_ADDR, sizeof(svc->addr),
&svc->addr);
NLA_PUT_U16(msg, IPVS_SVC_ATTR_PORT, svc->port);
}
nla_nest_end(msg, nl_service);
if (ipvs_nl_send_message(msg, ipvs_dests_parse_cb, &d))
goto ipvs_nl_dest_failure;
return d;
nla_put_failure:
nlmsg_free(msg);
ipvs_nl_dest_failure:
free(d);
return NULL;
}
#endif
if (svc->af != AF_INET) {
errno = EAFNOSUPPORT;
free(d);
return NULL;
}
len = sizeof(*dk) + sizeof(struct ip_vs_dest_entry_kern) * svc->num_dests;
if (!(dk = malloc(len))) {
free(d);
return NULL;
}
dk->fwmark = svc->fwmark;
dk->protocol = svc->protocol;
dk->addr = svc->addr.ip;
dk->port = svc->port;
dk->num_dests = svc->num_dests;
if (getsockopt(sockfd, IPPROTO_IP,
IP_VS_SO_GET_DESTS, dk, &len) < 0) {
free(d);
free(dk);
return NULL;
}
memcpy(d, dk, sizeof(struct ip_vs_get_dests_kern));
d->af = AF_INET;
d->addr.ip = d->__addr_v4;
for (i = 0; i < dk->num_dests; i++) {
memcpy(&d->entrytable[i], &dk->entrytable[i],
sizeof(struct ip_vs_dest_entry_kern));
d->entrytable[i].af = AF_INET;
d->entrytable[i].addr.ip = d->entrytable[i].__addr_v4;
}
free(dk);
return d;
}
