1 files changed, 119 insertions, 0 deletions
diff --git a/test/javaTests/src/org/criu/java/tests/ReadWrite.java b/test/javaTests/src/org/criu/java/tests/ReadWrite.java
new file mode 100644
index 000000000..fa98447ed
--- /dev/null
+++ b/test/javaTests/src/org/criu/java/tests/ReadWrite.java
@@ -0,0 +1,119 @@
+package org.criu.java.tests;
+
+import java.io.*;
+import java.lang.management.ManagementFactory;
+import java.lang.management.RuntimeMXBean;
+import java.nio.MappedByteBuffer;
+import java.nio.channels.FileChannel;
+import java.nio.channels.FileChannel.MapMode;
+import java.nio.file.StandardOpenOption;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+class ReadWrite {
+	private static String TESTNAME = "ReadWrite";
+
+	/**
+	 * Read from a file and write its content into another file,
+	 * checkpointing and restoring in between.
+	 *
+	 * @param args Not used.
+	 */
+	public static void main(String[] args) {
+		int i = 0;
+		String s, pid;
+		boolean similar;
+		MappedByteBuffer b = null;
+		Logger logger = null;
+		try {
+			File readFile = new File(Helper.SOURCE_FOLDER + "/" + "FileRead.java");
+			File writeFile = new File(Helper.OUTPUT_FOLDER_NAME + "/" + TESTNAME + "/" + "ReadWrite_file.txt");
+			logger = Logger.getLogger(Helper.PACKAGE_NAME + "." + TESTNAME);
+			File f = new File(Helper.MEMORY_MAPPED_FILE_NAME);
+			RuntimeMXBean bean = ManagementFactory.getRuntimeMXBean();
+			pid = bean.getName();
+			int val = Helper.init(TESTNAME, pid, logger);
+			if (0 != val) {
+				logger.log(Level.SEVERE, "Helper.init returned a non-zero code.");
+				b.putChar(Helper.MAPPED_INDEX, Helper.STATE_END);
+				System.exit(1);
+			}
+			logger.log(Level.INFO, "Test init done; pid written to pid file; beginning with test");
+			FileChannel channel = FileChannel.open(f.toPath(), StandardOpenOption.READ, StandardOpenOption.WRITE, StandardOpenOption.CREATE);
+			b = channel.map(MapMode.READ_WRITE, 0, Helper.MAPPED_REGION_SIZE);
+			channel.close();
+			/*
+			 * Mapped Byte Buffer should be in init state at the beginning of test
+			 */
+			if (Helper.STATE_INIT != b.getChar(Helper.MAPPED_INDEX)) {
+				logger.log(Level.SEVERE, "Error: Error in memory mapping, test is not in init state");
+				b.putChar(Helper.MAPPED_INDEX, Helper.STATE_END);
+				System.exit(1);
+			}
+			logger.log(Level.INFO, "Checking existence of files to be read!");
+			if (!readFile.exists()) {
+				logger.log(Level.SEVERE, "Error: File from which to read does not exist");
+				b.putChar(Helper.MAPPED_INDEX, Helper.STATE_END);
+				System.exit(1);
+			}
+
+			if (writeFile.exists()) {
+				writeFile.delete();
+			}
+			logger.log(Level.INFO, "Creating the writeFile");
+			boolean newFile = writeFile.createNewFile();
+			if (!newFile) {
+				logger.log(Level.SEVERE, "Error: Cannot create a new file to write to.");
+				b.putChar(Helper.MAPPED_INDEX, Helper.STATE_END);
+				System.exit(1);
+			}
+
+			BufferedReader brr = new BufferedReader(new FileReader(readFile));
+			BufferedWriter brw = new BufferedWriter(new FileWriter(writeFile));
+			logger.log(Level.INFO, "Start writing");
+
+			s = brr.readLine();
+
+			while (null != s) {
+				i++;
+				brw.write(s + "\n");
+
+				if (50 == i) {
+					/*
+					 * Checkpoint and Restore
+					 */
+					logger.log(Level.INFO, "Going to checkpoint");
+					Helper.checkpointAndWait(b, logger);
+					logger.log(Level.INFO, "Test has been restored!");
+				}
+				s = brr.readLine();
+			}
+
+			brw.flush();
+			logger.log(Level.INFO, "Checking content of the files.");
+			similar = Helper.compare(readFile, writeFile);
+
+			if (!similar) {
+				logger.log(Level.SEVERE, "Error: Files are not similar after writing");
+				b.putChar(Helper.MAPPED_INDEX, Helper.STATE_FAIL);
+				System.exit(1);
+			}
+			logger.log(Level.INFO, "Content of file is as expected");
+			logger.log(Level.INFO, Helper.PASS_MESSAGE);
+			b.putChar(Helper.MAPPED_INDEX, Helper.STATE_PASS);
+			System.exit(0);
+		} catch (Exception e) {
+			if (null != logger) {
+				StringWriter writer = new StringWriter();
+				PrintWriter printWriter = new PrintWriter(writer);
+				e.printStackTrace(printWriter);
+				logger.log(Level.SEVERE, "Exception occurred:" + e);
+				logger.log(Level.FINE, writer.toString());
+			}
+			if (null != b) {
+				b.putChar(Helper.MAPPED_INDEX, Helper.STATE_FAIL);
+			}
+			System.exit(5);
+		}
+	}
+}