X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=rquota%2Frquota.c;h=55919d54eb035215a8f7d5416cd17648097bc2e0;hb=eecdc4f3cd5c464ae02823a823993e720664e465;hp=4397ec82ed6eb73b840716ef5fc1be24e13d2df3;hpb=19e74f5a322739990fd47260f3b3347fbd49c302;p=deb_libnfs.git

diff --git a/rquota/rquota.c b/rquota/rquota.c
index 4397ec8..55919d5 100644
--- a/rquota/rquota.c
+++ b/rquota/rquota.c
@@ -15,6 +15,10 @@
    along with this program; if not, see <http://www.gnu.org/licenses/>.
 */
 
+#if defined(WIN32)
+#include <winsock2.h>
+#endif
+
 #include <stdio.h>
 #include <errno.h>
 #include <sys/stat.h>
@@ -54,12 +58,12 @@ int rpc_rquota1_null_async(struct rpc_context *rpc, rpc_cb cb, void *private_dat
 
 	pdu = rpc_allocate_pdu(rpc, RQUOTA_PROGRAM, RQUOTA_V1, RQUOTA1_NULL, cb, private_data, (xdrproc_t)xdr_void, 0);
 	if (pdu == NULL) {
-		rpc_set_error(rpc, "Out of memory. Failed to allocate pdu for rquota/null call");
+		rpc_set_error(rpc, "Out of memory. Failed to allocate pdu for rquota1/null call");
 		return -1;
 	}
 
 	if (rpc_queue_pdu(rpc, pdu) != 0) {
-		rpc_set_error(rpc, "Out of memory. Failed to queue pdu for rquota/null call");
+		rpc_set_error(rpc, "Out of memory. Failed to queue pdu for rquota1/null call");
 		rpc_free_pdu(rpc, pdu);
 		return -2;
 	}
@@ -74,7 +78,7 @@ int rpc_rquota1_getquota_async(struct rpc_context *rpc, rpc_cb cb, char *export,
 
 	pdu = rpc_allocate_pdu(rpc, RQUOTA_PROGRAM, RQUOTA_V1, RQUOTA1_GETQUOTA, cb, private_data, (xdrproc_t)xdr_GETQUOTA1res, sizeof(GETQUOTA1res));
 	if (pdu == NULL) {
-		rpc_set_error(rpc, "Out of memory. Failed to allocate pdu for rquota/getquota call");
+		rpc_set_error(rpc, "Out of memory. Failed to allocate pdu for rquota1/getquota call");
 		return -1;
 	}
 
@@ -88,7 +92,7 @@ int rpc_rquota1_getquota_async(struct rpc_context *rpc, rpc_cb cb, char *export,
 	}
 
 	if (rpc_queue_pdu(rpc, pdu) != 0) {
-		rpc_set_error(rpc, "Out of memory. Failed to queue pdu for rquota/getquota call");
+		rpc_set_error(rpc, "Out of memory. Failed to queue pdu for rquota1/getquota call");
 		rpc_free_pdu(rpc, pdu);
 		return -3;
 	}
@@ -103,7 +107,7 @@ int rpc_rquota1_getactivequota_async(struct rpc_context *rpc, rpc_cb cb, char *e
 
 	pdu = rpc_allocate_pdu(rpc, RQUOTA_PROGRAM, RQUOTA_V1, RQUOTA1_GETACTIVEQUOTA, cb, private_data, (xdrproc_t)xdr_GETQUOTA1res, sizeof(GETQUOTA1res));
 	if (pdu == NULL) {
-		rpc_set_error(rpc, "Out of memory. Failed to allocate pdu for rquota/getactivequota call");
+		rpc_set_error(rpc, "Out of memory. Failed to allocate pdu for rquota1/getactivequota call");
 		return -1;
 	}
 
@@ -117,7 +121,87 @@ int rpc_rquota1_getactivequota_async(struct rpc_context *rpc, rpc_cb cb, char *e
 	}
 
 	if (rpc_queue_pdu(rpc, pdu) != 0) {
-		rpc_set_error(rpc, "Out of memory. Failed to queue pdu for rquota/getactivequota call");
+		rpc_set_error(rpc, "Out of memory. Failed to queue pdu for rquota1/getactivequota call");
+		rpc_free_pdu(rpc, pdu);
+		return -3;
+	}
+
+	return 0;
+}
+
+
+int rpc_rquota2_null_async(struct rpc_context *rpc, rpc_cb cb, void *private_data)
+{
+	struct rpc_pdu *pdu;
+
+	pdu = rpc_allocate_pdu(rpc, RQUOTA_PROGRAM, RQUOTA_V2, RQUOTA2_NULL, cb, private_data, (xdrproc_t)xdr_void, 0);
+	if (pdu == NULL) {
+		rpc_set_error(rpc, "Out of memory. Failed to allocate pdu for rquota2/null call");
+		return -1;
+	}
+
+	if (rpc_queue_pdu(rpc, pdu) != 0) {
+		rpc_set_error(rpc, "Out of memory. Failed to queue pdu for rquota2/null call");
+		rpc_free_pdu(rpc, pdu);
+		return -2;
+	}
+
+	return 0;
+}
+
+int rpc_rquota2_getquota_async(struct rpc_context *rpc, rpc_cb cb, char *export, int type, int uid, void *private_data)
+{
+	struct rpc_pdu *pdu;
+	GETQUOTA2args args;
+
+	pdu = rpc_allocate_pdu(rpc, RQUOTA_PROGRAM, RQUOTA_V2, RQUOTA2_GETQUOTA, cb, private_data, (xdrproc_t)xdr_GETQUOTA1res, sizeof(GETQUOTA1res));
+	if (pdu == NULL) {
+		rpc_set_error(rpc, "Out of memory. Failed to allocate pdu for rquota2/getquota call");
+		return -1;
+	}
+
+	args.export  = export;
+	args.type    = type;
+	args.uid     = uid;
+
+	if (xdr_GETQUOTA2args(&pdu->xdr, &args) == 0) {
+		rpc_set_error(rpc, "XDR error: Failed to encode GETQUOTA2args");
+		rpc_free_pdu(rpc, pdu);
+		return -2;
+	}
+
+	if (rpc_queue_pdu(rpc, pdu) != 0) {
+		rpc_set_error(rpc, "Out of memory. Failed to queue pdu for rquota2/getquota call");
+		rpc_free_pdu(rpc, pdu);
+		return -3;
+	}
+
+	return 0;
+}
+
+int rpc_rquota2_getactivequota_async(struct rpc_context *rpc, rpc_cb cb, char *export, int type, int uid, void *private_data)
+{
+	struct rpc_pdu *pdu;
+	GETQUOTA2args args;
+
+	pdu = rpc_allocate_pdu(rpc, RQUOTA_PROGRAM, RQUOTA_V2, RQUOTA2_GETACTIVEQUOTA, cb, private_data, (xdrproc_t)xdr_GETQUOTA1res, sizeof(GETQUOTA1res));
+	if (pdu == NULL) {
+		rpc_set_error(rpc, "Out of memory. Failed to allocate pdu for rquota2/getactivequota call");
+		return -1;
+	}
+
+	args.export  = export;
+	args.type    = type;
+	args.uid     = uid;
+
+	if (xdr_GETQUOTA2args(&pdu->xdr, &args) == 0) {
+		rpc_set_error(rpc, "XDR error: Failed to encode GETQUOTA2args");
+		rpc_free_pdu(rpc, pdu);
+		return -2;
+	}
+
+	if (rpc_queue_pdu(rpc, pdu) != 0) {
+		rpc_set_error(rpc, "Out of memory. Failed to queue pdu for rquota2/getactivequota call");
 		rpc_free_pdu(rpc, pdu);
 		return -3;
 	}