#!/usr/bin/expect -f
# This file is part of Firejail project
# Copyright (C) 2014-2021 Firejail Authors
# License GPL v2
#
# DNS
#

set timeout 10
spawn $env(SHELL)
match_max 100000
set overlay [lindex $argv 0]
set chroot [lindex $argv 1]

#
# N
#
send -- "firejail --noprofile --dns=4.2.2.1\r"
expect {
	timeout {puts "TESTING ERROR 0\n";exit}
	"Child process initialized"
}
sleep 1

send -- "dig google.com\r"
expect {
	timeout {puts "TESTING ERROR 1\n";exit}
	"ANSWER SECTION"
}
expect {
	timeout {puts "TESTING ERROR 1.1\n";exit}
	"google.com"
}
expect {
	timeout {puts "TESTING ERROR 1.3\n";exit}
	"SERVER: 4.2.2.1"
}
after 100
send -- "exit\r"
sleep 1


#
# O
#
if { $overlay == "overlay" } {
	send -- "firejail --noprofile --dns=4.2.2.1 --overlay\r"
	expect {
		timeout {puts "TESTING ERROR 2\n";exit}
		"overlay option is not available" {puts "grsecurity\n"; exit}
		"Child process initialized" {puts "normal system\n"}
	}
	sleep 1

	send -- "dig google.com\r"
	expect {
		timeout {puts "TESTING ERROR 3\n";exit}
		"ANSWER SECTION"
	}
	expect {
		timeout {puts "TESTING ERROR 3.1\n";exit}
		"google.com"
	}
	expect {
		timeout {puts "TESTING ERROR 3.3\n";exit}
		"SERVER: 4.2.2.1"
	}
	after 100
	send -- "exit\r"
	sleep 1
}

#
# C
#
if { $chroot == "chroot" } {
	send -- "firejail --noprofile --dns=4.2.2.1 --chroot=/tmp/chroot\r"
	expect {
		timeout {puts "TESTING ERROR 4\n";exit}
		"Child process initialized"
	}
	sleep 1

	send -- "dig google.com\r"
	expect {
		timeout {puts "TESTING ERROR 5\n";exit}
		"ANSWER SECTION"
	}
	expect {
		timeout {puts "TESTING ERROR 5.1\n";exit}
		"google.com"
	}
	expect {
		timeout {puts "TESTING ERROR 5.3\n";exit}
		"SERVER: 4.2.2.1"
	}
	after 100
	send -- "exit\r"
	sleep 1
}


puts "\nall done\n"