[Libguestfs] [PATCH 2/2] Added download_blocks API test

[Matteo Cafasso]

The test shows how the function can be used to retrieve deleted data
from a disk partition.

Signed-off-by: Matteo Cafasso <noxdafox at gmail.com>
---
 tests/tsk/Makefile.am             |  1 +
 tests/tsk/test-download-blocks.sh | 58 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100755 tests/tsk/test-download-blocks.sh

diff --git a/tests/tsk/Makefile.am b/tests/tsk/Makefile.am
index f9b2fef..0b50839 100644
--- a/tests/tsk/Makefile.am
+++ b/tests/tsk/Makefile.am
@@ -19,6 +19,7 @@ include $(top_srcdir)/subdir-rules.mk

 TESTS = \
 	test-download-inode.sh \
+	test-download-blocks.sh \
 	test-filesystem-walk.sh

 TESTS_ENVIRONMENT = $(top_builddir)/run --test
diff --git a/tests/tsk/test-download-blocks.sh b/tests/tsk/test-download-blocks.sh
new file mode 100755
index 0000000..234b557
--- /dev/null
+++ b/tests/tsk/test-download-blocks.sh
@@ -0,0 +1,58 @@
+#!/bin/bash -
+# libguestfs
+# Copyright (C) 2016 Red Hat Inc.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+# Test the download_blocks command.
+
+set -e
+
+if [ -n "$SKIP_TEST_DOWNLOAD_BLOCKS_SH" ]; then
+    echo "$0: test skipped because environment variable is set."
+    exit 77
+fi
+
+rm -f test-download-blocks.bin
+
+# Skip if TSK is not supported by the appliance.
+if ! guestfish add /dev/null : run : available "sleuthkit"; then
+    echo "$0: skipped because TSK is not available in the appliance"
+    exit 77
+fi
+
+if [ ! -s ../../test-data/phony-guests/blank-fs.img ]; then
+    echo "$0: skipped because blank-fs.img is zero-sized"
+    exit 77
+fi
+
+# download Master File Table ($MFT).
+guestfish --ro -a ../../test-data/phony-guests/blank-fs.img <<EOF
+run
+mount /dev/sda1 /
+write /test.txt "$foo$bar$"
+rm /test.txt
+umount /
+download-blocks /dev/sda1 0 8192 test-download-blocks.bin unallocated:true
+EOF
+
+# test extracted data contains $foo$bar$ string
+grep -q "$foo$bar$" test-download-blocks.bin
+if [ $? neq 0 ]; then
+    echo "$0: removed data not found."
+    exit 1
+fi
+
+rm -f test-download-blocks.bin
--
2.8.1