X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=lib%2Finit.c;h=36d4880006d9f244b804fdf0a8b2631c2a220924;hb=54bafab240be2a172ce8075fd25497a149bdf77c;hp=2a451a16d51f4282b76b40f0d32ee416aae72ef0;hpb=6fda9871a037fdbac341f55f0b9a175e3977281e;p=deb_libnfs.git diff --git a/lib/init.c b/lib/init.c index 2a451a1..36d4880 100644 --- a/lib/init.c +++ b/lib/init.c @@ -57,7 +57,9 @@ struct rpc_context *rpc_init_context(void) memset(rpc, 0, sizeof(struct rpc_context)); rpc->magic = RPC_CONTEXT_MAGIC; - rpc->encodebuflen = 65536; + + /* Allow 1M of data (for writes) and some */ + rpc->encodebuflen = 1024 * 1024 + 4096; rpc->encodebuf = malloc(rpc->encodebuflen); if (rpc->encodebuf == NULL) { free(rpc); @@ -73,6 +75,14 @@ struct rpc_context *rpc_init_context(void) rpc->xid = salt + time(NULL) + getpid() << 16; salt += 0x01000000; rpc->fd = -1; + rpc->tcp_syncnt = RPC_PARAM_UNDEFINED; +#if defined(WIN32) || defined(ANDROID) + rpc->uid = 65534; + rpc->gid = 65534; +#else + rpc->uid = getuid(); + rpc->gid = getgid(); +#endif return rpc; } @@ -100,20 +110,40 @@ void rpc_set_auth(struct rpc_context *rpc, struct AUTH *auth) rpc->auth = auth; } +static void rpc_set_uid_gid(struct rpc_context *rpc, int uid, int gid) { + if (uid != rpc->uid || gid != rpc->gid) { + struct AUTH *auth = libnfs_authunix_create("libnfs", uid, gid, 0, NULL); + if (auth != NULL) { + rpc_set_auth(rpc, auth); + rpc->uid = uid; + rpc->gid = gid; + } + } +} + +void rpc_set_uid(struct rpc_context *rpc, int uid) { + rpc_set_uid_gid(rpc, uid, rpc->gid); +} + +void rpc_set_gid(struct rpc_context *rpc, int gid) { + rpc_set_uid_gid(rpc, rpc->uid, gid); +} void rpc_set_error(struct rpc_context *rpc, char *error_string, ...) { va_list ap; + char *old_error_string = rpc->error_string; assert(rpc->magic == RPC_CONTEXT_MAGIC); - if (rpc->error_string != NULL) { - free(rpc->error_string); - } va_start(ap, error_string); rpc->error_string = malloc(1024); vsnprintf(rpc->error_string, 1024, error_string, ap); va_end(ap); + + if (old_error_string != NULL) { + free(old_error_string); + } } char *rpc_get_error(struct rpc_context *rpc)