// SPDX-License-Identifier: BSD-2-Clause
// -*- Java -*-
//
// Copyright (c) 2005, Matthew J. Rutherford <rutherfo@cs.colorado.edu>
// Copyright (c) 2005, University of Colorado at Boulder
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
// * Redistributions of source code must retain the above copyright
//   notice, this list of conditions and the following disclaimer.
//
// * Redistributions in binary form must reproduce the above copyright
//   notice, this list of conditions and the following disclaimer in the
//   documentation and/or other materials provided with the distribution.
//
// * Neither the name of the University of Colorado at Boulder nor the
//   names of its contributors may be used to endorse or promote
//   products derived from this software without specific prior written
//   permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
package org.xbill.DNS;

import static org.junit.jupiter.api.Assertions.assertArrayEquals;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;

import java.io.IOException;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.List;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.xbill.DNS.APLRecord.Element;

public class APLRecordTest {
  static class Test_Element_init {
    InetAddress m_addr4;
    InetAddress m_addr6;

    @BeforeEach
    void setUp() throws UnknownHostException {
      m_addr4 = InetAddress.getByName("193.160.232.5");
      m_addr6 = InetAddress.getByName("2001:db8:85a3:8d3:1319:8a2e:370:7334");
    }

    @Test
    void valid_IPv4() {
      Element el = new Element(true, m_addr4, 16);
      assertEquals(Address.IPv4, el.family);
      assertTrue(el.negative);
      assertEquals(m_addr4, el.address);
      assertEquals(16, el.prefixLength);
    }

    @Test
    void invalid_IPv4() {
      assertThrows(IllegalArgumentException.class, () -> new Element(true, m_addr4, 33));
    }

    @Test
    void valid_IPv6() {
      Element el = new Element(false, m_addr6, 74);
      assertEquals(Address.IPv6, el.family);
      assertFalse(el.negative);
      assertEquals(m_addr6, el.address);
      assertEquals(74, el.prefixLength);
    }

    @Test
    void invalid_IPv6() {
      assertThrows(IllegalArgumentException.class, () -> new Element(true, m_addr6, 129));
    }
  }

  static class Test_init {
    Name m_an, m_rn;
    long m_ttl;
    ArrayList<Element> m_elements;
    InetAddress m_addr4;
    String m_addr4_string;
    byte[] m_addr4_bytes;
    InetAddress m_addr6;
    String m_addr6_string;
    byte[] m_addr6_bytes;

    @BeforeEach
    void setUp() throws TextParseException, UnknownHostException {
      m_an = Name.fromString("My.Absolute.Name.");
      m_rn = Name.fromString("My.Relative.Name");
      m_ttl = 0x13579;
      m_addr4_string = "193.160.232.5";
      m_addr4 = InetAddress.getByName(m_addr4_string);
      m_addr4_bytes = m_addr4.getAddress();

      m_addr6_string = "2001:db8:85a3:8d3:1319:8a2e:370:7334";
      m_addr6 = InetAddress.getByName(m_addr6_string);
      m_addr6_bytes = m_addr6.getAddress();

      m_elements = new ArrayList<>(2);
      Element e = new Element(true, m_addr4, 12);
      m_elements.add(e);

      e = new Element(false, m_addr6, 64);
      m_elements.add(e);
    }

    @Test
    void ctor_0arg() {
      APLRecord ar = new APLRecord();
      assertNull(ar.getName());
      assertEquals(0, ar.getType());
      assertEquals(0, ar.getDClass());
      assertEquals(0, ar.getTTL());
      assertNull(ar.getElements());
    }

    @Test
    void ctor_4arg_basic() {
      APLRecord ar = new APLRecord(m_an, DClass.IN, m_ttl, m_elements);
      assertEquals(m_an, ar.getName());
      assertEquals(Type.APL, ar.getType());
      assertEquals(DClass.IN, ar.getDClass());
      assertEquals(m_ttl, ar.getTTL());
      assertEquals(m_elements, ar.getElements());
    }

