src/auto-uid.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

#include <pwd.h>
#include <sys/types.h>
#include <unistd.h>

#include "buffer.h"
#include "exit.h"
#include "fmt.h"
#include "scan.h"

char inbuf[256];
buffer b = BUFFER_INIT(write, 1, inbuf, sizeof(inbuf));

void outs(char *s) /* was named puts, but Solaris pwd.h includes stdio.h. dorks. */
{
  if (buffer_puts(&b, s) == -1) _exit(111);
}

int main(int argc, char **argv)
{
  char *name;
  char *value;
  struct passwd *pw;
  char strnum[FMT_ULONG];

  name = argv[1];
  if (!name) _exit(100);
  value = argv[2];
  if (!value) _exit(100);

  pw = getpwnam(value);
  if (!pw) {
    buffer_puts(buffer_2, "fatal: unable to find user ");
    buffer_puts(buffer_2, value);
    buffer_puts(buffer_2, "\n");
    buffer_flush(buffer_2);
    _exit(111);
  }

  strnum[fmt_ulong(strnum, (unsigned long)pw->pw_uid)] = 0;

  outs("int ");
  outs(name);
  outs(" = ");
  outs(strnum);
  outs(";\n");
  if (buffer_flush(&b) == -1) _exit(111);
  _exit(0);
}