Initial checkin

go_login.go

@@ -0,0 +1,109 @@
+// Golang bindings for basic login/utmp accounting
+package go_login
+
+//#include <stdio.h>
+//#include <stdlib.h>
+//#include <sys/file.h>
+//#include <string.h>
+//#include <bsd/string.h>
+//#include <unistd.h>
+//#include <stdint.h>
+//#include <time.h>
+//#include <pwd.h>
+//
+//#include <utmp.h>
+//#include <utmpx.h>
+//#include <lastlog.h>
+//
+//typedef char char_t;
+//
+//
+//void pututmp(struct utmpx* entry, char* name, char* host) {
+//  int32_t stat = system("echo ---- pre ----;who");
+//
+//  entry->ut_type = USER_PROCESS;
+//  entry->ut_pid = getpid();
+//  strcpy(entry->ut_line, ttyname(STDIN_FILENO) + strlen("/dev/"));
+//  /* only correct for ptys named /dev/tty[pqr][0-9a-z] */
+//
+//  strcpy(entry->ut_id, ttyname(STDIN_FILENO) + strlen("/dev/tty"));
+//  entry->ut_time = time(NULL);
+//  strcpy(entry->ut_user, name);
+//  strcpy(entry->ut_host, host);
+//  entry->ut_addr = 0;
+//  setutxent();
+//  pututxline(entry);
+//
+//  stat = system("echo ---- post ----;who");
+//}
+//
+//void unpututmp(struct utmpx* entry) {
+//  entry->ut_type = DEAD_PROCESS;
+//  memset(entry->ut_line, 0, UT_LINESIZE);
+//  entry->ut_time = 0;
+//  memset(entry->ut_user, 0, UT_NAMESIZE);
+//  setutxent();
+//  pututxline(entry);
+//
+//  int32_t stat = system("echo ---- cleanup ----;who; lastlog");
+//
+//  endutxent();
+//}
+//
+//int putlastlogentry(int64_t t, int uid, char* line, char* host) {
+//  int retval = 0;
+//  FILE *f;
+//  struct lastlog l;
+//
+//  strncpy(l.ll_line, line, UT_LINESIZE);
+//  l.ll_line[UT_LINESIZE-1] = '\0';
+//  strncpy(l.ll_host, host, UT_HOSTSIZE);
+//  l.ll_host[UT_HOSTSIZE-1] = '\0';
+//
+//  l.ll_time = (time_t)t;
+//  //printf("l: ll_line '%s', ll_host '%s', ll_time %d\n", l.ll_line, l.ll_host, l.ll_time);
+//
+//  /* Write lastlog entry at fixed offset (uid * sizeof(struct lastlog) */
+//  if( NULL != (f = fopen("/var/log/lastlog", "rw+")) ) {
+//    if( !fseek(f, (uid * sizeof(struct lastlog)), SEEK_SET) ) {
+//      int fd = fileno(f);
+//      if( write(fd, &l, sizeof(l)) == sizeof(l) ) {
+//        retval = 1;
+//        int32_t stat = system("echo ---- lastlog ----; lastlog");
+//      }
+//    }
+//    fclose(f);
+//  }
+//  return retval;
+//}
+import "C"
+
+import (
+  "fmt"
+  "os/user"
+  "time"
+)
+
+func Put_utmp(user string, host string) (*C.struct_utmpx) {
+  var entry C.struct_utmpx
+
+  C.pututmp(&entry, C.CString(user), C.CString(host))
+  return &entry
+}
+
+func Unput_utmp(entry *C.struct_utmpx) {
+  C.unpututmp(entry)
+}
+
+func Put_lastlog_entry(app string, usr string, host string) {
+  u, e := user.Lookup(usr)
+  if e != nil {
+    return
+  }
+  var uid uint32
+  fmt.Sscanf(u.Uid, "%d", &uid)
+
+  t := time.Now().Unix()
+  stat := C.putlastlogentry(C.int64_t(t), C.int(uid), C.CString(app), C.CString(host))
+  fmt.Println("stat was:",stat)
+}