    @Test
    void ctor_4arg_empty_elements() {
      APLRecord ar = new APLRecord(m_an, DClass.IN, m_ttl, new ArrayList<>());
      assertEquals(new ArrayList<>(), ar.getElements());
    }

    @Test
    void ctor_4arg_relative_name() {
      assertThrows(
          RelativeNameException.class, () -> new APLRecord(m_rn, DClass.IN, m_ttl, m_elements));
    }
  }

  static class Test_rrFromWire {
    InetAddress m_addr4;
    byte[] m_addr4_bytes;
    InetAddress m_addr6;
    byte[] m_addr6_bytes;

    @BeforeEach
    void setUp() throws UnknownHostException {
      m_addr4 = InetAddress.getByName("193.160.232.5");
      m_addr4_bytes = m_addr4.getAddress();

      m_addr6 = InetAddress.getByName("2001:db8:85a3:8d3:1319:8a2e:370:7334");
      m_addr6_bytes = m_addr6.getAddress();
    }

    @Test
    void validIPv4() throws IOException {
      byte[] raw =
          new byte[] {
            0,
            1,
            8,
            (byte) 0x84,
            m_addr4_bytes[0],
            m_addr4_bytes[1],
            m_addr4_bytes[2],
            m_addr4_bytes[3]
          };

      DNSInput di = new DNSInput(raw);
      APLRecord ar = new APLRecord();
      ar.rrFromWire(di);

      ArrayList<Element> exp = new ArrayList<>();
      exp.add(new Element(true, m_addr4, 8));
      assertEquals(exp, ar.getElements());
    }

    @Test
    void validIPv4_short_address() throws IOException {
      byte[] raw =
          new byte[] {0, 1, 20, (byte) 0x83, m_addr4_bytes[0], m_addr4_bytes[1], m_addr4_bytes[2]};

      DNSInput di = new DNSInput(raw);
      APLRecord ar = new APLRecord();
      ar.rrFromWire(di);

      InetAddress a = InetAddress.getByName("193.160.232.0");

      ArrayList<Element> exp = new ArrayList<>();
      exp.add(new Element(true, a, 20));
      assertEquals(exp, ar.getElements());
    }

    @Test
    void invalid_IPv4_prefix() {
      byte[] raw =
          new byte[] {
            0,
            1,
            33,
            (byte) 0x84,
            m_addr4_bytes[0],
            m_addr4_bytes[1],
            m_addr4_bytes[2],
            m_addr4_bytes[3]
          };

      DNSInput di = new DNSInput(raw);
      APLRecord ar = new APLRecord();
      assertThrows(WireParseException.class, () -> ar.rrFromWire(di));
    }

    @Test
    void invalid_IPv4_length() {
      byte[] raw =
          new byte[] {
            0,
            1,
            8,
            (byte) 0x85,
            m_addr4_bytes[0],
            m_addr4_bytes[1],
            m_addr4_bytes[2],
            m_addr4_bytes[3],
            10
          };

      DNSInput di = new DNSInput(raw);
      APLRecord ar = new APLRecord();
      assertThrows(WireParseException.class, () -> ar.rrFromWire(di));
    }

    @Test
    void multiple_validIPv4() throws IOException {
      byte[] raw =
          new byte[] {
            0,
            1,
            8,
            (byte) 0x84,
            m_addr4_bytes[0],
            m_addr4_bytes[1],
            m_addr4_bytes[2],
            m_addr4_bytes[3],
            0,
            1,
            30,
            (byte) 0x4,
            m_addr4_bytes[0],
            m_addr4_bytes[1],
            m_addr4_bytes[2],
            m_addr4_bytes[3],
          };

      DNSInput di = new DNSInput(raw);
      APLRecord ar = new APLRecord();
      ar.rrFromWire(di);

      ArrayList<Element> exp = new ArrayList<>();
      exp.add(new Element(true, m_addr4, 8));
      exp.add(new Element(false, m_addr4, 30));
      assertEquals(exp, ar.getElements());
    }

