diff --git a/tools/include/xenstore.h b/tools/include/xenstore.h index 423422dc50..bf6d767a22 100644 --- a/tools/include/xenstore.h +++ b/tools/include/xenstore.h @@ -277,6 +277,43 @@ bool xs_get_features_domain(struct xs_handle *h, unsigned int domid, bool xs_set_features_domain(struct xs_handle *h, unsigned int domid, unsigned int features); +/* Get names of supported quota. + * Returns an array of quota names supported by Xenstore. On return + * num will contain the number of quota names in the array. + * Will return NULL on failure. + * The returned array should be freed via free() after usage, this will + * free the array AND the strings of the quota names. + */ +const char **xs_get_quota_names(struct xs_handle *h, unsigned int *num); + +/* Get the value of one global quota. + * Returns the global quota value specified by "quota" in "value". + * Return false on failure. + */ +bool xs_get_global_quota(struct xs_handle *h, const char *quota, + unsigned int *value); + +/* Set the value of one global quota. + * Sets the global quota value specified by "quota" to "value". + * Return false on failure. + */ +bool xs_set_global_quota(struct xs_handle *h, const char *quota, + unsigned int value); + +/* Get the value of one domain quota. + * Returns a domain's quota value specified by "quota" in "value". + * Return false on failure. + */ +bool xs_get_domain_quota(struct xs_handle *h, unsigned int domid, + const char *quota, unsigned int *value); + +/* Set the value of one domain quota. + * Sets a domain's quota value specified by "quota" to "value". + * Return false on failure. + */ +bool xs_set_domain_quota(struct xs_handle *h, unsigned int domid, + const char *quota, unsigned int value); + char *xs_control_command(struct xs_handle *h, const char *cmd, void *data, unsigned int len); /* Deprecated: use xs_control_command() instead. */ diff --git a/tools/libs/store/Makefile b/tools/libs/store/Makefile index fed43b0008..b52d1f35ad 100644 --- a/tools/libs/store/Makefile +++ b/tools/libs/store/Makefile @@ -2,7 +2,7 @@ XEN_ROOT=$(CURDIR)/../../.. include $(XEN_ROOT)/tools/Rules.mk MAJOR = 4 -MINOR = 1 +MINOR = 2 version-script := libxenstore.map ifeq ($(CONFIG_Linux),y) diff --git a/tools/libs/store/libxenstore.map b/tools/libs/store/libxenstore.map index cd9df86749..a08ddd549f 100644 --- a/tools/libs/store/libxenstore.map +++ b/tools/libs/store/libxenstore.map @@ -45,3 +45,11 @@ VERS_4.1 { xs_get_features_domain; xs_set_features_domain; } VERS_4.0; +VERS_4.2 { + global: + xs_get_quota_names; + xs_get_global_quota; + xs_set_global_quota; + xs_get_domain_quota; + xs_set_domain_quota; +} VERS_4.1; diff --git a/tools/libs/store/xs.c b/tools/libs/store/xs.c index 8f4b90a3cf..06462445e0 100644 --- a/tools/libs/store/xs.c +++ b/tools/libs/store/xs.c @@ -1456,6 +1456,117 @@ bool xs_set_features_domain(struct xs_handle *h, unsigned int domid, return xs_bool(xs_talkv(h, iov, ARRAY_SIZE(iov), NULL)); } +const char **xs_get_quota_names(struct xs_handle *h, unsigned int *num) +{ + struct xsd_sockmsg msg = { .type = XS_GET_QUOTA }; + struct iovec iov[1]; + const char **quota; + char *reply; + char *c; + unsigned int i; + + iov[0].iov_base = &msg; + iov[0].iov_len = sizeof(msg); + + reply = xs_talkv(h, iov, ARRAY_SIZE(iov), NULL); + if (!reply) + return NULL; + + *num = 1; + for (c = reply; *c; c++) + if (*c == ' ') + (*num)++; + + quota = malloc(*num * sizeof(char *) + strlen(reply) + 1); + c = (char *)(quota + *num); + strcpy(c, reply); + for (i = 0; i < *num; i++) { + quota[i] = c; + c = strchr(c, ' '); + if (c) { + *c = 0; + c++; + } + } + + return quota; +} + +bool xs_get_global_quota(struct xs_handle *h, const char *quota, + unsigned int *value) +{ + struct xsd_sockmsg msg = { .type = XS_GET_QUOTA }; + struct iovec iov[2]; + + iov[0].iov_base = &msg; + iov[0].iov_len = sizeof(msg); + iov[1].iov_base = (char *)quota; + iov[1].iov_len = strlen(quota) + 1; + + return xs_uint(xs_talkv(h, iov, ARRAY_SIZE(iov), NULL), value); +} + +bool xs_set_global_quota(struct xs_handle *h, const char *quota, + unsigned int value) +{ + struct xsd_sockmsg msg = { .type = XS_SET_QUOTA }; + char val_str[MAX_STRLEN(value)]; + struct iovec iov[3]; + + snprintf(val_str, sizeof(val_str), "%u", value); + + iov[0].iov_base = &msg; + iov[0].iov_len = sizeof(msg); + iov[1].iov_base = (char *)quota; + iov[1].iov_len = strlen(quota) + 1; + iov[2].iov_base = val_str; + iov[2].iov_len = strlen(val_str) + 1; + + return xs_bool(xs_talkv(h, iov, ARRAY_SIZE(iov), NULL)); +} + +bool xs_get_domain_quota(struct xs_handle *h, unsigned int domid, + const char *quota, unsigned int *value) +{ + struct xsd_sockmsg msg = { .type = XS_GET_QUOTA }; + char domid_str[MAX_STRLEN(domid)]; + struct iovec iov[3]; + + snprintf(domid_str, sizeof(domid_str), "%u", domid); + + iov[0].iov_base = &msg; + iov[0].iov_len = sizeof(msg); + iov[1].iov_base = domid_str; + iov[1].iov_len = strlen(domid_str) + 1; + iov[2].iov_base = (char *)quota; + iov[2].iov_len = strlen(quota) + 1; + + return xs_uint(xs_talkv(h, iov, ARRAY_SIZE(iov), NULL), value); +} + +bool xs_set_domain_quota(struct xs_handle *h, unsigned int domid, + const char *quota, unsigned int value) +{ + struct xsd_sockmsg msg = { .type = XS_SET_QUOTA }; + char domid_str[MAX_STRLEN(domid)]; + char val_str[MAX_STRLEN(value)]; + struct iovec iov[4]; + + snprintf(domid_str, sizeof(domid_str), "%u", domid); + snprintf(val_str, sizeof(val_str), "%u", value); + + iov[0].iov_base = &msg; + iov[0].iov_len = sizeof(msg); + iov[1].iov_base = domid_str; + iov[1].iov_len = strlen(domid_str) + 1; + iov[2].iov_base = (char *)quota; + iov[2].iov_len = strlen(quota) + 1; + iov[3].iov_base = val_str; + iov[3].iov_len = strlen(val_str) + 1; + + return xs_bool(xs_talkv(h, iov, ARRAY_SIZE(iov), NULL)); +} + char *xs_control_command(struct xs_handle *h, const char *cmd, void *data, unsigned int len) {