Update socket_tag -> SocketTag & lowercase tags

src/accountsdb/download.zig

@@ -7,7 +7,6 @@ const ContactInfo = @import("../gossip/data.zig").ContactInfo;
 const GossipTable = @import("../gossip/table.zig").GossipTable;
 const SlotAndHash = @import("./snapshots.zig").SlotAndHash;
 const Logger = @import("../trace/log.zig").Logger;
-const socket_tag = @import("../gossip/data.zig").socket_tag;
 const Hash = @import("../core/hash.zig").Hash;
 
 const DOWNLOAD_PROGRESS_UPDATES_NS = 30 * std.time.ns_per_s;
@@ -96,7 +95,7 @@ pub fn findPeersToDownloadFromAssumeCapacity(
             result.invalid_shred_version += 1;
             continue;
         }
-        _ = peer_contact_info.getSocket(socket_tag.RPC) orelse {
+        _ = peer_contact_info.getSocket(.rpc) orelse {
             result.no_rpc_count += 1;
             continue;
         };
@@ -212,7 +211,7 @@ pub fn downloadSnapshotsFromGossip(
             });
             defer allocator.free(snapshot_filename);
 
-            const rpc_socket = peer.contact_info.getSocket(socket_tag.RPC).?;
+            const rpc_socket = peer.contact_info.getSocket(.rpc).?;
             const rpc_url_bounded = rpc_socket.toStringBounded();
             const rpc_url = rpc_url_bounded.constSlice();
 

src/cmd/cmd.zig

@@ -35,7 +35,7 @@ const parallelUnpackZstdTarBall = sig.accounts_db.parallelUnpackZstdTarBall;
 const requestIpEcho = sig.net.requestIpEcho;
 const servePrometheus = sig.prometheus.servePrometheus;
 
-const socket_tag = sig.gossip.socket_tag;
+const SocketTag = sig.gossip.SocketTag;
 
 // TODO: use better allocator, unless GPA becomes more performant.
 
@@ -411,8 +411,8 @@ fn validator() !void {
         ip_echo_data.shred_version, // TODO atomic owned at top level? or owned by gossip is good?
         ip_echo_data.ip,
         &.{
-            .{ .tag = .REPAIR, .port = repair_port },
-            .{ .tag = .TURBINE_RECV, .port = turbine_recv_port },
+            .{ .tag = .repair, .port = repair_port },
+            .{ .tag = .turbine_recv, .port = turbine_recv_port },
         },
     );
     defer gossip_service.deinit();
