@Before
public void before() throws Exception {
// force the xn field in the request
Random random = mock(Random.class);
Mockito.when(random.nextInt(0xffff)).thenReturn(65535);
whenNew(Random.class).withNoArguments().thenReturn(random);
mockStatic(System.class);
when(System.currentTimeMillis()).thenReturn(144140678000L).thenReturn(144140678345L);
when(System.nanoTime()).thenReturn(100000000L, 100000000L + 345123000L);
mockStatic(Logger.class);
when(Logger.getLogger("com.comcast.cdn.traffic_control.traffic_router.core.access"))
.thenReturn(accessLogger);
header = new Header();
header.setID(65535);
header.setFlag(Flags.QR);
client = Inet4Address.getByAddress(new byte[] {(byte) 192, (byte) 168, 23, 45});
nameServer = mock(NameServer.class);
}
public static void main(String[] args) throws Exception {
ZoneTransferIn xfrin;
TSIG key = null;
int ixfr_serial = -1;
String server = null;
int port = SimpleResolver.DEFAULT_PORT;
boolean fallback = false;
Name zname;
int arg = 0;
while (arg < args.length) {
if (args[arg].equals("-i")) {
ixfr_serial = Integer.parseInt(args[++arg]);
if (ixfr_serial < 0) usage("invalid serial number");
} else if (args[arg].equals("-k")) {
String s = args[++arg];
int index = s.indexOf('/');
if (index < 0) usage("invalid key");
key = new TSIG(s.substring(0, index), s.substring(index + 1));
} else if (args[arg].equals("-s")) {
server = args[++arg];
} else if (args[arg].equals("-p")) {
port = Integer.parseInt(args[++arg]);
if (port < 0 || port > 0xFFFF) usage("invalid port");
} else if (args[arg].equals("-f")) {
fallback = true;
} else if (args[arg].startsWith("-")) {
usage("invalid option");
} else {
break;
}
arg++;
}
if (arg >= args.length) usage("no zone name specified");
zname = Name.fromString(args[arg]);
if (server == null) {
Lookup l = new Lookup(zname, Type.NS);
Record[] ns = l.run();
if (ns == null) {
System.out.println("failed to look up NS record: " + l.getErrorString());
System.exit(1);
}
server = ns[0].rdataToString();
System.out.println("sending to server '" + server + "'");
}
if (ixfr_serial >= 0)
xfrin = ZoneTransferIn.newIXFR(zname, ixfr_serial, fallback, server, port, key);
else xfrin = ZoneTransferIn.newAXFR(zname, server, port, key);
List response = xfrin.run();
if (xfrin.isAXFR()) {
if (ixfr_serial >= 0) System.out.println("AXFR-like IXFR response");
else System.out.println("AXFR response");
Iterator it = response.iterator();
while (it.hasNext()) System.out.println(it.next());
} else if (xfrin.isIXFR()) {
System.out.println("IXFR response");
Iterator it = response.iterator();
while (it.hasNext()) {
ZoneTransferIn.Delta delta;
delta = (ZoneTransferIn.Delta) it.next();
System.out.println("delta from " + delta.start + " to " + delta.end);
System.out.println("deletes");
Iterator it2 = delta.deletes.iterator();
while (it2.hasNext()) System.out.println(it2.next());
System.out.println("adds");
it2 = delta.adds.iterator();
while (it2.hasNext()) System.out.println(it2.next());
}
} else if (xfrin.isCurrent()) {
System.out.println("up to date");
}
}
private static void usage(String s) {
System.out.println("Error: " + s);
System.out.println(
"usage: xfrin [-i serial] [-k keyname/secret] " + "[-s server] [-p port] [-f] zone");
System.exit(1);
}