    @Test
    void validIPv6() throws IOException {
      byte[] raw =
          new byte[] {
            0,
            2,
            (byte) 115,
            (byte) 0x10,
            m_addr6_bytes[0],
            m_addr6_bytes[1],
            m_addr6_bytes[2],
            m_addr6_bytes[3],
            m_addr6_bytes[4],
            m_addr6_bytes[5],
            m_addr6_bytes[6],
            m_addr6_bytes[7],
            m_addr6_bytes[8],
            m_addr6_bytes[9],
            m_addr6_bytes[10],
            m_addr6_bytes[11],
            m_addr6_bytes[12],
            m_addr6_bytes[13],
            m_addr6_bytes[14],
            m_addr6_bytes[15]
          };

      DNSInput di = new DNSInput(raw);
      APLRecord ar = new APLRecord();
      ar.rrFromWire(di);

      ArrayList<Element> exp = new ArrayList<>();
      exp.add(new Element(false, m_addr6, 115));
      assertEquals(exp, ar.getElements());
    }

    @Test
    void valid_nonIP() throws IOException {
      byte[] raw = new byte[] {0, 3, (byte) 130, (byte) 0x85, 1, 2, 3, 4, 5};

      DNSInput di = new DNSInput(raw);
      APLRecord ar = new APLRecord();
      ar.rrFromWire(di);

      List<Element> l = ar.getElements();
      assertEquals(1, l.size());

      Element el = l.get(0);
      assertEquals(3, el.family);
      assertTrue(el.negative);
      assertEquals(130, el.prefixLength);
      assertArrayEquals(new byte[] {1, 2, 3, 4, 5}, (byte[]) el.address);
    }
  }

  static class Test_rdataFromString {
    InetAddress m_addr4;
    String m_addr4_string;
    byte[] m_addr4_bytes;
    InetAddress m_addr6;
    String m_addr6_string;
    byte[] m_addr6_bytes;

    @BeforeEach
    void setUp() throws UnknownHostException {
      m_addr4_string = "193.160.232.5";
      m_addr4 = InetAddress.getByName(m_addr4_string);
      m_addr4_bytes = m_addr4.getAddress();

      m_addr6_string = "2001:db8:85a3:8d3:1319:8a2e:370:7334";
      m_addr6 = InetAddress.getByName(m_addr6_string);
      m_addr6_bytes = m_addr6.getAddress();
    }

    @Test
    void validIPv4() throws IOException {
      Tokenizer t = new Tokenizer("1:" + m_addr4_string + "/11\n");
      APLRecord ar = new APLRecord();
      ar.rdataFromString(t, null);

      ArrayList<Element> exp = new ArrayList<>();
      exp.add(new Element(false, m_addr4, 11));

      assertEquals(exp, ar.getElements());

      // make sure extra token is put back
      assertEquals(Tokenizer.EOL, t.get().type);
    }

    @Test
    void valid_multi() throws IOException {
      Tokenizer t = new Tokenizer("1:" + m_addr4_string + "/11 !2:" + m_addr6_string + "/100");
      APLRecord ar = new APLRecord();
      ar.rdataFromString(t, null);

      ArrayList<Element> exp = new ArrayList<>();
      exp.add(new Element(false, m_addr4, 11));
      exp.add(new Element(true, m_addr6, 100));

      assertEquals(exp, ar.getElements());
    }

    @Test
    void validIPv6() throws IOException {
      Tokenizer t = new Tokenizer("!2:" + m_addr6_string + "/36\n");
      APLRecord ar = new APLRecord();
      ar.rdataFromString(t, null);

      ArrayList<Element> exp = new ArrayList<>();
      exp.add(new Element(true, m_addr6, 36));

      assertEquals(exp, ar.getElements());

      // make sure extra token is put back
      assertEquals(Tokenizer.EOL, t.get().type);
    }

