#pragma implementation
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/utsname.h>
#include <fcntl.h>
#include <unistd.h>
#include <utime.h>
#include "vutil.h"
#include <sys/ioctl.h>
#include <ext2fs/ext2_fs.h>

bool testmode;
int debug;

int file_copy (const char *src, const char *dst, struct stat64 &st)
{
	int ret = -1;
	FILE *fin = fopen (src,"r");
	if (fin == NULL){
		fprintf (stderr,"Can't open file %s (%s)\n",src,strerror(errno));
	}else{
		FILE *fout = fopen (dst,"w");
		if (fout == NULL){
			fprintf (stderr,"Can't open file %s (%s)\n",dst,strerror(errno));
		}else{
			char buf[8192];
			int len;
			while ((len=fread(buf,1,sizeof(buf),fin))>0){
				fwrite (buf,1,len,fout);
			}
			fflush (fout);
			ret = 0;
			if (fchown (fileno(fout),st.st_uid,st.st_gid)==-1){
				fprintf (stderr,"Can't chown file %s (%s)\n"
					,dst,strerror(errno));
				ret = -1;
			}else if (fchmod (fileno(fout),st.st_mode)==-1){
				fprintf (stderr,"Can't chmod file %s (%s)\n"
					,dst,strerror(errno));
				ret = -1;
			}
			fclose(fout);
			struct utimbuf timbuf;
			timbuf.modtime = st.st_mtime;
			timbuf.actime = st.st_atime;
			if (utime (dst,&timbuf)==-1){
				fprintf (stderr,"Can't set time stamp on file %s (%s)\n"
					,dst,strerror(errno));
			}
		}
		fclose (fin);
	}
	return ret;
}

/*
	Set the immutable flag on a file
*/
int setext2flag (const char *fname, bool set, int ext2flags)
{
	int ret = -1;
	if (testmode){
		ret = 0;
	}else{
		int fd = open (fname,O_RDONLY);
		if (fd == -1){
			fprintf (stderr,"Can't open file %s (%s)\n",fname 
				,strerror(errno));
		}else{
			int flags = set ? ext2flags : 0;
			ret = ioctl (fd,EXT2_IOC_SETFLAGS,&flags);
			close (fd);
			if (ret == -1){
				fprintf (stderr,"Can't %s immutable flag on file %s (%s)\n"
					,(set ? "set" : "unset")
					,fname
					,strerror(errno));
			}
		}
	}
	return ret;
}

int vbuild_mkdir (const char *path, mode_t mode)
{
	int ret = -1;
	if (testmode){
		printf ("mkdir %s; chmod %o %s\n",path,mode,path);
		ret = 0;
	}else{
		ret = mkdir (path,mode);
		if (ret == -1 && errno == EEXIST){
			struct stat64 st;
			if (lstat64(path,&st)!=-1	&& S_ISDIR(st.st_mode)){
				ret = chmod (path,mode);
			}
		}
	}
	return ret;
}

int vbuild_mknod(const char *path, mode_t mode, dev_t dev)
{
	int ret = -1;
	if (testmode){
		printf ("mknod %s %o %02x:%02x\n",path,mode,major(dev),minor(dev));
		ret = 0;
	}else{
		ret = mknod (path,mode,dev);
		if (ret == -1 && errno == EEXIST){
			struct stat64 st;
			//lstat64(path,&st);
			if (lstat64(path,&st)!=-1
				&& (st.st_mode & S_IFMT) == (mode & S_IFMT)
				&& st.st_rdev == dev){
				ret = chmod (path,mode);
			}
		}
	}
	return ret;
}
int vbuild_symlink(const char *src, const char *dst)
{
	int ret = -1;
	if (testmode){
		printf ("ln -s %s %s\n",src,dst);
		ret = 0;
	}else{
		ret = symlink (src,dst);
	}
	return ret;
}

int vbuild_link(const char *src, const char *dst)
{
	int ret = -1;
	if (testmode){
		printf ("ln %s %s\n",src,dst);
		ret = 0;
	}else{
		ret = link (src,dst);
	}
	return ret;
}

int vbuild_unlink(const char *path)
{
	int ret = -1;
	if (testmode){
		printf ("unlink %s\n",path);
		ret = 0;
	}else{
		ret = unlink (path);
	}
	return ret;
}

int vbuild_chown(const char *path, uid_t uid, gid_t gid)
{
	int ret = -1;
	if (testmode){
		printf ("chown %d.%d %s\n",uid,gid,path);
		ret = 0;
	}else{
		ret = chown (path,uid,gid);
	}
	return ret;
}

int vbuild_file_copy(
	const char *src,
	const char *dst,
	struct stat64 &st)
{
	int ret = -1;
	if (testmode){
		printf ("cp -a %s %s\n",src,dst);
		ret = 0;
	}else{
		ret = file_copy (src,dst,st);
	}
	return ret;
}

/*
	Load the list of all packages in a vserver
*/
void vutil_loadallpkg (string &refserver, list<PACKAGE> &packages)
{
	FILE *fin = vutil_execdistcmd (K_PKGVERSION,refserver,NULL);
	if (fin != NULL){
		char line[1000];
		while (fgets(line,sizeof(line)-1,fin)!=NULL){
			int last = strlen(line)-1;
			if (last >= 0 && line[last] == '\n') line[last] = '\0';
			packages.push_back (PACKAGE(line));
		}
		pclose (fin);
	}
}

