/*
   test_getpeercred.c - simple test for the peercred module
   This file is part of the nss-pam-ldapd library.

   Copyright (C) 2008, 2011 Arthur de Jong

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 2.1 of the License, or (at your option) any later version.

   This library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with this library; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
   02110-1301 USA
*/

#include "config.h"

#include <stdio.h>
#include <string.h>
#include <assert.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#ifdef HAVE_GRP_H
#include <grp.h>
#endif /* HAVE_GRP_H */
#include <errno.h>

#include "common.h"

#include "compat/attrs.h"
#include "compat/getpeercred.h"

/* create a named socket */
static int create_socket(const char *name)
{
  int sock;
  struct sockaddr_un addr;
  /* create a socket */
  assertok((sock=socket(PF_UNIX,SOCK_STREAM,0))>=0);
  /* remove existing named socket */
  unlink(name);
  /* create socket address structure */
  memset(&addr,0,sizeof(struct sockaddr_un));
  addr.sun_family=AF_UNIX;
  strncpy(addr.sun_path,name,sizeof(addr.sun_path));
  addr.sun_path[sizeof(addr.sun_path)-1]='\0';
  /* bind to the named socket */
  assertok(bind(sock,(struct sockaddr *)&addr,sizeof(struct sockaddr_un))==0);
  /* close the file descriptor on exit */
  assertok(fcntl(sock,F_SETFD,FD_CLOEXEC)>=0);
  /* start listening for connections */
  assertok(listen(sock,SOMAXCONN)>=0);
  /* we're done */
  return sock;
}

/* accept a connection on the socket */
static int acceptconnection(int sock)
{
  int csock;
  int j;
  struct sockaddr_storage addr;
  socklen_t alen;
  /* accept a new connection */
  alen=(socklen_t)sizeof(struct sockaddr_storage);
  assertok((csock=accept(sock,(struct sockaddr *)&addr,&alen))>=0);
  /* make sure O_NONBLOCK is not inherited */
  assertok((j=fcntl(csock,F_GETFL,0))>=0);
  assertok(fcntl(csock,F_SETFL,j&~O_NONBLOCK)>=0);
  /* return socket */
  return csock;
}

/* open a connection to the named socket */
static int open_socket(const char *name)
{
  int sock;
  struct sockaddr_un addr;
  /* create a socket */
  assertok((sock=socket(PF_UNIX,SOCK_STREAM,0))>=0);
  /* create socket address structure */
  memset(&addr,0,sizeof(struct sockaddr_un));
  addr.sun_family=AF_UNIX;
  strncpy(addr.sun_path,name,sizeof(addr.sun_path));
  addr.sun_path[sizeof(addr.sun_path)-1]='\0';
  /* connect to the socket */
  assertok(connect(sock,(struct sockaddr *)&addr,(socklen_t)sizeof(struct sockaddr_un))>=0);
  /* return the socket */
  return sock;
}

#define SOCKETNAME "/tmp/test_getpeercred.sock"

#define assertwarn(assertion) \
  if (!(assertion)) \
    fprintf(stderr,"test_getpeercred: %s:%d: %s: Assertion `%s' failed\n", \
            __FILE__, __LINE__, __ASSERT_FUNCTION, __STRING(assertion));

/* the main program... */
int main(int UNUSED(argc),char UNUSED(*argv[]))
{
  int ssock;
  int csock;
  int fsock;
  uid_t uid;
  gid_t gid;
  pid_t pid;
  /* create a socket to listen on */
  ssock=create_socket(SOCKETNAME);
  /* open a connection to the socket */
  csock=open_socket(SOCKETNAME);
  /* get a connection from the server socket */
  fsock=acceptconnection(ssock);
  /* look up client information */
  assert(getpeercred(fsock,&uid,&gid,&pid)==0);
  assert(uid==geteuid());
  assertwarn(gid==getegid());
  assertwarn(pid==getpid());
  /* remove the socket */
  unlink(SOCKETNAME);
  return 0;
}