Add support for client port in audit record

Closes Ericsson#90
eperott · May 10, 2019 · 0e28c1b · 0e28c1b
1 parent ff6b4da
commit 0e28c1b
Show file tree

Hide file tree

Showing 29 changed files with 423 additions and 220 deletions.
diff --git a/...java/com/ericsson/bss/cassandra/ecaudit/common/chronicle/AuditRecordReadMarshallable.java b/...java/com/ericsson/bss/cassandra/ecaudit/common/chronicle/AuditRecordReadMarshallable.java
@@ -16,6 +16,7 @@
 package com.ericsson.bss.cassandra.ecaudit.common.chronicle;
 
 import java.net.InetAddress;
+import java.net.InetSocketAddress;
 import java.net.UnknownHostException;
 import java.util.UUID;
 
@@ -48,8 +49,8 @@ public void readMarshallable(@NotNull WireIn wire) throws IORuntimeException
         SimpleAuditRecord.Builder builder = SimpleAuditRecord
                                             .builder()
                                             .withTimestamp(wire.read(WireTags.KEY_TIMESTAMP).int64())
-                                            .withClientAddress(readAddress(wire, WireTags.KEY_CLIENT))
-                                            .withCoordinatorAddress(readAddress(wire, WireTags.KEY_COORDINATOR))
+                                            .withClientAddress(readInetSocketAddress(wire, WireTags.KEY_CLIENT_IP, WireTags.KEY_CLIENT_PORT))
+                                            .withCoordinatorAddress(readInetAddress(wire, WireTags.KEY_COORDINATOR_IP))
                                             .withUser(wire.read(WireTags.KEY_USER).text());
 
         if (WireTags.VALUE_TYPE_BATCH_ENTRY.equals(type))
@@ -83,15 +84,29 @@ private String readType(WireIn wire) throws IORuntimeException
         return type;
     }
 
-    private InetAddress readAddress(WireIn wire, String key) throws IORuntimeException
+    private InetSocketAddress readInetSocketAddress(WireIn wire, String ipKey, String portKey) throws IORuntimeException
+    {
+        try
+        {
+            InetAddress inetAddress = InetAddress.getByAddress(wire.read(ipKey).bytes());
+            int port = wire.read(portKey).int32();
+            return new InetSocketAddress(inetAddress, port);
+        }
+        catch (UnknownHostException e)
+        {
+            throw new IORuntimeException("Corrupt " + ipKey + " field", e);
+        }
+    }
+
+    private InetAddress readInetAddress(WireIn wire, String key) throws IORuntimeException
     {
         try
         {
             return InetAddress.getByAddress(wire.read(key).bytes());
         }
         catch (UnknownHostException e)
         {
-            throw new IORuntimeException("Corrupt " + key + " IP address field", e);
+            throw new IORuntimeException("Corrupt " + key + " field", e);
         }
     }
 

diff --git a/...ava/com/ericsson/bss/cassandra/ecaudit/common/chronicle/AuditRecordWriteMarshallable.java b/...ava/com/ericsson/bss/cassandra/ecaudit/common/chronicle/AuditRecordWriteMarshallable.java
@@ -37,8 +37,9 @@ public void writeMarshallable(@NotNull WireOut wire)
         wire.write(WireTags.KEY_VERSION).int16(WireTags.VALUE_VERSION_CURRENT);
         wire.write(WireTags.KEY_TYPE).text(type);
         wire.write(WireTags.KEY_TIMESTAMP).int64(auditRecord.getTimestamp());
-        wire.write(WireTags.KEY_CLIENT).bytes(auditRecord.getClientAddress().getAddress());
-        wire.write(WireTags.KEY_COORDINATOR).bytes(auditRecord.getCoordinatorAddress().getAddress());
+        wire.write(WireTags.KEY_CLIENT_IP).bytes(auditRecord.getClientAddress().getAddress().getAddress());
+        wire.write(WireTags.KEY_CLIENT_PORT).int32(auditRecord.getClientAddress().getPort());
+        wire.write(WireTags.KEY_COORDINATOR_IP).bytes(auditRecord.getCoordinatorAddress().getAddress());
         wire.write(WireTags.KEY_USER).text(auditRecord.getUser());
         auditRecord.getBatchId().ifPresent(batchId -> wire.write(WireTags.KEY_BATCH_ID).uuid(batchId));
         wire.write(WireTags.KEY_STATUS).text(auditRecord.getStatus().name());

diff --git a/common/src/main/java/com/ericsson/bss/cassandra/ecaudit/common/chronicle/WireTags.java b/common/src/main/java/com/ericsson/bss/cassandra/ecaudit/common/chronicle/WireTags.java
@@ -20,8 +20,9 @@ class WireTags
     static final String KEY_VERSION = "version";
     static final String KEY_TYPE = "type";
     static final String KEY_TIMESTAMP = "timestamp";
-    static final String KEY_CLIENT = "client";
-    static final String KEY_COORDINATOR = "coordinator";
+    static final String KEY_CLIENT_IP = "client_ip";
+    static final String KEY_CLIENT_PORT = "client_port";
+    static final String KEY_COORDINATOR_IP = "coordinator_ip";
     static final String KEY_USER = "user";
     static final String KEY_BATCH_ID = "batchId";
     static final String KEY_STATUS = "status";

diff --git a/common/src/main/java/com/ericsson/bss/cassandra/ecaudit/common/record/AuditRecord.java b/common/src/main/java/com/ericsson/bss/cassandra/ecaudit/common/record/AuditRecord.java
@@ -16,14 +16,15 @@
 package com.ericsson.bss.cassandra.ecaudit.common.record;
 
 import java.net.InetAddress;
+import java.net.InetSocketAddress;
 import java.util.Optional;
 import java.util.UUID;
 
 public interface AuditRecord
 {
     Long getTimestamp();
 
-    InetAddress getClientAddress();
+    InetSocketAddress getClientAddress();
 
     InetAddress getCoordinatorAddress();
 

diff --git a/common/src/main/java/com/ericsson/bss/cassandra/ecaudit/common/record/SimpleAuditRecord.java b/common/src/main/java/com/ericsson/bss/cassandra/ecaudit/common/record/SimpleAuditRecord.java
@@ -16,12 +16,13 @@
 package com.ericsson.bss.cassandra.ecaudit.common.record;
 
 import java.net.InetAddress;
+import java.net.InetSocketAddress;
 import java.util.Optional;
 import java.util.UUID;
 
 public class SimpleAuditRecord implements AuditRecord
 {
-    private final InetAddress clientAddress;
+    private final InetSocketAddress clientAddress;
     private final InetAddress coordinatorAddress;
     private final String user;
     private final UUID batchId;
@@ -47,7 +48,7 @@ public Long getTimestamp()
     }
 
     @Override
-    public InetAddress getClientAddress()
+    public InetSocketAddress getClientAddress()
     {
         return clientAddress;
     }
@@ -89,15 +90,15 @@ public static Builder builder()
 
     public static class Builder
     {
-        private InetAddress clientAddress;
+        private InetSocketAddress clientAddress;
         private InetAddress coordinatorAddress;
         private String user;
         private UUID batchId;
         private Status status;
         private AuditOperation operation;
         private long timestamp;
 
-        public Builder withClientAddress(InetAddress clientAddress)
+        public Builder withClientAddress(InetSocketAddress clientAddress)
         {
             this.clientAddress = clientAddress;
             return this;