int vutil_lstat (string path, struct stat64 &st)
{
	int ret = 0;
	if (lstat64(path.c_str(),&st) == -1){
		fprintf (stderr,"Can't lstat file %s (%s)\n"
			,path.c_str(),strerror(errno));
		ret = -1;
	}
	return ret;
}

const char K_PKGVERSION[]="pkgversion";
const char K_DUMPFILES[]="dumpfiles";
const char K_UNIFILES[]="unifiles";

FILE *vutil_execdistcmd (const char *key, const string &vserver, const char *args)
{
	string cmd = "/usr/lib/vserver/distrib-info ";
	cmd += vserver;
	cmd += " ";
	cmd += key;
	if (args != NULL){
		cmd += " ";
		cmd += args;
	}
	FILE *ret = popen (cmd.c_str(),"r");
	if (ret == NULL){
		fprintf (stderr,"Can't execute command %s\n",cmd.c_str());
	}
	return ret;
}

int vutil_readconf (const char *name, VSERVER_CONF &conf)
{
	int ret = -1;
	string cmd = "/usr/lib/vserver/printconf.sh ";
	cmd += name;
	FILE *fin = popen (cmd.c_str(),"r");
	if (fin == NULL){
		fprintf (stderr,"Can't execute command %s\n",cmd.c_str());
	}else{
		char buf[1000];
		while (fgets(buf,sizeof(buf)-1,fin)!=NULL){
			int last = strlen(buf)-1;
			if (last >= 0) buf[last] = '\0';
			char *pt = strchr(buf,'=');
			if (pt != NULL){
				*pt++ = '\0';
				if (strcmp(buf,"ONBOOT")==0){
					conf.onboot = strcasecmp(pt,"yes")==0;
				}else if (strcmp(buf,"IPROOT")==0){
					conf.iproot = pt;
				}else if (strcmp(buf,"IPROOTDEV")==0){
					conf.iprootdev = pt;
				}else if (strcmp(buf,"IPROOTMASK")==0){
					conf.iprootmask = pt;
				}else if (strcmp(buf,"S_HOSTNAME")==0){
					conf.s_hostname = pt;
				}else if (strcmp(buf,"S_DOMAINNAME")==0){
					conf.s_domainname = pt;
				}else if (strcmp(buf,"S_CAPS")==0){
					conf.s_caps = pt;
				}else if (strcmp(buf,"S_FLAGS")==0){
					conf.s_flags = pt;
				}else if (strcmp(buf,"S_NICE")==0){
					conf.s_nice = atoi(pt);
				}else if (strcmp(buf,"PRIORITY")==0){
					conf.priority = atoi(pt);
				}else if (strcmp(buf,"VSERVERS_ROOT")==0){
					conf.vserversroot = pt;
				}else if (strcmp(buf,"VSERVERDIR")==0){
					conf.vserverdir = pt;
				}else if (strcmp(buf,"BUILDFROM")==0){
					conf.buildfrom = pt;
				}else if (strcmp(buf,"BUILDARG")==0){
					conf.buildarg = pt;
				}else if (strcmp(buf,"BUILDUSING")==0){
					conf.buildusing = pt;
				}else if (strcmp(buf,"VARCH")==0){
					conf.varch = pt;
				}else if (strcmp(buf,"VKRELEASE")==0){
					conf.vkrelease = pt;
				}else if (strcmp(buf,"ULIMIT")==0){
					conf.ulimit = pt;
				}else if (strcmp(buf,"PROFILE")==0){
					conf.profile = pt;
				}else if (strcmp(buf,"GENERATEMTAB")==0){
					conf.generatemtab = strcasecmp(pt,"yes")==0;
				}else if (strcmp(buf,"REQ_S_CONTEXT")==0){
					conf.xid = -1;
					if (pt[0] != '\0') conf.xid = atoi(pt);
				}
			}
		}
		ret = pclose (fin);
		// fprintf (stderr,"ret = %d\n",ret);
		if (ret != 0) ret = -1;
	}
	return ret;
}

int vutil_readconf (const string &name, VSERVER_CONF &conf)
{
	return vutil_readconf (name.c_str(),conf);
}

/*
	Return the extended attribute flag for IUNLINK.
	(It changed between 2.4 and 2.6)
*/
unsigned vutil_get_iunlinkflag()
{
	unsigned ret = EXT2_IMMUTABLE_LINK_FL;
	struct utsname uts;
	if (uname(&uts)!=-1){
		//printf ("sysname :%s:\n",uts.sysname);
		//printf ("release :%s:\n",uts.release);
		if (strcmp(uts.release,"2.6.")< 0){
			ret = EXT2_IMMUTABLE_LINK_FL24;
		}
	}
	return ret;
}

/*
	Return the extended attribute flag for IBARRIER.
	(It changed between 2.4 and 2.6)
*/
unsigned vutil_get_ibarrierflag()
{
	unsigned ret = EXT2_IBARRIER_FL;
	struct utsname uts;
	if (uname(&uts)!=-1){
		//printf ("sysname :%s:\n",uts.sysname);
		//printf ("release :%s:\n",uts.release);
		if (strcmp(uts.release,"2.6.")< 0){
			ret = EXT2_IBARRIER_FL24;
		}
	}
	return ret;
}