@@ -515,7 +515,7 @@ fn initGossip(
     entrypoints: []const SocketAddr,
     shred_version: u16,
     gossip_host_ip: IpAddr,
-    sockets: []const struct { tag: socket_tag, port: u16 },
+    sockets: []const struct { tag: SocketTag, port: u16 },
 ) !GossipService {
     const gossip_port: u16 = config.current.gossip.port;
     logger.infof("gossip host: {any}", .{gossip_host_ip});
@@ -524,7 +524,7 @@ fn initGossip(
     // setup contact info
     const my_pubkey = Pubkey.fromPublicKey(&my_keypair.public_key);
     var contact_info = ContactInfo.init(gpa_allocator, my_pubkey, getWallclockMs(), 0);
-    try contact_info.setSocket(socket_tag.GOSSIP, SocketAddr.init(gossip_host_ip, gossip_port));
+    try contact_info.setSocket(.gossip, SocketAddr.init(gossip_host_ip, gossip_port));
     for (sockets) |s| try contact_info.setSocket(s.tag, SocketAddr.init(gossip_host_ip, s.port));
     contact_info.shred_version = shred_version;
 

src/gossip/active_set.zig

@@ -6,7 +6,6 @@ const _gossip_data = @import("../gossip/data.zig");
 const SignedGossipData = _gossip_data.SignedGossipData;
 const getWallclockMs = _gossip_data.getWallclockMs;
 const ContactInfo = _gossip_data.ContactInfo;
-const socket_tag = _gossip_data.socket_tag;
 const LegacyContactInfo = _gossip_data.LegacyContactInfo;
 
 const Pubkey = @import("../core/pubkey.zig").Pubkey;
@@ -106,7 +105,7 @@ pub const ActiveSet = struct {
         while (iter.next()) |entry| {
             // lookup peer contact info
             const peer_info = table.getContactInfo(entry.key_ptr.*) orelse continue;
-            const peer_gossip_addr = peer_info.getSocket(socket_tag.GOSSIP) orelse continue;
+            const peer_gossip_addr = peer_info.getSocket(.gossip) orelse continue;
 
             peer_gossip_addr.sanitize() catch continue;
 

src/gossip/data.zig

@@ -447,7 +447,7 @@ pub const GossipData = union(enum(u32)) {
     pub fn gossipAddr(self: *const @This()) ?SocketAddr {
         return switch (self.*) {
             .LegacyContactInfo => |*v| if (v.gossip.isUnspecified()) null else v.gossip,
-            .ContactInfo => |*v| v.getSocket(socket_tag.GOSSIP),
+            .ContactInfo => |*v| v.getSocket(.gossip),
             else => null,
         };
     }
@@ -537,32 +537,32 @@ pub const LegacyContactInfo = struct {
     /// call ContactInfo.deinit to free
     pub fn toContactInfo(self: *const LegacyContactInfo, allocator: std.mem.Allocator) !ContactInfo {
         var ci = ContactInfo.init(allocator, self.id, self.wallclock, self.shred_version);
-        try ci.setSocket(socket_tag.GOSSIP, self.gossip);
-        try ci.setSocket(socket_tag.TURBINE_RECV, self.turbine_recv);
-        try ci.setSocket(socket_tag.TURBINE_RECV_QUIC, self.turbine_recv_quic);
-        try ci.setSocket(socket_tag.REPAIR, self.repair);
-        try ci.setSocket(socket_tag.TPU, self.tpu);
-        try ci.setSocket(socket_tag.TPU_FORWARDS, self.tpu_forwards);
-        try ci.setSocket(socket_tag.TPU_VOTE, self.tpu_vote);
-        try ci.setSocket(socket_tag.RPC, self.rpc);
-        try ci.setSocket(socket_tag.RPC_PUBSUB, self.rpc_pubsub);
-        try ci.setSocket(socket_tag.SERVE_REPAIR, self.serve_repair);
+        try ci.setSocket(.gossip, self.gossip);
+        try ci.setSocket(.turbine_recv, self.turbine_recv);
+        try ci.setSocket(.turbine_recv_quic, self.turbine_recv_quic);
+        try ci.setSocket(.repair, self.repair);
+        try ci.setSocket(.tpu, self.tpu);
+        try ci.setSocket(.tpu_forwards, self.tpu_forwards);
+        try ci.setSocket(.tpu_vote, self.tpu_vote);
+        try ci.setSocket(.rpc, self.rpc);
+        try ci.setSocket(.rpc_pubsub, self.rpc_pubsub);
+        try ci.setSocket(.serve_repair, self.serve_repair);
         return ci;
     }
 
     pub fn fromContactInfo(ci: *const ContactInfo) LegacyContactInfo {
         return .{
             .id = ci.pubkey,
-            .gossip = ci.getSocket(socket_tag.GOSSIP) orelse SocketAddr.UNSPECIFIED,
-            .turbine_recv = ci.getSocket(socket_tag.TURBINE_RECV) orelse SocketAddr.UNSPECIFIED,
-            .turbine_recv_quic = ci.getSocket(socket_tag.TURBINE_RECV_QUIC) orelse SocketAddr.UNSPECIFIED,
-            .repair = ci.getSocket(socket_tag.REPAIR) orelse SocketAddr.UNSPECIFIED,
-            .tpu = ci.getSocket(socket_tag.TPU) orelse SocketAddr.UNSPECIFIED,
-            .tpu_forwards = ci.getSocket(socket_tag.TPU_FORWARDS) orelse SocketAddr.UNSPECIFIED,
-            .tpu_vote = ci.getSocket(socket_tag.TPU_VOTE) orelse SocketAddr.UNSPECIFIED,
-            .rpc = ci.getSocket(socket_tag.RPC) orelse SocketAddr.UNSPECIFIED,
-            .rpc_pubsub = ci.getSocket(socket_tag.RPC_PUBSUB) orelse SocketAddr.UNSPECIFIED,
-            .serve_repair = ci.getSocket(socket_tag.SERVE_REPAIR) orelse SocketAddr.UNSPECIFIED,
+            .gossip = ci.getSocket(.gossip) orelse SocketAddr.UNSPECIFIED,
+            .turbine_recv = ci.getSocket(.turbine_recv) orelse SocketAddr.UNSPECIFIED,
+            .turbine_recv_quic = ci.getSocket(.turbine_recv_quic) orelse SocketAddr.UNSPECIFIED,
+            .repair = ci.getSocket(.repair) orelse SocketAddr.UNSPECIFIED,
+            .tpu = ci.getSocket(.tpu) orelse SocketAddr.UNSPECIFIED,
+            .tpu_forwards = ci.getSocket(.tpu_forwards) orelse SocketAddr.UNSPECIFIED,
+            .tpu_vote = ci.getSocket(.tpu_vote) orelse SocketAddr.UNSPECIFIED,
+            .rpc = ci.getSocket(.rpc) orelse SocketAddr.UNSPECIFIED,
+            .rpc_pubsub = ci.getSocket(.rpc_pubsub) orelse SocketAddr.UNSPECIFIED,
+            .serve_repair = ci.getSocket(.serve_repair) orelse SocketAddr.UNSPECIFIED,
             .wallclock = ci.wallclock,
             .shred_version = ci.shred_version,
         };
@@ -977,26 +977,26 @@ pub const SnapshotHashes = struct {
     }
 };
 
-pub const socket_tag = enum(u8) {
-    GOSSIP = 0,
-    REPAIR = 1,
-    RPC = 2,
-    RPC_PUBSUB = 3,
-    SERVE_REPAIR = 4,
-    TPU = 5,
-    TPU_FORWARDS = 6,
-    TPU_FORWARDS_QUIC = 7,
-    TPU_QUIC = 8,
-    TPU_VOTE = 9,
+pub const SocketTag = enum(u8) {
+    gossip = 0,
+    repair = 1,
+    rpc = 2,
+    rpc_pubsub = 3,
+    serve_repair = 4,
+    tpu = 5,
+    tpu_forwards = 6,
+    tpu_forwards_quic = 7,
+    tpu_quic = 8,
+    tpu_vote = 9,
     /// Analogous to [SOCKET_TAG_TVU](https://github.com/anza-xyz/agave/blob/0d34a1a160129c4293dac248e14231e9e773b4ce/gossip/src/contact_info.rs#L36)
-    TURBINE_RECV = 10,
+    turbine_recv = 10,
     /// Analogous to [SOCKET_TAG_TVU_QUIC](https://github.com/anza-xyz/agave/blob/0d34a1a160129c4293dac248e14231e9e773b4ce/gossip/src/contact_info.rs#L37)
-    TURBINE_RECV_QUIC = 11,
+    turbine_recv_quic = 11,
     _,
 
     pub const BincodeSize = u8;
 };
-pub const SOCKET_CACHE_SIZE: usize = @intFromEnum(socket_tag.TURBINE_RECV_QUIC) + 1;
+pub const SOCKET_CACHE_SIZE: usize = @intFromEnum(SocketTag.turbine_recv_quic) + 1;
 
 pub const ContactInfo = struct {
     pubkey: Pubkey,
@@ -1029,7 +1029,7 @@ pub const ContactInfo = struct {
 
     pub fn initSpy(allocator: std.mem.Allocator, id: Pubkey, gossip_socket_addr: SocketAddr, shred_version: u16) !Self {
         var contact_info = Self.init(allocator, id, @intCast(std.time.microTimestamp()), shred_version);
-        try contact_info.setSocket(socket_tag.GOSSIP, gossip_socket_addr);
+        try contact_info.setSocket(.gossip, gossip_socket_addr);
         return contact_info;
     }
 
@@ -1068,7 +1068,7 @@ pub const ContactInfo = struct {
         }
 
         for (0..6) |_| {
-            sockets.append(.{ .key = .TURBINE_RECV, .index = 20, .offset = 30 }) catch unreachable;
+            sockets.append(.{ .key = .turbine_recv, .index = 20, .offset = 30 }) catch unreachable;
         }
 
         return ContactInfo{
@@ -1087,15 +1087,15 @@ pub const ContactInfo = struct {
         try sanitizeWallclock(self.wallclock);
     }
 
-    pub fn getSocket(self: *const Self, key: socket_tag) ?SocketAddr {
+    pub fn getSocket(self: *const Self, key: SocketTag) ?SocketAddr {
         const socket = &self.cache[@intFromEnum(key)];
         if (socket.eql(&SocketAddr.UNSPECIFIED)) {
             return null;
         }
         return socket.*;
     }
 
-    pub fn setSocket(self: *Self, key: socket_tag, socket_addr: SocketAddr) !void {
+    pub fn setSocket(self: *Self, key: SocketTag, socket_addr: SocketAddr) !void {
         self.removeSocket(key);
 
         const offset: u16, const index: ?usize = blk: {
@@ -1122,7 +1122,7 @@ pub const ContactInfo = struct {
         self.cache[@intFromEnum(key)] = socket_addr;
     }
 
-    pub fn removeSocket(self: *Self, key: socket_tag) void {
+    pub fn removeSocket(self: *Self, key: SocketTag) void {
         // find existing socket index
         const existing_socket_index = for (self.sockets.items, 0..) |socket, idx| {
             if (socket.key == key) break idx;
@@ -1222,7 +1222,7 @@ const Sockets = struct {
 
 pub const SocketEntry = struct {
     /// GossipMessageIdentifier, e.g. turbine_recv, tpu, etc
-    key: socket_tag,
+    key: SocketTag,
     /// IpAddr index in the accompanying addrs vector.
     index: u8,
     /// Port offset with respect to the previous entry.
@@ -1320,12 +1320,12 @@ test "gossip.data: set & get socket on contact info" {
 
     var ci = ContactInfo.init(testing.allocator, Pubkey.random(rng), @as(u64, @intCast(std.time.microTimestamp())), 0);
     defer ci.deinit();
-    try ci.setSocket(socket_tag.RPC, SocketAddr.initIpv4(.{ 127, 0, 0, 1 }, 8899));
+    try ci.setSocket(.rpc, SocketAddr.initIpv4(.{ 127, 0, 0, 1 }, 8899));
 
-    var set_socket = ci.getSocket(socket_tag.RPC);
+    var set_socket = ci.getSocket(.rpc);
     try testing.expect(set_socket.?.eql(&SocketAddr.initIpv4(.{ 127, 0, 0, 1 }, 8899)));
     try testing.expect(ci.addrs.items[0].eql(&IpAddr.newIpv4(127, 0, 0, 1)));
-    try testing.expect(ci.sockets.items[0].eql(&.{ .key = .RPC, .index = 0, .offset = 8899 }));
+    try testing.expect(ci.sockets.items[0].eql(&.{ .key = .rpc, .index = 0, .offset = 8899 }));
 }
 
 test "gossip.data: contact info bincode serialize matches rust bincode" {
@@ -1386,8 +1386,8 @@ test "gossip.data: ContactInfo bincode roundtrip maintains data integrity" {
 test "gossip.data: SocketEntry serializer works" {
     testing.log_level = .debug;
 
-    comptime std.debug.assert(@intFromEnum(socket_tag.RPC_PUBSUB) == 3);
-    const se: SocketEntry = .{ .key = .RPC_PUBSUB, .index = 3, .offset = 30304 };
+    comptime std.debug.assert(@intFromEnum(SocketTag.rpc_pubsub) == 3);
+    const se: SocketEntry = .{ .key = .rpc_pubsub, .index = 3, .offset = 30304 };
 
     var buf = std.ArrayList(u8).init(testing.allocator);
     defer buf.deinit();

src/gossip/fuzz.zig

@@ -14,7 +14,6 @@ const _gossip_data = @import("data.zig");
 const LegacyContactInfo = _gossip_data.LegacyContactInfo;
 const SignedGossipData = _gossip_data.SignedGossipData;
 const ContactInfo = _gossip_data.ContactInfo;
-const socket_tag = _gossip_data.socket_tag;
 const AtomicBool = std.atomic.Value(bool);
 
 const SocketAddr = @import("../net/net.zig").SocketAddr;
@@ -289,7 +288,7 @@ pub fn run() !void {
 
     const fuzz_pubkey = Pubkey.fromPublicKey(&fuzz_keypair.public_key);
     var fuzz_contact_info = ContactInfo.init(allocator, fuzz_pubkey, 0, 19);
-    try fuzz_contact_info.setSocket(socket_tag.GOSSIP, fuzz_address);
+    try fuzz_contact_info.setSocket(.gossip, fuzz_address);
 
     var fuzz_exit = AtomicBool.init(false);
     var gossip_service_fuzzer = try GossipService.init(

src/gossip/lib.zig

@@ -19,4 +19,4 @@ pub const Ping = ping_pong.Ping;
 pub const Pong = ping_pong.Pong;
 
 pub const getWallclockMs = data.getWallclockMs;
-pub const socket_tag = data.socket_tag;
+pub const SocketTag = data.SocketTag;

src/gossip/ping_pong.zig

@@ -4,7 +4,6 @@ const Hash = @import("../core/hash.zig").Hash;
 const Signature = @import("../core/signature.zig").Signature;
 const _gossip_data = @import("data.zig");
 const ContactInfo = _gossip_data.ContactInfo;
-const socket_tag = _gossip_data.socket_tag;
 const getWallclockMs = _gossip_data.getWallclockMs;
 
 const DefaultPrng = std.rand.DefaultPrng;
@@ -222,7 +221,7 @@ pub const PingCache = struct {
         var pings = std.ArrayList(PingAndSocketAddr).init(allocator);
 
         for (peers, 0..) |*peer, i| {
-            if (peer.getSocket(socket_tag.GOSSIP)) |gossip_addr| {
+            if (peer.getSocket(.gossip)) |gossip_addr| {
                 const result = self.check(now, PubkeyAndSocketAddr{ .pubkey = peer.pubkey, .socket_addr = gossip_addr }, &our_keypair);
                 if (result.passes_ping_check) {
                     try valid_peers.append(i);