add client IP to audit logs

models/audit.go

@@ -12,10 +12,11 @@ type Audit struct {
 	SecretId  int       `db:"SecretId" json:"secretId"`
 	EventText string    `db:"EventText" json:"eventText"`
 	EventTime time.Time `db:"EventTime" json:"eventTime"`
+	IpAddress string    `db:"IpAddress" json:"ipAddress"`
 }
 
-// AutidLogAdd adds a new audit record to the database
-func (a *Audit) AutidLogAdd() (*Audit, error) {
+// AuditLogAdd adds a new audit record to the database
+func (a *Audit) AuditLogAdd() (*Audit, error) {
 	var err error
 
 	// Populate timestamp field if not already set
@@ -23,16 +24,16 @@ func (a *Audit) AutidLogAdd() (*Audit, error) {
 		a.EventTime = time.Now().UTC()
 	}
 
-	result, err := db.NamedExec(("INSERT INTO audit (UserId, SecretId, EventText, EventTime) VALUES (:UserId, :SecretId, :EventText, :EventTime);"), a)
+	result, err := db.NamedExec(("INSERT INTO audit (UserId, SecretId, EventText, EventTime, IpAddress) VALUES (:UserId, :SecretId, :EventText, :EventTime, :IpAddress);"), a)
 
 	if err != nil {
-		log.Printf("AutidLogAdd error executing sql record : '%s'\n", err)
+		log.Printf("AuditLogAdd error executing sql record : '%s'\n", err)
 		return &Audit{}, err
 	} else {
 		affected, _ := result.RowsAffected()
 		id, _ := result.LastInsertId()
 		a.AuditId = int(id)
-		log.Printf("AutidLogAdd insert returned result id '%d' affecting %d row(s).\n", id, affected)
+		log.Printf("AuditLogAdd insert returned result id '%d' affecting %d row(s).\n", id, affected)
 	}
 
 	return a, nil

models/db.go

@@ -86,6 +86,7 @@ const createAudit string = `
 		UserId INTEGER DEFAULT 0,
 		SecretId INTEGER DEFAULT 0,
 		EventText VARCHAR,
+		IpAddress VARCHAR,
 		EventTime datetime
 	);
 `
@@ -394,6 +395,17 @@ func CreateTables() {
 			os.Exit(1)
 		}
 	}
+
+	// Add IpAddress column to audit table
+	auditIPCheck, _ := CheckColumnExists("audit", "IpAddress")
+	if !auditIPCheck {
+		// Add the column for LdapGroup in the roles table
+		_, err := db.Exec("ALTER TABLE audit ADD COLUMN IpAddress VARCHAR;")
+		if err != nil {
+			log.Printf("Error altering audit table to add IpAddress column : '%s'\n", err)
+			os.Exit(1)
+		}
+	}
 }
 
 // Count the number of records in the sqlite database