From: Enrico Scholz Date: Mon, 19 Jan 2004 18:11:59 +0000 (+0000) Subject: initial checkin X-Git-Tag: VERSION_0_10~887 X-Git-Url: http://git.linux-vserver.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=4e116a34e870349dba10828da7f8324ab9d52d59;p=util-vserver.git initial checkin git-svn-id: http://svn.linux-vserver.org/svn/util-vserver/trunk@642 94cd875c-1c1d-0410-91d2-eb244daf1a30 --- diff --git a/util-vserver/src/vunify-compare.ic b/util-vserver/src/vunify-compare.ic new file mode 100644 index 0000000..567b03f --- /dev/null +++ b/util-vserver/src/vunify-compare.ic @@ -0,0 +1,38 @@ +// $Id$ --*- c -*-- + +// Copyright (C) 2004 Enrico Scholz +// +// 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; version 2 of the License. +// +// 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + + +static bool +compareUnify(struct stat const *lhs, struct stat const *rhs) +{ + return (lhs->st_dev ==rhs->st_dev && + lhs->st_ino !=rhs->st_ino && + lhs->st_mode ==rhs->st_mode && + lhs->st_uid ==rhs->st_uid && + lhs->st_gid ==rhs->st_gid && + lhs->st_size ==rhs->st_size && + lhs->st_mtime==rhs->st_mtime); + + // TODO: acl? time? +} + +static bool +compareDeUnify(struct stat const *lhs, struct stat const *rhs) +{ + return (lhs->st_dev ==rhs->st_dev && + lhs->st_ino ==rhs->st_ino); +} diff --git a/util-vserver/src/vunify-doit.ic b/util-vserver/src/vunify-doit.ic new file mode 100644 index 0000000..9a5f5a5 --- /dev/null +++ b/util-vserver/src/vunify-doit.ic @@ -0,0 +1,172 @@ +// $Id$ --*- c -*-- + +// Copyright (C) 2004 Enrico Scholz +// +// 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; version 2 of the License. +// +// 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + +#include "wrappers-io.h" + +#include +#include + +static bool +doitUnify(char const *src, char const *dst) +{ + size_t l = strlen(dst); + char tmpfile[l + sizeof(".XXXXXX")]; + int fd; + int fd_src; + bool res = false; + + // at first, set the ILI flags on 'src' + fd_src = open(src, O_RDONLY); + if (fd_src==-1) { + perror("open"); + return false; + } + if (vc_X_set_ext2flags(fd_src, + VC_IMMUTABLE_FILE_FL|VC_IMMUTABLE_LINK_FL, 0)==-1) { + perror("vc_X_set_ext2flags()"); + close(fd_src); + return false; + } + close(fd_src); + + // now, create a temporary filename + memcpy(tmpfile, dst, l); + memcpy(tmpfile+l, ".XXXXXX", 8); + fd = mkstemp(tmpfile); + close(fd); + + if (fd==-1) { + perror("mkstemp()"); + return false; + } + + // and rename the old file to this name + if (rename(dst, tmpfile)==-1) { + perror("rename()"); + goto err; + } + + // now, link the src-file to dst + if (link(src, dst)==-1) { + perror("link()"); + + unlink(dst); + if (rename(tmpfile, dst)==-1) { + perror("FATAL error in rename()"); + _exit(1); + } + goto err; + } + + res = true; + + err: + unlink(tmpfile); + + return res; +} + +static bool +doitDeUnify(char const UNUSED *src, char const *dst) +{ + size_t l = strlen(dst); + char tmpfile[l + sizeof(".XXXXXX")]; + int fd_src, fd_tmp; + struct stat st; + struct utimbuf utm; + + fd_src = open(dst, O_RDONLY); + if (fd_src==-1) { + perror("open()"); + return false; + } + + if (fstat(fd_src, &st)==-1) { + perror("fstat()"); + close(fd_src); + return false; + } + + memcpy(tmpfile, dst, l); + memcpy(tmpfile+l, ".XXXXXX", 8); + fd_tmp = mkstemp(tmpfile); + + if (fd_tmp==-1) { + perror("mkstemp()"); + tmpfile[0] = '\0'; + goto err; + } + + if (fchown(fd_tmp, st.st_uid, st.st_gid)==-1 || + fchmod(fd_tmp, st.st_mode)==-1) { + perror("fchown()/fchmod()"); + goto err; + } + + // todo: acl? + + for (;;) { + char buf[0x4000]; + ssize_t len = read(fd_src, buf, sizeof buf); + if (len==-1) { + perror("read()"); + goto err; + } + if (len==0) break; + + if (!WwriteAll(fd_tmp, buf, len)) goto err; + } + + if (close(fd_src)==-1) { + perror("close()"); + goto err; + } + if (close(fd_tmp)==-1) { + perror("close()"); + goto err; + } + + utm.actime = st.st_atime; + utm.modtime = st.st_mtime; + + // ALERT: race !!! + if (utime(tmpfile, &utm)==-1) { + perror("utime()"); + goto err1; + } + + if (unlink(dst)==-1) { + perror("unlink()"); + goto err1; + } + + // ALERT: race !!! + if (rename(tmpfile, dst)==-1) { + perror("FATAL error in rename()"); + _exit(1); + } + + return true; + + err: + close(fd_src); + close(fd_tmp); + err1: + if (tmpfile[0]) unlink(tmpfile); + + return false; +}