    @Test
    void no_colon() {
      Tokenizer t = new Tokenizer("!1192.68.0.1/20");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }

    @Test
    void colon_and_slash_swapped() {
      Tokenizer t = new Tokenizer("!1/192.68.0.1:20");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }

    @Test
    void no_slash() {
      Tokenizer t = new Tokenizer("!1:192.68.0.1|20");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }

    @Test
    void empty_family() {
      Tokenizer t = new Tokenizer("!:192.68.0.1/20");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }

    @Test
    void malformed_family() {
      Tokenizer t = new Tokenizer("family:192.68.0.1/20");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }

    @Test
    void invalid_family() {
      Tokenizer t = new Tokenizer("3:192.68.0.1/20");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }

    @Test
    void empty_prefix() {
      Tokenizer t = new Tokenizer("1:192.68.0.1/");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }

    @Test
    void malformed_prefix() {
      Tokenizer t = new Tokenizer("1:192.68.0.1/prefix");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }

    @Test
    void invalid_prefix() {
      Tokenizer t = new Tokenizer("1:192.68.0.1/33");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }

    @Test
    void empty_address() {
      Tokenizer t = new Tokenizer("1:/33");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }

    @Test
    void malformed_address() {
      Tokenizer t = new Tokenizer("1:A.B.C.D/33");
      APLRecord ar = new APLRecord();
      assertThrows(TextParseException.class, () -> ar.rdataFromString(t, null));
    }
  }

  static class Test_rrToString {
    Name m_an, m_rn;
    long m_ttl;
    ArrayList<Element> m_elements;
    InetAddress m_addr4;
    String m_addr4_string;
    byte[] m_addr4_bytes;
    InetAddress m_addr6;
    String m_addr6_string;
    byte[] m_addr6_bytes;

    @BeforeEach
    void setUp() throws TextParseException, UnknownHostException {
      m_an = Name.fromString("My.Absolute.Name.");
      m_rn = Name.fromString("My.Relative.Name");
      m_ttl = 0x13579;
      m_addr4_string = "193.160.232.5";
      m_addr4 = InetAddress.getByName(m_addr4_string);
      m_addr4_bytes = m_addr4.getAddress();

      m_addr6_string = "2001:db8:85a3:8d3:1319:8a2e:370:7334";
      m_addr6 = InetAddress.getByName(m_addr6_string);
      m_addr6_bytes = m_addr6.getAddress();

      m_elements = new ArrayList<>(2);
      Element e = new Element(true, m_addr4, 12);
      m_elements.add(e);

      e = new Element(false, m_addr6, 64);
      m_elements.add(e);
    }

    @Test
    void test() {
      APLRecord ar = new APLRecord(m_an, DClass.IN, m_ttl, m_elements);
      assertEquals("!1:" + m_addr4_string + "/12 2:" + m_addr6_string + "/64", ar.rrToString());
    }
  }

  static class Test_rrToWire {
    Name m_an, m_rn;
    long m_ttl;
    ArrayList<Element> m_elements;
    InetAddress m_addr4;
    String m_addr4_string;
    byte[] m_addr4_bytes;
    InetAddress m_addr6;
    String m_addr6_string;
    byte[] m_addr6_bytes;

    @BeforeEach
    void setUp() throws TextParseException, UnknownHostException {
      m_an = Name.fromString("My.Absolute.Name.");
      m_rn = Name.fromString("My.Relative.Name");
      m_ttl = 0x13579;
      m_addr4_string = "193.160.232.5";
      m_addr4 = InetAddress.getByName(m_addr4_string);
      m_addr4_bytes = m_addr4.getAddress();

      m_addr6_string = "2001:db8:85a3:8d3:1319:8a2e:370:7334";
      m_addr6 = InetAddress.getByName(m_addr6_string);
      m_addr6_bytes = m_addr6.getAddress();

      m_elements = new ArrayList<>(2);
      Element e = new Element(true, m_addr4, 12);
      m_elements.add(e);

      e = new Element(false, m_addr6, 64);
      m_elements.add(e);
    }

