[Add Local Addr & Port 2/3] Add columns to socket tracer (#1807)

Summary: Adds data columns for local IP address and port to the socket
tracer, which are populated by #1808 and #1809. This will support
standalone pem entity relationships.

Type of change: /kind feature

Test Plan: Existing targets

Signed-off-by: Benjamin Kilimnik <bkilimnik@pixielabs.ai>

Author: benkilimnik

src/carnot/goplanner/schema_test.go

@@ -44,6 +44,16 @@ relation_map {
 			column_type: INT64
 			column_semantic_type: ST_PORT
 		}
+		columns {
+			column_name: "local_addr"
+			column_type: STRING
+			column_semantic_type: ST_IP_ADDRESS
+		}
+		columns {
+			column_name: "local_port"
+			column_type: INT64
+			column_semantic_type: ST_PORT
+		}
 		columns {
 			column_name: "protocol"
 			column_type: INT64

src/stirling/source_connectors/socket_tracer/amqp_table.h

@@ -33,6 +33,8 @@ static constexpr DataElement kAMQPElements[] = {
         canonical_data_elements::kUPID,
         canonical_data_elements::kRemoteAddr,
         canonical_data_elements::kRemotePort,
+        canonical_data_elements::kLocalAddr,
+        canonical_data_elements::kLocalPort,
         canonical_data_elements::kTraceRole,
         {
         "frame_type", "AMQP request command",

src/stirling/source_connectors/socket_tracer/canonical_types.h

@@ -45,6 +45,22 @@ constexpr DataElement kRemotePort = {
     types::PatternType::GENERAL,
 };
 
+constexpr DataElement kLocalAddr = {
+    "local_addr",
+    "IP address of the local endpoint.",
+    types::DataType::STRING,
+    types::SemanticType::ST_IP_ADDRESS,
+    types::PatternType::GENERAL,
+};
+
+constexpr DataElement kLocalPort = {
+    "local_port",
+    "Port of the local endpoint.",
+    types::DataType::INT64,
+    types::SemanticType::ST_PORT,
+    types::PatternType::GENERAL,
+};
+
 constexpr DataElement kTraceRole = {
     "trace_role",
     "The role (client-or-server) of the process that owns the connections.",

src/stirling/source_connectors/socket_tracer/cass_table.h

@@ -39,6 +39,8 @@ static constexpr DataElement kCQLElements[] = {
         canonical_data_elements::kUPID,
         canonical_data_elements::kRemoteAddr,
         canonical_data_elements::kRemotePort,
+        canonical_data_elements::kLocalAddr,
+        canonical_data_elements::kLocalPort,
         canonical_data_elements::kTraceRole,
         {"req_op", "Request opcode",
          types::DataType::INT64,

src/stirling/source_connectors/socket_tracer/conn_tracker.h

@@ -337,6 +337,11 @@ class ConnTracker : NotCopyMoveable {
    */
   const SockAddr& remote_endpoint() const { return open_info_.remote_addr; }
 
+  /**
+   * Get local IP endpoint of the connection.
+   */
+  const SockAddr& local_endpoint() const { return open_info_.local_addr; }
+
   /**
    * Get the connection information (e.g. remote IP, port, PID, etc.) for this connection.
    */
@@ -801,6 +806,8 @@ std::string DebugString(const ConnTracker& c, std::string_view prefix) {
   info += absl::Substitute("state=$0\n", magic_enum::enum_name(c.state()));
   info += absl::Substitute("$0remote_addr=$1:$2\n", prefix, c.remote_endpoint().AddrStr(),
                            c.remote_endpoint().port());
+  info += absl::Substitute("$0local_addr=$1:$2\n", prefix, c.local_endpoint().AddrStr(),
+                           c.local_endpoint().port());
   info += absl::Substitute("$0protocol=$1\n", prefix, magic_enum::enum_name(c.protocol()));
   if constexpr (std::is_same_v<TFrameType, protocols::http2::Stream>) {
     info += c.http2_client_streams_.DebugString(absl::StrCat(prefix, "  "));

src/stirling/source_connectors/socket_tracer/dns_table.h

@@ -34,6 +34,8 @@ static constexpr DataElement kDNSElements[] = {
         canonical_data_elements::kUPID,
         canonical_data_elements::kRemoteAddr,
         canonical_data_elements::kRemotePort,
+        canonical_data_elements::kLocalAddr,
+        canonical_data_elements::kLocalPort,
         canonical_data_elements::kTraceRole,
         {"req_header", "Request header",
          types::DataType::STRING,

src/stirling/source_connectors/socket_tracer/http_table.h

@@ -43,6 +43,8 @@ constexpr DataElement kHTTPElements[] = {
         canonical_data_elements::kUPID,
         canonical_data_elements::kRemoteAddr,
         canonical_data_elements::kRemotePort,
+        canonical_data_elements::kLocalAddr,
+        canonical_data_elements::kLocalPort,
         canonical_data_elements::kTraceRole,
         {"major_version", "HTTP major version, can be 1 or 2",
          types::DataType::INT64,
@@ -112,6 +114,8 @@ constexpr int kHTTPTimeIdx = kHTTPTable.ColIndex("time_");
 constexpr int kHTTPUPIDIdx = kHTTPTable.ColIndex("upid");
 constexpr int kHTTPRemoteAddrIdx = kHTTPTable.ColIndex("remote_addr");
 constexpr int kHTTPRemotePortIdx = kHTTPTable.ColIndex("remote_port");
+constexpr int kHTTPLocalAddrIdx = kHTTPTable.ColIndex("local_addr");
+constexpr int kHTTPLocalPortIdx = kHTTPTable.ColIndex("local_port");
 constexpr int kHTTPTraceRoleIdx = kHTTPTable.ColIndex("trace_role");
 constexpr int kHTTPMajorVersionIdx = kHTTPTable.ColIndex("major_version");
 constexpr int kHTTPMinorVersionIdx = kHTTPTable.ColIndex("minor_version");

src/stirling/source_connectors/socket_tracer/kafka_table.h

@@ -37,6 +37,8 @@ static constexpr DataElement kKafkaElements[] = {
       canonical_data_elements::kUPID,
       canonical_data_elements::kRemoteAddr,
       canonical_data_elements::kRemotePort,
+      canonical_data_elements::kLocalAddr,
+      canonical_data_elements::kLocalPort,
       canonical_data_elements::kTraceRole,
       {"req_cmd", "Kafka request command",
        types::DataType::INT64,

src/stirling/source_connectors/socket_tracer/mongodb_table.h

@@ -34,6 +34,8 @@ static constexpr DataElement kMongoDBElements[] = {
         canonical_data_elements::kUPID,
         canonical_data_elements::kRemoteAddr,
         canonical_data_elements::kRemotePort,
+        canonical_data_elements::kLocalAddr,
+        canonical_data_elements::kLocalPort,
         canonical_data_elements::kTraceRole,
         {"req_cmd", "MongoDB request command",
          types::DataType::STRING,

src/stirling/source_connectors/socket_tracer/mux_table.h

@@ -34,6 +34,8 @@ static constexpr DataElement kMuxElements[] = {
         canonical_data_elements::kUPID,
         canonical_data_elements::kRemoteAddr,
         canonical_data_elements::kRemotePort,
+        canonical_data_elements::kLocalAddr,
+        canonical_data_elements::kLocalPort,
         canonical_data_elements::kTraceRole,
         {"req_type", "Mux message request type",
          types::DataType::INT64,