Imported Upstream version 0.2.5

[platform/upstream/libtirpc.git] / tirpc / rpc / auth.h

diff --git a/tirpc/rpc/auth.h b/tirpc/rpc/auth.h
index 734e6b9..7c8f813 100644 (file)
--- a/tirpc/rpc/auth.h
+++ b/tirpc/rpc/auth.h
@@ -203,8 +203,22 @@ typedef struct __auth {
 
        } *ah_ops;
        void *ah_private;
+       int ah_refcnt;
 } AUTH;
 
+static __inline int
+auth_get(AUTH *auth)
+{
+       return __sync_add_and_fetch(&auth->ah_refcnt, 1);
+}
+
+static __inline int
+auth_put(AUTH *auth)
+{
+       return __sync_sub_and_fetch(&auth->ah_refcnt, 1);
+}
+
+
 
 /*
  * Authentication ops.
@@ -234,10 +248,19 @@ typedef struct __auth {
 #define auth_refresh(auth, msg)                \
                ((*((auth)->ah_ops->ah_refresh))(auth, msg))
 
-#define AUTH_DESTROY(auth)             \
-               ((*((auth)->ah_ops->ah_destroy))(auth))
-#define auth_destroy(auth)             \
-               ((*((auth)->ah_ops->ah_destroy))(auth))
+#define AUTH_DESTROY(auth)                                             \
+               do {                                                    \
+                       int refs;                                       \
+                       if ((refs = auth_put((auth))) == 0)             \
+                               ((*((auth)->ah_ops->ah_destroy))(auth));\
+               } while (0)
+
+#define auth_destroy(auth)                                             \
+               do {                                                    \
+                       int refs;                                       \
+                       if ((refs = auth_put((auth))) == 0)             \
+                               ((*((auth)->ah_ops->ah_destroy))(auth));\
+               } while (0)
 
 #define AUTH_WRAP(auth, xdrs, xfunc, xwhere)            \
                ((*((auth)->ah_ops->ah_wrap))(auth, xdrs, \
@@ -373,9 +396,10 @@ __END_DECLS
 __BEGIN_DECLS
 struct svc_req;
 struct rpc_msg;
-enum auth_stat _svcauth_null (struct svc_req *, struct rpc_msg *);
+enum auth_stat _svcauth_none (struct svc_req *, struct rpc_msg *);
 enum auth_stat _svcauth_short (struct svc_req *, struct rpc_msg *);
 enum auth_stat _svcauth_unix (struct svc_req *, struct rpc_msg *);
+enum auth_stat _svcauth_gss (struct svc_req *, struct rpc_msg *, bool_t *);
 __END_DECLS
 
 #define AUTH_NONE      0               /* no authentication */