|
@@ -38,8 +38,6 @@
|
|
|
#include <linux/netlink.h>
|
|
|
#include <syslog.h>
|
|
|
|
|
|
-static char vss_recv_buffer[4096];
|
|
|
-static char vss_send_buffer[4096];
|
|
|
static struct sockaddr_nl addr;
|
|
|
|
|
|
#ifndef SOL_NETLINK
|
|
@@ -147,6 +145,9 @@ int main(void)
|
|
|
struct cn_msg *incoming_cn_msg;
|
|
|
int op;
|
|
|
struct hv_vss_msg *vss_msg;
|
|
|
+ char *vss_send_buffer;
|
|
|
+ char *vss_recv_buffer;
|
|
|
+ size_t vss_recv_buffer_len;
|
|
|
|
|
|
if (daemon(1, 0))
|
|
|
return 1;
|
|
@@ -154,6 +155,14 @@ int main(void)
|
|
|
openlog("Hyper-V VSS", 0, LOG_USER);
|
|
|
syslog(LOG_INFO, "VSS starting; pid is:%d", getpid());
|
|
|
|
|
|
+ vss_recv_buffer_len = NLMSG_HDRLEN + sizeof(struct cn_msg) + sizeof(struct hv_vss_msg);
|
|
|
+ vss_send_buffer = calloc(1, vss_recv_buffer_len);
|
|
|
+ vss_recv_buffer = calloc(1, vss_recv_buffer_len);
|
|
|
+ if (!(vss_send_buffer && vss_recv_buffer)) {
|
|
|
+ syslog(LOG_ERR, "Failed to allocate netlink buffers");
|
|
|
+ exit(EXIT_FAILURE);
|
|
|
+ }
|
|
|
+
|
|
|
fd = socket(AF_NETLINK, SOCK_DGRAM, NETLINK_CONNECTOR);
|
|
|
if (fd < 0) {
|
|
|
syslog(LOG_ERR, "netlink socket creation failed; error:%d %s",
|
|
@@ -215,7 +224,7 @@ int main(void)
|
|
|
continue;
|
|
|
}
|
|
|
|
|
|
- len = recvfrom(fd, vss_recv_buffer, sizeof(vss_recv_buffer), 0,
|
|
|
+ len = recvfrom(fd, vss_recv_buffer, vss_recv_buffer_len, 0,
|
|
|
addr_p, &addr_l);
|
|
|
|
|
|
if (len < 0) {
|