1 files changed, 190 insertions, 4 deletions
diff --git a/nss/protocols.c b/nss/protocols.c
index 165c24f..244cdac 100644
--- a/nss/protocols.c
+++ b/nss/protocols.c
@@ -2,7 +2,8 @@
    protocols.c - NSS lookup functions for protocol database
 
    Copyright (C) 2006 West Consulting
-   Copyright (C) 2006, 2007, 2008 Arthur de Jong
+   Copyright (C) 2006, 2007, 2008, 2010 Arthur de Jong
+   Copyright (C) 2010 Symas Corporation
 
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
@@ -41,14 +42,26 @@ static nss_status_t read_protoent(
   return NSS_STATUS_SUCCESS;
 }
 
-nss_status_t _nss_ldap_getprotobyname_r(const char *name,struct protoent *result,char *buffer,size_t buflen,int *errnop)
+#ifdef HAVE_NSSWITCH_H
+nss_status_t _nss_nslcd_getprotobyname_r(
+#else /* not HAVE_NSSWITCH_H */
+nss_status_t _nss_ldap_getprotobyname_r(
+#endif /* HAVE_NSSWITCH_H */
+        const char *name,struct protoent *result,char *buffer,
+        size_t buflen,int *errnop)
 {
   NSS_BYNAME(NSLCD_ACTION_PROTOCOL_BYNAME,
              name,
              read_protoent(fp,result,buffer,buflen,errnop));
 }
 
-nss_status_t _nss_ldap_getprotobynumber_r(int number,struct protoent *result,char *buffer,size_t buflen,int *errnop)
+#ifdef HAVE_NSSWITCH_H
+nss_status_t _nss_nslcd_getprotobynumber_r(
+#else /* not HAVE_NSSWITCH_H */
+nss_status_t _nss_ldap_getprotobynumber_r(
+#endif /* HAVE_NSSWITCH_H */
+        int number,struct protoent *result,char *buffer,
+        size_t buflen,int *errnop)
 {
   NSS_BYINT32(NSLCD_ACTION_PROTOCOL_BYNUMBER,
               number,
@@ -58,18 +71,191 @@ nss_status_t _nss_ldap_getprotobynumber_r(int number,struct protoent *result,cha
 /* thread-local file pointer to an ongoing request */
 static __thread TFILE *protoentfp;
 
+#ifdef HAVE_NSSWITCH_H
+nss_status_t _nss_ldap_setprotoent(nss_backend_t *proto_context,void *fakeargs)
+#else /* not HAVE_NSSWITCH_H */
 nss_status_t _nss_ldap_setprotoent(int UNUSED(stayopen))
+#endif /* HAVE_NSSWITCH_H */
 {
   NSS_SETENT(protoentfp);
 }
 
-nss_status_t _nss_ldap_getprotoent_r(struct protoent *result,char *buffer,size_t buflen,int *errnop)
+#ifdef HAVE_NSSWITCH_H
+nss_status_t _nss_nslcd_getprotoent_r(
+#else /* not HAVE_NSSWITCH_H */
+nss_status_t _nss_ldap_getprotoent_r(
+#endif /* HAVE_NSSWITCH_H */
+        struct protoent *result,char *buffer,size_t buflen,int *errnop)
 {
   NSS_GETENT(protoentfp,NSLCD_ACTION_PROTOCOL_ALL,
              read_protoent(protoentfp,result,buffer,buflen,errnop));
 }
 
+#ifdef HAVE_NSSWITCH_H
+nss_status_t _nss_ldap_endprotoent(nss_backend_t *proto_context,void *fakeargs)
+#else /* not HAVE_NSSWITCH_H */
 nss_status_t _nss_ldap_endprotoent(void)
+#endif /* HAVE_NSSWITCH_H */
+
 {
   NSS_ENDENT(protoentfp);
 }
+
+#ifdef HAVE_NSSWITCH_H
+
+static nss_status_t _nss_ldap_getprotobyname_r(nss_backend_t *be,void *args)
+{
+  struct protoent priv_proto;
+  struct protoent *proto=NSS_ARGS(args)->buf.result?(struct protoent *)NSS_ARGS(args)->buf.result:&priv_proto;
+  char *name=(char *)NSS_ARGS(args)->key.name;
+  char *buffer=NSS_ARGS(args)->buf.buffer;
+  size_t buflen=NSS_ARGS(args)->buf.buflen;
+  char *data_ptr;
+  nss_status_t status;
+  if (NSS_ARGS(args)->buf.buflen < 0)
+  {
+    NSS_ARGS(args)->erange=1;
+    return NSS_STATUS_TRYAGAIN;
+  }
+  status=_nss_nslcd_getprotobyname_r(name,proto ,buffer,buflen,&errno);
+  if (status!=NSS_STATUS_SUCCESS)
+    return status;
+  if (!NSS_ARGS(args)->buf.result)
+  {
+    /* result==NULL, return file format */
+    data_ptr=(char *)malloc(buflen);
+    sprintf(data_ptr,"%s\t\t%d",proto->p_name,proto->p_proto);
+    if (proto->p_aliases)
+    {
+      int i;
+      for (i=0; proto->p_aliases[i]; i++)
+      {
+        strcat(data_ptr," ");
+        strcat(data_ptr,proto->p_aliases[i]);
+      }
+    }
+    strcpy(buffer,data_ptr);
+    free(data_ptr);
+    NSS_ARGS(args)->returnval=NSS_ARGS(args)->buf.buffer;
+    NSS_ARGS(args)->returnlen=strlen(NSS_ARGS(args)->buf.buffer);
+  }
+  else
+  {
+    NSS_ARGS(args)->returnval=NSS_ARGS(args)->buf.result;
+  }
+  return status;
+}
+
+static nss_status_t _nss_ldap_getprotobynumber_r(nss_backend_t *be,void *args)
+{
+  struct protoent priv_proto;
+  struct protoent *proto=NSS_ARGS(args)->buf.result?(struct protoent *)NSS_ARGS(args)->buf.result:&priv_proto;
+  int number=NSS_ARGS(args)->key.number;
+  char *buffer=NSS_ARGS(args)->buf.buffer;
+  size_t buflen=NSS_ARGS(args)->buf.buflen;
+  char *data_ptr;
+  nss_status_t status;
+  if (NSS_ARGS(args)->buf.buflen<0)
+  {
+    NSS_ARGS(args)->erange=1;
+    return NSS_STATUS_TRYAGAIN;
+  }
+  status=_nss_nslcd_getprotobynumber_r(number,proto,buffer,buflen,&errno);
+  if (status!=NSS_STATUS_SUCCESS)
+    return status;
+  if (!NSS_ARGS(args)->buf.result)
+  {
+    /* result==NULL, return file format */
+    data_ptr=(char *)malloc(buflen);
+    sprintf(data_ptr,"%s\t\t%d",proto->p_name,proto->p_proto);
+    if (proto->p_aliases)
+    {
+      int i;
+      for (i=0; proto->p_aliases[i]; i++)
+      {
+        strcat(data_ptr," ");
+        strcat(data_ptr,proto->p_aliases[i]);
+      }
+    }
+    strcpy(buffer,data_ptr);
+    free(data_ptr);
+    NSS_ARGS(args)->returnval=NSS_ARGS(args)->buf.buffer;
+    NSS_ARGS(args)->returnlen=strlen(NSS_ARGS(args)->buf.buffer);
+  }
+  else
+  {
+    NSS_ARGS(args)->returnval=NSS_ARGS(args)->buf.result;
+  }
+  return status;
+}
+
+static nss_status_t _nss_ldap_getprotoent_r(nss_backend_t *proto_context,void *args)
+{
+  struct protoent priv_proto;
+  struct protoent *proto=NSS_ARGS(args)->buf.result?(struct protoent *)NSS_ARGS(args)->buf.result:&priv_proto;
+  char *buffer=NSS_ARGS(args)->buf.buffer;
+  size_t buflen=NSS_ARGS(args)->buf.buflen;
+  char *data_ptr;
+  nss_status_t status;
+  if (NSS_ARGS(args)->buf.buflen < 0)
+  {
+    NSS_ARGS(args)->erange=1;
+    return NSS_STATUS_TRYAGAIN;
+  }
+  status=_nss_nslcd_getprotoent_r(proto,buffer,buflen,&errno);
+  if (status!=NSS_STATUS_SUCCESS)
+    return status;
+  if (!NSS_ARGS(args)->buf.result)
+  {
+    /* result==NULL, return file format */
+    data_ptr=(char *)malloc(buflen);
+    sprintf(data_ptr,"%s\t\t%d",proto->p_name,proto->p_proto);
+    if (proto->p_aliases)
+    {
+      int i;
+      for (i=0; proto->p_aliases[i]; i++)
+      {
+        strcat(data_ptr," ");
+        strcat(data_ptr,proto->p_aliases[i]);
+      }
+    }
+    strcpy(buffer,data_ptr);
+    free(data_ptr);
+    NSS_ARGS(args)->returnval=NSS_ARGS(args)->buf.buffer;
+    NSS_ARGS(args)->returnlen=strlen(NSS_ARGS(args)->buf.buffer);
+  }
+  else
+  {
+    NSS_ARGS(args)->returnval=NSS_ARGS(args)->buf.result;
+  }
+  return status;
+}
+
+static nss_status_t _nss_ldap_protocols_destr(nss_backend_t *proto_context,void *args)
+{
+  return _nss_ldap_default_destr(proto_context,args);
+}
+
+static nss_backend_op_t proto_ops[]={
+  _nss_ldap_protocols_destr,
+  _nss_ldap_endprotoent,
+  _nss_ldap_setprotoent,
+  _nss_ldap_getprotoent_r,
+  _nss_ldap_getprotobyname_r,
+  _nss_ldap_getprotobynumber_r
+};
+
+nss_backend_t *_nss_ldap_protocols_constr(const char *db_name,
+                            const char *src_name,const char *cfg_args)
+{
+  nss_ldap_backend_t *be;
+  if (!(be=(nss_ldap_backend_t *)malloc(sizeof(*be))))
+    return NULL;
+  be->ops=proto_ops;
+  be->n_ops=sizeof(proto_ops)/sizeof(nss_backend_op_t);
+  if (_nss_ldap_default_constr(be)!=NSS_STATUS_SUCCESS)
+    return NULL;
+  return (nss_backend_t *)be;
+}
+
+#endif /* HAVE_NSSWITCH_H */