From: Sukadev Bhattiprolu <sukadev@linux.vnet.ibm.com>
Date: Mon, 12 Jul 2010 13:13:18 +0000 (+0200)
Subject: Ensure frezer state has changed
X-Git-Tag: lxc-0.7.2~26
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=ba31511e05aa098a22552b1c8da39e80ae4a090c;p=thirdparty%2Flxc.git

Ensure frezer state has changed

A write to the freezer.state file does not gurantee that the state has
changed. To ensure that the freezer state is either FROZEN or THAWED,
read the freezer state and if it has not changed, repeat the write.

Changelog[v2]:
	- Minor reorg of code
	- Comments from Daniel Lezcano:
		- lseek() before each read/write of freezer.state
		- Have lxc_freeze_unfreeze() return -1 on error

Signed-off-by: Sukadev Bhattiprolu <sukadev@linux.vnet.ibm.com>
Signed-off-by: Daniel Lezcano <dlezcano@fr.ibm.com>
---

diff --git a/src/lxc/freezer.c b/src/lxc/freezer.c
index cff954ec9..07eede5c6 100644
--- a/src/lxc/freezer.c
+++ b/src/lxc/freezer.c
@@ -42,6 +42,7 @@ static int freeze_unfreeze(const char *name, int freeze)
 {
 	char *nsgroup;
 	char freezer[MAXPATHLEN], *f;
+	char tmpf[32];
 	int fd, ret;
 	
 	ret = lxc_cgroup_path_get(&nsgroup, name);
@@ -50,7 +51,7 @@ static int freeze_unfreeze(const char *name, int freeze)
 
 	snprintf(freezer, MAXPATHLEN, "%s/freezer.state", nsgroup);
 
-	fd = open(freezer, O_WRONLY);
+	fd = open(freezer, O_RDWR);
 	if (fd < 0) {
 		SYSERROR("failed to open freezer for '%s'", name);
 		return -1;
@@ -58,22 +59,57 @@ static int freeze_unfreeze(const char *name, int freeze)
 
 	if (freeze) {
 		f = "FROZEN";
-		ret = write(fd, f, strlen(f) + 1) < 0;
+		ret = write(fd, f, strlen(f) + 1);
 	} else {
 		f = "THAWED";
-		ret = write(fd, f, strlen(f) + 1) < 0;
+		ret = write(fd, f, strlen(f) + 1);
 
 		/* compatibility code with old freezer interface */
-		if (ret) {
+		if (ret < 0) {
 			f = "RUNNING";
 			ret = write(fd, f, strlen(f) + 1) < 0;
 		}
 	}
 
-	close(fd);
-	if (ret) 
-		SYSERROR("failed to write to '%s'", freezer);
+	if (ret < 0) {
+		SYSERROR("failed to write '%s' to '%s'", f, freezer);
+		goto out;
+	}
+
+	while (1) {
+		ret = lseek(fd, 0L, SEEK_SET);
+		if (ret < 0) {
+			SYSERROR("failed to lseek on file '%s'", freezer);
+			goto out;
+		}
+
+		ret = read(fd, tmpf, sizeof(tmpf));
+		if (ret < 0) {
+			SYSERROR("failed to read to '%s'", freezer);
+			goto out;
+		}
+
+		ret = strncmp(f, tmpf, strlen(f));
+		if (!ret)
+			break;		/* Success */
 
+		sleep(1);
+
+		ret = lseek(fd, 0L, SEEK_SET);
+		if (ret < 0) {
+			SYSERROR("failed to lseek on file '%s'", freezer);
+			goto out;
+		}
+
+		ret = write(fd, f, strlen(f) + 1);
+		if (ret < 0) {
+			SYSERROR("failed to write '%s' to '%s'", f, freezer);
+			goto out;
+		}
+	}
+
+out:
+	close(fd);
 	return ret;
 }