licenses = ["notice"],
)

filegroup(
    name = "safe_ptr_hdr",
    srcs = ["safe_ptr.h"],
    visibility = [
        "//tensorflow:internal",
    ],
)

cc_library(
    name = "safe_ptr",
    srcs = [
        "safe_ptr.cc",
        "//tensorflow/c/eager:headers",
    ],
    hdrs = ["safe_ptr.h"],
    visibility = [

# filegroup(
#     name = "copybara_library",
#     srcs = glob(
#         [
#             "*.bara.sky",
#             "*.dic",
#             "*.ftrcp",
#         ],
#         exclude = [
#             "copy.bara.sky",
#         ],
#     ),
#     visibility = ["//visibility:public"],
# )
#
# filegroup(
#     name = "copybara_config",
#     srcs = ["copy.bara.sky"],
#     data = [

# TODO(ybaturina): Remove once TF is migrated to CUDA 12.9.
genrule(
    name = "nvidia_constraints",
    srcs = ["nvidia-requirements.txt"],
    outs = ["nvidia-constraints.txt"],
    cmd = """sed -E "s/>=/==/" $(location nvidia-requirements.txt) > $@;""",
)

compile_pip_requirements(
    name = "requirements",
    srcs = [
        "nvidia-requirements.txt",
        "requirements.in",
        ":nvidia_constraints",
    ],

        byte[] src = new byte[100];

        // Encode "test.example.com"
        src[0] = 4; // Length of "test"
        src[1] = 't';
        src[2] = 'e';
        src[3] = 's';
        src[4] = 't';
        src[5] = 7; // Length of "example"
        src[6] = 'e';
        src[7] = 'x';
        src[8] = 'a';
        src[9] = 'm';
        src[10] = 'p';
        src[11] = 'l';

        byte[] src = new byte[20];
        src[0] = (byte) 0xC0;
        src[1] = (byte) 0x0C;
        NameServicePacket.writeInt2(NameServicePacket.A, src, 2); // recordType
        NameServicePacket.writeInt2(NameServicePacket.IN, src, 4); // recordClass
        NameServicePacket.writeInt4(100, src, 6); // ttl
        NameServicePacket.writeInt2(6, src, 10); // rDataLength (for one NbtAddress)

     *
     * @param src the source byte array
     * @param si the starting index in the source array
     * @return the decoded integer value
     */
    public static int dec_uint32be(final byte[] src, final int si) {
        return (src[si] & 0xFF) << 24 | (src[si + 1] & 0xFF) << 16 | (src[si + 2] & 0xFF) << 8 | src[si + 3] & 0xFF;
    }

    /**
     * Decodes a 16-bit unsigned integer from little-endian byte order.

        src[srcIndex + 5] = 0x01;
        // TTL (4 bytes)
        src[srcIndex + 6] = 0x00;
        src[srcIndex + 7] = 0x00;
        src[srcIndex + 8] = 0x00;
        src[srcIndex + 9] = 0x00;
        // RData length
        src[srcIndex + 10] = 0x00;
        src[srcIndex + 11] = 0x1F; // 31 bytes: 1 + 18 + 6 + 6
        // Number of names (1)
        src[srcIndex + 12] = 0x01;
        // Name entry (18 bytes)

    }

    static int readULong(final byte[] src, final int index) {
        return src[index] & 0xff | (src[index + 1] & 0xff) << 8 | (src[index + 2] & 0xff) << 16 | (src[index + 3] & 0xff) << 24;
    }

    static int readUShort(final byte[] src, final int index) {
        return src[index] & 0xff | (src[index + 1] & 0xff) << 8;
    }

    static byte[] readSecurityBuffer(final byte[] src, final int index) {

        assertEquals(0xFFFFFFFF, SMBUtil.readInt4(src, 4));
        assertEquals(0x00000000, SMBUtil.readInt4(src, 8));
        assertEquals(0x89ABCDEF, SMBUtil.readInt4(src, 12));
    }

    @Test
    void testReadInt8() {
        byte[] src = new byte[] {
                // 0x123456789ABCDEF0L

        byte[] src = new byte[200];
        // Setup minimal valid header
        src[0] = 0x00;
        src[1] = 0x01; // Transaction ID
        src[2] = 0x00; // Flags
        src[3] = 0x00; // Flags
        src[4] = 0x00;
        src[5] = 0x01; // Question count

        // Act
        int result = nodeStatusRequest.readWireFormat(src, 0);

        // Assert