#define _GNU_SOURCE
#include "pwf.h"
#include <grp.h>
#include <string.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <byteswap.h>
#include <errno.h>
#include "nscd.h"

int getgrouplist(const char* user, gid_t gid, gid_t* groups, int* ngroups) {
  int rv, nlim, ret = -1;
  ssize_t i, n = 1;
  struct group gr;
  struct group* res;
  FILE* f;
  int swap = 0;
  int32_t resp[INITGR_LEN];
  uint32_t* nscdbuf = 0;
  char* buf = 0;
  char** mem = 0;
  size_t nmem = 0;
  size_t size;
  nlim = *ngroups;
  if (nlim >= 1)
    *groups++ = gid;

  f = __nscd_query(GETINITGR, user, resp, sizeof resp, &swap);
  if (!f)
    goto cleanup;
  if (resp[INITGRFOUND]) {
    nscdbuf = calloc(resp[INITGRNGRPS], sizeof(uint32_t));
    if (!nscdbuf)
      goto cleanup;
    if (!fread(nscdbuf, sizeof(*nscdbuf) * resp[INITGRNGRPS], 1, f)) {
      if (!ferror(f))
        errno = EIO;
      goto cleanup;
    }
    if (swap) {
      for (i = 0; i < resp[INITGRNGRPS]; i++)
        nscdbuf[i] = bswap_32(nscdbuf[i]);
    }
  }
  fclose(f);

  f = fopen("/etc/group", "rbe");
  if (!f && errno != ENOENT && errno != ENOTDIR)
    goto cleanup;

  if (f) {
    while (!(rv = __getgrent_a(f, &gr, &buf, &size, &mem, &nmem, &res)) &&
           res) {
      if (nscdbuf)
        for (i = 0; i < resp[INITGRNGRPS]; i++) {
          if (nscdbuf[i] == gr.gr_gid)
            nscdbuf[i] = gid;
        }
      for (i = 0; gr.gr_mem[i] && strcmp(user, gr.gr_mem[i]); i++)
        ;
      if (!gr.gr_mem[i])
        continue;
      if (++n <= nlim)
        *groups++ = gr.gr_gid;
    }
    if (rv) {
      errno = rv;
      goto cleanup;
    }
  }
  if (nscdbuf) {
    for (i = 0; i < resp[INITGRNGRPS]; i++) {
      if (nscdbuf[i] != gid)
        if (++n <= nlim)
          *groups++ = nscdbuf[i];
    }
  }

  ret = n > nlim ? -1 : n;
  *ngroups = n;

cleanup:
  if (f)
    fclose(f);
  free(nscdbuf);
  free(buf);
  free(mem);
  return ret;
}