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>
This commit is contained in:
Sukadev Bhattiprolu 2010-07-12 15:13:18 +02:00 committed by Daniel Lezcano
parent ebb9ec72ce
commit ba31511e05

View File

@ -42,6 +42,7 @@ static int freeze_unfreeze(const char *name, int freeze)
{ {
char *nsgroup; char *nsgroup;
char freezer[MAXPATHLEN], *f; char freezer[MAXPATHLEN], *f;
char tmpf[32];
int fd, ret; int fd, ret;
ret = lxc_cgroup_path_get(&nsgroup, name); 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); snprintf(freezer, MAXPATHLEN, "%s/freezer.state", nsgroup);
fd = open(freezer, O_WRONLY); fd = open(freezer, O_RDWR);
if (fd < 0) { if (fd < 0) {
SYSERROR("failed to open freezer for '%s'", name); SYSERROR("failed to open freezer for '%s'", name);
return -1; return -1;
@ -58,22 +59,57 @@ static int freeze_unfreeze(const char *name, int freeze)
if (freeze) { if (freeze) {
f = "FROZEN"; f = "FROZEN";
ret = write(fd, f, strlen(f) + 1) < 0; ret = write(fd, f, strlen(f) + 1);
} else { } else {
f = "THAWED"; f = "THAWED";
ret = write(fd, f, strlen(f) + 1) < 0; ret = write(fd, f, strlen(f) + 1);
/* compatibility code with old freezer interface */ /* compatibility code with old freezer interface */
if (ret) { if (ret < 0) {
f = "RUNNING"; f = "RUNNING";
ret = write(fd, f, strlen(f) + 1) < 0; ret = write(fd, f, strlen(f) + 1) < 0;
} }
} }
close(fd); if (ret < 0) {
if (ret) SYSERROR("failed to write '%s' to '%s'", f, freezer);
SYSERROR("failed to write to '%s'", 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; return ret;
} }