    @Test
    void empty() {
      APLRecord ar = new APLRecord(m_an, DClass.IN, m_ttl, new ArrayList<>());
      DNSOutput dout = new DNSOutput();

      ar.rrToWire(dout, null, true);
      assertArrayEquals(new byte[0], dout.toByteArray());
    }

    @Test
    void basic() {
      APLRecord ar = new APLRecord(m_an, DClass.IN, m_ttl, m_elements);

      byte[] exp =
          new byte[] {
            0,
            1,
            12,
            (byte) 0x84,
            m_addr4_bytes[0],
            m_addr4_bytes[1],
            m_addr4_bytes[2],
            m_addr4_bytes[3],
            0,
            2,
            64,
            0x10,
            m_addr6_bytes[0],
            m_addr6_bytes[1],
            m_addr6_bytes[2],
            m_addr6_bytes[3],
            m_addr6_bytes[4],
            m_addr6_bytes[5],
            m_addr6_bytes[6],
            m_addr6_bytes[7],
            m_addr6_bytes[8],
            m_addr6_bytes[9],
            m_addr6_bytes[10],
            m_addr6_bytes[11],
            m_addr6_bytes[12],
            m_addr6_bytes[13],
            m_addr6_bytes[14],
            m_addr6_bytes[15]
          };

      DNSOutput dout = new DNSOutput();

      ar.rrToWire(dout, null, true);
      assertArrayEquals(exp, dout.toByteArray());
    }

    @Test
    void non_IP() throws IOException {
      byte[] exp = new byte[] {0, 3, (byte) 130, (byte) 0x85, 1, 2, 3, 4, 5};

      DNSInput di = new DNSInput(exp);
      APLRecord ar = new APLRecord();
      ar.rrFromWire(di);

      DNSOutput dout = new DNSOutput();

      ar.rrToWire(dout, null, true);
      assertArrayEquals(exp, dout.toByteArray());
    }

    @Test
    void address_with_embedded_zero() throws UnknownHostException {
      InetAddress a = InetAddress.getByName("232.0.11.1");
      ArrayList<Element> elements = new ArrayList<>();
      elements.add(new Element(true, a, 31));

      APLRecord ar = new APLRecord(m_an, DClass.IN, m_ttl, elements);

      byte[] exp = new byte[] {0, 1, 31, (byte) 0x84, (byte) 232, 0, 11, 1};

      DNSOutput dout = new DNSOutput();

      ar.rrToWire(dout, null, true);
      assertArrayEquals(exp, dout.toByteArray());
    }

    @Test
    void short_address() throws UnknownHostException {
      InetAddress a = InetAddress.getByName("232.0.11.0");
      ArrayList<Element> elements = new ArrayList<>();
      elements.add(new Element(true, a, 31));

      APLRecord ar = new APLRecord(m_an, DClass.IN, m_ttl, elements);

      byte[] exp = new byte[] {0, 1, 31, (byte) 0x83, (byte) 232, 0, 11};

      DNSOutput dout = new DNSOutput();

      ar.rrToWire(dout, null, true);
      assertArrayEquals(exp, dout.toByteArray());
    }

    @Test
    void wildcard_address() throws UnknownHostException {
      InetAddress a = InetAddress.getByName("0.0.0.0");
      ArrayList<Element> elements = new ArrayList<>();
      elements.add(new Element(true, a, 31));

      APLRecord ar = new APLRecord(m_an, DClass.IN, m_ttl, elements);

      byte[] exp = new byte[] {0, 1, 31, (byte) 0x80};

      DNSOutput dout = new DNSOutput();

      ar.rrToWire(dout, null, true);
      assertArrayEquals(exp, dout.toByteArray());
    }
  }
}
