path: root/client/engine/Rsync.lib

#!/bin/bash
#/**
# ogCreateFileImage  [ REPO | CACHE ] image_name extension size
#@brief   Crear el archivo 
#@param 1 Repositorio  [ REPO | CACHE ]
#@param 2 Nombre Imagen
#@param 3 Tipo imagen [ img |diff ]
#@param 4 Tamaño de la imagen
#@return  instrucción para ser ejecutada.
#*/

function ogCreateFileImage () {
local SIZEREQUIRED IMGDIR IMGFILE DIRMOUNT LOOPDEVICE  IMGSIZE IMGEXT

if [ "$*" == "help" ]; then
    ogHelp "$FUNCNAME" : "$MSG_HELP_ogCreateFileImage" \
	   "$FUNCNAME [ REPO|CACHE ] image_name extension  size(K)" \
           "$FUNCNAME REPO Ubuntu12 img 300000" \
           "$FUNCNAME CACHE Windows7 diff 20000000"
    return
fi


if [ $# -lt 4 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME  [ REPO|CACHE ] image_name extension  size(k)"
    return $?
fi

SIZEREQUIRED=$4
[ $SIZEREQUIRED -lt 300000 ] && SIZEREQUIRED=300000

if [ "$1" == "CACHE" -o "$1" == "cache" ]; then
	IMGDIR="$(ogGetParentPath "$1" "/$2")"
	[ "$3" == "img" ] && IMGEXT="img" ||  IMGEXT="img.diff"
	IMGFILE="${IMGDIR}/$(basename "/$2").$IMGEXT"
	## Si no existe, crear subdirectorio de la imagen.
	if [ $? != 0 ]; then
            ogEcho log session  "      $MSG_HELP_ogMakeDir \"$1 $(dirname "$2")." 
            ogMakeDir "$1" "$(dirname "/$2")" || return $(ogRaiseError $OG_ERR_NOTWRITE "$3 /$4"; echo $?)
            IMGDIR="$(ogGetParentPath "$1" "/$2")" || return $(ogRaiseError $OG_ERR_NOTWRITE "$3 /$4"; echo $?)
	fi
	DIRMOUNT="/tmp/$(ogGetMountImageDir "$2" "$3")"
        mkdir -p "$DIRMOUNT"
        LOOPDEVICE=$(losetup -f)
        # Si existe el fichero de la imagen se hace copia de seguridad, si  no existe se crea.
        if [ -f "$IMGFILE" ]; then
		# Si la imagen esta montada la desmonto
                if [ -r "$DIRMOUNT/ogimg.info" ]; then
                    umount "$DIRMOUNT"
                    [ $? -ne 0 ] && return $(ogRaiseError $OG_ERR_DONTUNMOUNT_IMAGE "$1 $2.$IMGEXT"; echo $?)
                fi

		if [ "$BACKUP" == "true" ]; then
                    # Copia seguridad 
                    ogEcho log session  "     $MSG_SCRIPTS_FILE_RENAME \"$IMGFILE\" ->  \"$IMGFILE.ant\"." 
                    cp -f  "$IMGFILE" "$IMGFILE.ant"  
                    mv -f "$IMGFILE.torrent" "$IMGFILE.torrent.ant" 2>/dev/null
                    rm -f "$IMGFILE.sum"
		fi

                IMGSIZE=$(ls -l --block-size=1024 "$IMGFILE" | awk '{print $5}')
                if [ $IMGSIZE -lt $SIZEREQUIRED ];then
                   ogEcho log session  "      $MSG_SYNC_RESIZE" 
                   echo "      truncate --size=>$SIZEREQUIRED k $IMGFILE" 
                   truncate --size=">$SIZEREQUIRED"k "$IMGFILE" &> $OGLOGCOMMAND
		   # FS de la imagen segun la version del kernel: < 3.7 ext4, si >=  btrfs 
		   #if [[ "$(uname -r)" < "3.7" ]]; then
		   if [ $(uname -r|cut -d. -f2) -lt 7 ]; then
                	losetup $LOOPDEVICE "$IMGFILE"
			echo "      resize2fs -f $LOOPDEVICE"
			resize2fs -f $LOOPDEVICE &> $OGLOGCOMMAND
			
		   else
		   	echo "      ogMountImage $1 "$2" $3"
		   	ogMountImage $1 "$2" $3
                   	echo "      btrfs filesystem resize max $DIRMOUNT"
                   	btrfs filesystem resize max "$DIRMOUNT" &> $OGLOGCOMMAND
		   fi
                fi
        else
                echo "      dd if=/dev/zero of=$IMGFILE bs=1024 count=$SIZEREQUIRED"
                dd if=/dev/zero of="$IMGFILE" bs=1024 count=$SIZEREQUIRED 2>&1 | tee -a $OGLOGCOMMAND
                #Formateamos imagen
                losetup $LOOPDEVICE $IMGFILE
		# FS de la imagen segun la version del kernel: < 3.7 ext4, si >=  btrfs 
		#if [[ "$(uname -r)" < "3.7" ]]; then
		if [ $(uname -r|cut -d. -f2) -lt 7 ]; then
		   echo "      mkfs.ext4 -i 4096  -b 4096 -L "${2##*\/}" $LOOPDEVICE"
		   mkfs.ext4 -i 4096  -b 4096 -L "${2##*\/}" $LOOPDEVICE 2>&1 |tee -a $OGLOGCOMMAND
		else
                   echo "      mkfs.btrfs  -L ${2##*\/} $LOOPDEVICE "
                   mkfs.btrfs  -L "${2##*\/}" $LOOPDEVICE 2>&1 | tee -a $OGLOGCOMMAND
		fi
	fi
	# Monto la imagen 
	ogMountImage $1 "$2" $3 &>/dev/null
	[ $? -eq 0 ] || return $( ogRaiseError $OG_ERR_IMAGE "$3 $4"; echo $?)
	touch "$DIRMOUNT/ogimg.info"

	# Si existe dispositivo de loop lo borro.
        [ $LOOPDEVICE ] && losetup -d $LOOPDEVICE 2>&1 &>/dev/null

else
	[ -z $REPOIP ] && REPOIP=$(ogGetRepoIp)
        echo "      hose $REPOIP 2009 --out sh -c \"echo -ne CREATE_IMAGE $2 $3 $SIZEREQUIRED \"" 
        hose $REPOIP 2009 --out sh -c "echo -ne CREATE_IMAGE \"$2\" $3 $SIZEREQUIRED"
fi

}

function ogCreateInfoImage () {
#/**
#  ogCreateInfoImage
#@brief   Crear listados con la informacion de la imagen, los situa en /tmp.
#@param 1 num_disk
#@param 2 num_part
#@param 3 Repositorio  [ REPO | CACHE ] (opcional en las completas)
#@param 4 Nombre Imagen Basica (opcional en las completas)
#@param 5 Tipo imagen [ img | diff ]
#*/
local  IMGTYPE IMGDIRAUX DIRMOUNT DESTRSYNC PASSWORD OPTRSYNC USERRSYNC ORIG FSTYPE PART DIREMPTY IMGLIST IMGINFO IMGACL 

# Ayuda o menos de 5 parametros y la imagen no es basica
if [ "$*" == "help" -o   $# -lt 5 -a "$3" != "img"  ]; then
	ogHelp "$FUNCNAME" : "$MSG_HELP_ogCreateInfoImage" \
		"$FUNCNAME num_disk num_part [ REPO|CACHE ] [ base_image_name ] extension  " \
       		"base image -> $FUNCNAME 1 2 img" \
     		"diff image -> $FUNCNAME 1 1 CACHE Windows7 diff "
    	return
fi

if [ $# -lt 3 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME  num_disk num_part [ REPO|CACHE ] [ base_image_name]  extension  "
    return $?
fi

if [ $3 == "img" ]; then
   IMGTYPE="img"
else
   # Comprobamos que las extension sea valida
   ogCheckStringInGroup $5 "img diff" || return $( ogRaiseError  $OG_ERR_FORMAT "$MSG_SYNC_EXTENSION"; echo $?)
   IMGTYPE=$5
   if [ "$IMGTYPE" == "diff" ]; then
	# Imagen completa con la que comparo la particion.
	IMGDIRAUX="$(ogGetMountImageDir "$4" "img")"
	if [ "$3" == "CACHE" -o "$3" == "cache" ]; then
           DIRMOUNT="/tmp/$IMGDIRAUX"
           DESTRSYNC="$DIRMOUNT"
	else
	   [ -z $REPOIP ] && REPOIP=$(ogGetRepoIp)
           DIRMOUNT="$OGIMG/$IMGDIRAUX"
           USERRSYNC="opengnsys"
           PASSWORD="--password-file=/scripts/passrsync"
           DESTRSYNC="$USERRSYNC@$REPOIP::ogimages/$IMGDIRAUX"
	fi
   fi
fi


ORIG=$(ogMount $1 $2)
FSTYPE=$(ogGetFsType $1 $2)
PART=$(ogDiskToDev "$1" "$2" 2>/dev/null)

# Creamos la lista del contenido y lo situamos en la particion a copiar.
DIREMPTY="/tmp/empty$$"
IMGLIST="/tmp/ogimg.list"
IMGINFO="/tmp/ogimg.info"
IMGACL="/tmp/ogimg.acl"

# Borramos archivos antiguos.
rm -f /tmp/ogimg.* 2>/dev/null
rm -f $ORIG/ogimg.* 2>/dev/null

# En las diferenciales no sabemos el tamaño -> ponemos una constante.
SIZEDATA=${SIZEDATA:-"SIZEDATA"}

# Incluimos información de la imagen. Segun el kernel sera ext4 o btrfs.
#if [[ "$(uname -r)" < "3.7" ]]; then
if [ $(uname -r|cut -d. -f2) -lt 7 ]; then
	echo "#EXT4:NO:$FSTYPE:$SIZEDATA" > $IMGINFO
else
	echo "#BTRFS:LZO:$FSTYPE:$SIZEDATA" > $IMGINFO
fi

if [ "$IMGTYPE" == "img" ]; then
	# Imagen Basica
	echo "      rsync -aHAXvn --delete $ORIG/ $DIREMPTY >> $IMGINFO" 
	rsync -aHAXvn --delete $ORIG/ $DIREMPTY>> $IMGINFO
	sed -i -e s/"^sent.*.bytes\/sec"//g -e s/^total.*.speedup.*.$//g -e s/"sending.*.list"//g $IMGINFO
	sed -i  '/^\.\//d' $IMGINFO

else
	# Imagen Diferencial
	echo "      rsync -aHAXvn$OPTRSYNC  --delete  $ORIG/ $DESTRSYNC a $IMGLIST" 
	rsync -aHAXvn$OPTRSYNC $PASSWORD  --delete  "$ORIG/" "$DESTRSYNC" >> $IMGLIST
	sed -i -e s/"^sent.*.bytes\/sec"//g -e s/^total.*.speedup.*.$//g -e s/"sending.*.list"//g $IMGLIST
	sed -i  '/^\.\//d' $IMGLIST

	# Creamos informacion de la imagen
	grep -e '\->' -e  '\=>' $IMGLIST > /tmp/ogimg.ln
	grep -e  ^deleting  $IMGLIST | sed s/^deleting\ //g | grep -v ^ogimg  > /tmp/ogimg.rm
	#grep -v -e '\->' -e  '\=>'  -e ^deleting  $IMGLIST >> $IMGINFO
	grep -v -e '\->' -e  '\=>'  -e ^deleting -e ^created  $IMGLIST >> $IMGINFO

	rm -f $IMGLIST

	# Comprobamos que los ficheros de diferencias no esten vacios o salimos con error.
	if [ $(grep -v -e "^$"  -e "^#" $IMGINFO /tmp/ogimg.ln /tmp/ogimg.rm |wc -l) -eq 0 ]; then 
		ogRaiseError $OG_ERR_NOTDIFFERENT "$1 $2 $3 $4 $5" 
		return $0	
	fi

fi

# Guardamos el contenido de las acl (Solo win) Necesario particion desmontada (esta asi)
ogUnmount $1 $2
if [ $FSTYPE == "NTFS" ]; then
	echo "      ntfs-3g.secaudit -b $PART /"
	ntfs-3g.secaudit -b $PART / > $IMGACL 
fi


}


#/**
#  ogAclFilter
#@brief Del fichero de acl de la partición extraemos las acl de los ficheros de la diferencial
#@param No.
#@return  (nada)
#*/
function ogAclFilter () {
local  IMGACL IMGINFO FILES ACLTMP

# Ayuda o menos de 5 parametros y la imagen no es basica
if [ "$*" == "help" ]; then
	ogHelp "$MSG_HELP_ogAclFilter" \
		"$FUNCNAME [ no parameters ]  "
    	return
fi

IMGACL="/tmp/ogimg.acl"
IMGINFO="/tmp/ogimg.info"
FILES="/tmp/files$$"
ACLTMP="/tmp/acl$$.tmp"
ACLFILES="/tmp/aclfiles$$"

# comprobamos que existan los archivos de origen. Si no salimos sin error.
[ -f $IMGACL -a -f $IMGINFO ] || return 0

echo "" > $ACLTMP
grep -n  -e "File" -e "Directory" $IMGACL > $ACLFILES

# Al fichero le quitamos las líneas sobrantes: comentarios y lineas vacias.
sed -e s/"^#.*$"//g $IMGINFO > $FILES
sed -i '/^$/d'  $FILES


# Recorremos el listado y extraemos la acl correspondiente al fichero o directorio.
while read LINE; do
    #read END INI <<< "$(grep -n  -e "File" -e "Directory" $IMGACL| grep -A 1 "$LINE"| awk -F :  '!(NR%2){print $1" "p}{p=$1}' )"
    read END INI <<< "$(grep -A 1 "$LINE" $ACLFILES | awk -F :  '!(NR%2){print $1" "p}{p=$1}' )"
    let NUM=$END-$INI-1
    # Si algún archivo no se encuentra, el error lo mandamos a /dev/null
    sed -n -e $INI,+"$NUM"p  $IMGACL 2>/dev/null >> $ACLTMP
    echo "aclfilter: $LINE" >> $OGLOGCOMMAND
done < $FILES

cp $ACLTMP $IMGACL
rm -f $FILES $ACLTMP $ACLFILES 

}

#/**
#  ogRestoreInfoImage
#@brief Crear o modificar enlaces y restaurar las ACL. La informacion esta ya copiada a la particion.
#@param 1 num_disk
#@param 2 num_part
#*/
function ogRestoreInfoImage () {
local  DEST PART IMGACL IMGLN OPTLN LINEA DESTLN ORIGLN TYPELN 

# Ayuda o menos de 5 parametros y la imagen no es basica
if [ "$*" == "help" ]; then
        ogHelp 	"$FUNCNAME": "$MSG_HELP_ogRestoreInfoImage" \
		"$FUNCNAME num_disk num_part" \
       		"base image -> $FUNCNAME 1 2 " \
		"diff image -> $FUNCNAME 1 1 "
        return
fi

if [ $# -lt 2 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME num_disk num_part  "
    return $?
fi

DEST=$(ogMount $1 $2)
PART=$(ogDiskToDev "$1" "$2" 2>/dev/null)
IMGACL="ogimg.acl"
IMGLN="ogimg.ln"

#Creamos o modificamos los enlaces.
# La imagen diferencial tiene ogimg.ln 
# para la completa lo generamos con los enlaces que contengan  /mnt/
[ -r "$DEST/$IMGLN" ] || grep -e "->" -e "=>" "$DEST/ogimg.info"|grep "/mnt/" > "$DEST/$IMGLN"
if [ $(wc -l "$DEST/$IMGLN"|cut -f1 -d" ") -ne 0 ]; then
   while read LINEA
   do
        ORIGLN="${LINEA#*> }"
        # Si origen hace referencia a la particion lo modificamos
        echo $ORIGLN|grep "/mnt/"> /dev/null && ORIGLN="$DEST/${ORIGLN#/mnt/*/}"
	# rsync marca - los enlaces simbolicos y = enlaces "duros"
        LINEA="${LINEA%>*}"
        TYPELN="${LINEA##* }"
        DESTLN="${LINEA% *}"

   	if [ "$TYPELN" == "-" ]
   	then
           OPTLN='-s'
   	else
           OPTLN=''
   	fi
   	cd "$DEST/$(dirname "$DESTLN")"
	rm -f "$(basename "$DESTLN")"
   	ln $OPTLN "$ORIGLN" "$(basename "$DESTLN")"
   	echo -n "."
   done < "$DEST/$IMGLN"  2>/dev/null
   echo ""

fi

# Restauramos acl
if [ "$(ogGetFsType $1 $2)" == "NTFS"  -a -f "$DEST/$IMGACL" ] ; then
	cd /
        cp "$DEST/$IMGACL" /tmp
        ogUnmount "$1" "$2"
        echo "ntfs-3g.secaudit -se $PART /tmp/$IMGACL"
        ntfs-3g.secaudit -se $PART /tmp/$IMGACL
	# Para evitar que de falso error
	echo ""	
fi

}

function ogSyncCreate () {
#/**
# ogSyncCreate 
#@brief   sincroniza los datos de la partición a la imagen para crearla. La imagen esta montada en un directorio.
#@param 1 num_disk
#@param 2 num_part
#@param 3 Repositorio  [ REPO | CACHE ]
#@param 4 Nombre Imagen
#@param 5 Tipo imagen [ img | diff ]
#*/
local ORIG DIRAUX DIRMOUNT DESTRSYNC USERRSYNC PASSWORD OPTRSYNC

# Limpiamos los archivo de log
echo  "" >$OGLOGCOMMAND;

if [ "$*" == "help" ]; then
	ogHelp 	"$FUNCNAME": "$MSG_HELP_ogSyncCreate" \
		"$FUNCNAME num_disk num_part [ REPO|CACHE ] image_name extension  " \
       		"$FUNCNAME 1 2 REPO Ubuntu12 img" \
     		"$FUNCNAME 1 1 CACHE Windows7 diff "
    	return
fi


if [ $# -lt 4 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME  num_disk num_part [ REPO|CACHE ] image_name extension  "
    return $?
fi

ORIG=$(ogMount $1 $2)
DIRMOUNT="$(ogGetMountImageDir "$4" $5)"
# Si la imagen es diferencial la lista de ficheros a transferir esta dentro de la imagen.
if [ "$5" == "diff" ]; then
	FILESFROM=" --files-from=/tmp/ogimg.info"
	# Borramos los directorios
	sed -i '/\/$/d' /tmp/ogimg.info
else
	FILESFROM=""
fi

if [ "$3" == "CACHE" -o "$3" == "cache" ]; then
        DESTRSYNC="/tmp/$DIRMOUNT"
else
	[ -z $REPOIP ] && REPOIP=$(ogGetRepoIp)
        PASSWORD="--password-file=/scripts/passrsync"
        [ "$ogrsyncz" ==  "true" ] && OPTRSYNC="z "
	[ "$ogrsyncw" == "true" ] && OPTRSYNC="W$OPTRSYNC"
	USERRSYNC="opengnsys"
        DESTRSYNC="$USERRSYNC@$REPOIP::ogimages/$DIRMOUNT"
fi
# Sincronizamos los datos de la partición a la imagen
echo "      rsync -aHAX$OPTRSYNC --progress --inplace --delete $FILESFROM  $ORIG/ $DESTRSYNC" 
rsync -aHAX$OPTRSYNC $PASSWORD --progress --inplace --delete $FILESFROM  "$ORIG/" "$DESTRSYNC" 2>&1| egrep "^deleting|^sent|^sending|^total|%"  &>> $OGLOGCOMMAND
echo "      rsync -aHAX$OPTRSYNC --inplace /tmp/ogimg* $DESTRSYNC" 
rsync -aHAX$OPTRSYNC  $PASSWORD  --inplace /tmp/ogimg* "$DESTRSYNC"

}


#/**
# ogSyncRestore
#@brief   sincroniza los datos de la imagen a la partición para restaurarla.
#@param 1 Repositorio  [ REPO | CACHE ]
#@param 2 Nombre Imagen
#@param 3 Tipo imagen [ img | diff ]
#@param 4 num_disk
#@param 5 num_part
#*/
function ogSyncRestore () {
local DIRMOUNT ORIG DESTRSYNC PASSWORD OPTRSYNC USERRSYNC IMGINFO FILESFROM

# Limpiamos los archivo de log
echo  "" >$OGLOGCOMMAND;

if [ "$*" == "help" ]; then
	ogHelp 	"$FUNCNAME": "$MSG_HELP_ogSyncRestore" \
		"$FUNCNAME [ REPO|CACHE ] image_name extension  num_disk num_part  " \
		"$FUNCNAME REPO Ubuntu12 img 1 2" \
		"$FUNCNAME CACHE Windows7 diff 1 1"
    	return
fi


if [ $# -lt 5 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME  [ REPO|CACHE ] image_name extension   num_disk num_part "
    return $?
fi


DIRMOUNT="$(ogGetMountImageDir "$2" "$3")"
DESTRSYNC=$(ogGetMountPoint $4 $5)

# Borramos ficheros de informacion de restauraciones antiguas
rm -rf $DESTRSYNC/ogimg.*

# Origen y destino de la sincronizacion y en REPO opciones rsync
if [ "$1" == "CACHE" -o "$1" == "cache" ]; then
        ORIG="/tmp/$DIRMOUNT"
else
	[ -z $REPOIP ] && REPOIP=$(ogGetRepoIp)
        PASSWORD="--password-file=/scripts/passrsync"
        [ "$ogrsyncz" ==  "true" ] && OPTRSYNC="z "
	[ "$ogrsyncw" == "true" ] && OPTRSYNC="W$OPTRSYNC"
	USERRSYNC="opengnsys"
        ORIG="$USERRSYNC@$REPOIP::ogimages/$DIRMOUNT"
fi

# Opciones rsync en cache y repo
# Para la imagen basica, opcion de borrar archivos de la particion que no existen en la imagen
[ "$3" == "img" ] && [ "$ogrsyncdel" != "false" ] && OPTRSYNC="$OPTRSYNC --delete"

# Nos traemos listado ficheros y bajamos la imagen

ogEcho log session "      $MSG_SYNC_RESTORE"

# Si la imagen es diferencial nos traemos los archivos de informacion de la imagen.
if [ "$3" == "diff" ]; then
	# Lista de archivos a copiar:
	IMGINFO="ogimg.info"
	FILESFROM=" --files-from=$DESTRSYNC/$IMGINFO"

        echo "      rsync -aHAX$OPTRSYNC --progress  $ORIG/ogimg* $DESTRSYNC" 
        rsync -aHAX$OPTRSYNC $PASSWORD --progress  "$ORIG"/ogimg* $DESTRSYNC 
        # Borramos linea de información de la imagen, sino busca un fichero con ese nombre
        sed -i   '/^\#/d' $DESTRSYNC/$IMGINFO

	cd $DESTRSYNC
	# Diferencial: Borramos archivos sobrantes.
	ogEcho log session "      $MSG_SYNC_DELETE"  
	cat "$DESTRSYNC/ogimg.rm" 2>/dev/null | xargs rm -rf

fi

echo "      rsync -aHAX$OPTRSYNC --progress  $FILESFROM  $ORIG/ $DESTRSYNC"
rsync -aHAX$OPTRSYNC $PASSWORD --progress  $FILESFROM  "$ORIG/" "$DESTRSYNC"  2>&1 | egrep "^deleting|^sent|^sending|^total|%"  &>> $OGLOGCOMMAND


}

function ogMountImage () {
#/**
# ogMountImage
#@brief   Monta la imagen para sincronizar.
#@param 1 Repositorio  [ REPO | CACHE ]
#@param 2 Nombre Imagen
#@param 3 Tipo imagen [ img |diff ]
#@return punto de montaje
#*/
local IMGEXT IMGFILE DIRMOUNT

if [ "$*" == "help" ]; then
	ogHelp "$FUNCNAME": "$MSG_HELP_ogMountImage" \
		"$FUNCNAME [ REPO|CACHE ] image_name [ extension ]" \
		"$FUNCNAME REPO Ubuntu12" \
		"$FUNCNAME CACHE Windows7 diff"
	return
fi


if [ $# -lt 2 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME [ REPO|CACHE ]  image_name [ extension ]"
    return $?
fi

[ "$3" == "" -o "$3" == "img" ] && IMGEXT="img" || IMGEXT="img.diff"

DIRMOUNT="$(ogGetMountImageDir "$2" ${IMGEXT#*\.})"

if [ "$1" == "REPO" -o "$1" == "repo" ]; then
        [ -z $REPOIP ] && REPOIP=$(ogGetRepoIp)
        hose $REPOIP 2009 --out sh -c "echo -ne MOUNT_IMAGE \"$2\" ${IMGEXT#*\.}"
        echo "$OGIMG/$DIRMOUNT"
else
	# Si está montado nada que hacer.
	df | grep "$DIRMOUNT$" 2>&1 >/dev/null && echo "/tmp/$DIRMOUNT" && return 0

        IMGFILE="$(ogGetPath "$1" /"$2.$IMGEXT")" \
		|| return $(ogRaiseError $OG_ERR_NOTFOUND "$1 $2.$IMGEXT"; echo $?)
        mkdir -p "/tmp/$DIRMOUNT"
	
   	# FS de la imagen segun la version del kernel: < 3.7 ext4, si >=  btrfs 
	if [ $(uname -r|cut -d. -f2) -lt 7 ]; then
	    mount -t ext4 -o loop "$IMGFILE" "/tmp/$DIRMOUNT" 1>/dev/null
	else
	    mount -o compress=lzo  "$IMGFILE" "/tmp/$DIRMOUNT" 1>/dev/null
	fi

	# Comprobamos que se ha montado bien
	[ $? -eq 0 ] || return $(ogRaiseError $OG_ERR_DONTMOUNT_IMAGE "$1 $2 $3"; echo $?)
        echo "/tmp/$DIRMOUNT"
fi

}





function ogUnmountImage () {
#/**
# ogUnmountImage  [ REPO | CACHE ] Image_name [ extension ]
#@brief   Desmonta la imagen para sincronizar.
#@param 1 Repositorio  [ REPO | CACHE ]
#@param 2 Nombre Imagen
#@param 3 Tipo imagen [ img |diff ]
#*/
local IMGTYPE DIRMOUNT

if [ "$*" == "help" ]; then
    ogHelp "$FUNCNAME": "$MSG_HELP_ogUnmountImage" \
		"$FUNCNAME [ REPO|CACHE ] image_name [ extension ]" \
		"$FUNCNAME REPO Ubuntu12" \
		"$FUNCNAME CACHE Windows7 diff"
    return
fi

if [ $# -lt 2 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME  [ REPO|CACHE ]  image_name [ extension ]"
    return $?
fi

[ "$3" == "" ] && IMGTYPE="img" || IMGTYPE="$3"

if [ "$1" == "CACHE" -o "$1" == "cache" ]; then
	DIRMOUNT="/tmp/$(ogGetMountImageDir "$2" $IMGTYPE)"
	umount "$DIRMOUNT"
	rmdir "$DIRMOUNT"
else
	[ -z $REPOIP ] && REPOIP=$(ogGetRepoIp)
        echo "      hose $REPOIP 2009 --out sh -c echo -ne UMOUNT_IMAGE \"$2\" $IMGTYPE"
        hose $REPOIP 2009 --out sh -c "echo -ne UMOUNT_IMAGE \"$2\" $IMGTYPE"
fi

}

function ogGetMountImageDir () {
#/**
# ogGetMountImageDir
#@brief   Devuelve el directorio de montaje de la imagen.
#@param 1 Nombre Imagen
#@param 2 Tipo imagen [ img |diff ]
#*/
local DIRMOUNT
if [ "$*" == "help" ]; then
    ogHelp "$FUNCNAME": "$MSG_HELP_ogGetMountImageDir" \
		"$FUNCNAME image_name [ extension ]" \
		"$FUNCNAME Ubuntu12" \
		"$FUNCNAME Windows7 diff"
    return
fi


if [ $# -lt 1 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME image_name [ extension ]"
    return $?
fi


DIRMOUNT="mount/$1"
[ "$2" == "diff" ] && DIRMOUNT="$DIRMOUNT.diff"
echo "$DIRMOUNT"


}


#/**
#  ogWaitMountImage image_name extension imagen_size
#@brief Se espera un tiempo a que se monte la imagen en el servidor.
#@brief Cuando se esta creando la imagen hay que dar el tamaño, para que espere el tiempo de creación.	
#@param 1 Respositorio [ REPO | CACHE ]
#@param 2 Nombre Imagen 
#@param 3 Tipo imagen [ img | diff ] 
#@param 4 Tamaño imagen (opcional)
#*/
function ogWaitMountImage () {
local SIZE TIME DIRMOUNT TIMEOUT TIMEAUX

TIME=$SECONDS

# Ayuda o menos de 5 parametros y la imagen no es basica
if [ "$*" == "help" ]; then
        ogHelp 	"$FUNCNAME": "$MSG_HELP_ogWaitMountImage" \
		"$FUNCNAME [ REPO | CACHE ] image_name extension [ image_size ] " \
		"$FUNCNAME REPO Ubuntu12 img 30000000" \
		"$FUNCNAME CACHE Windows7 diff "
        return
fi

if [ $# -lt 2 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME [ REPO | CACHE ]  image_name extension [ image_size ] " 
    return $?
fi

SIZE=${4:-"300000"}

if [ "$1" == "CACHE" -o "$1" == "cache" ]; then
	DIRMOUNT="/tmp/$(ogGetMountImageDir "$2" $3)"
else
	DIRMOUNT="$OGIMG/$(ogGetMountImageDir "$2" $3)"
fi

echo -n -e "      $MSG_SYNC_SLEEP: $DIRMOUNT\n      #"  | tee -a $OGLOGSESSION $OGLOGFILE

# time-out segun el tamaño de la imagen. por defecto: 100000k -> 3s
let TIMEOUT=$SIZE/$CREATESPEED
[ $TIMEOUT -lt 60 ] && TIMEOUT=60
until [ -f "$DIRMOUNT/ogimg.info" ] ; do
        TIMEAUX=$[SECONDS-TIME]
        [  "$TIMEAUX" -lt "$TIMEOUT" ] || return $(ogRaiseError $OG_ERR_DONTMOUNT_IMAGE "$3 $4 $IMGEXT"; echo $?)
        echo -n "#"  | tee -a $OGLOGSESSION $OGLOGFILE
        sleep 5
done
echo  "" | tee -a $OGLOGSESSION $OGLOGFILE

}


function ogReduceImage () {
#/**
# ogReduceImage
#@brief   Reduce el archivo de la imagen a tamaño datos + 500M
#@param 1 Repositorio  [ REPO | CACHE ]
#@param 2 Nombre Imagen
#@param 3 Tipo Imagen [ img |diff ]
#@return 
#@exception OG_ERR_FORMAT     # 1 formato incorrecto.
#@exception OG_ERR_NOTFOUND        # 2 Fichero o dispositivo no encontrado.
#*/
local IMGEXT DIRMOUNT AVAILABLE USED IMGDIR IMGFILE ENDSIZE LOOPDEVICE
if [ "$*" == "help" ]; then
    ogHelp "$FUNCNAME": "$MSG_HELP_ogReduceImage" \
		"$FUNCNAME [ REPO|CACHE ] image_name [ extension ]" \
		"$FUNCNAME REPO Ubuntu12" \
		"$FUNCNAME CACHE Windows7 diff"
    return
fi

if [ $# -lt 2 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME [ REPO|CACHE ] image_name [ extension ]"
    return $?
fi


[ "$3" == "" -o "$3" == "img" ] && IMGEXT="img" || IMGEXT="img.diff"
IMGDIR="$(ogGetParentPath "$1" "/$2")"
IMGFILE="${IMGDIR}/$(basename "/$2").$IMGEXT"

# Solo es posible para imagenes EXT4. Si no es EXT4 salimos sin error
file "$IMGFILE" | grep -i " ext4 filesystem " 2>&1 > /dev/null || return 0

if [ "$1" == "CACHE" -o "$1" == "cache" ]; then
        # Al montar se comprueba la existencia de la imagen
        DIRMOUNT="$(ogMountImage $1 "$2" ${IMGEXT#*\.}) "

        AVAILABLE=$(df -k|grep "$DIRMOUNT$"|awk '{print $4}')
        # Si el espacio libre menor que 500Mb nos salimos
        if [ $AVAILABLE -lt 200000 ]; then
        	ogUnmountImage $1 "$2" ${IMGEXT#*\.}
	 	return 0
	fi


        # Calculamos la diferencia entre el tamaño interno y externo
        EXTSIZE=$(ls -l --block-size=1024 "$IMGFILE" | cut -f5 -d" ")
        INTSIZE=$(df -k|grep "$DIRMOUNT"|awk '{print $2}')
        let EDGESIZE=$EXTSIZE-$INTSIZE

        ogUnmountImage $1 "$2" ${IMGEXT#*\.}
	LOOPDEVICE=$(losetup -f)
	losetup $LOOPDEVICE "$IMGFILE"

	# Redimensiono sistema de ficheros
	echo "      resize2fs -fpM $LOOPDEVICE"
	resize2fs -fpM $LOOPDEVICE |tee -a $OGLOGCOMMAND
        ogMountImage $1 "$2" ${IMGEXT#*\.}

	# Calculamos el tamaño final del archivo
        INTSIZE=$(df -k|grep "$DIRMOUNT"|awk '{print $2}')
        let EXTSIZE=$INTSIZE+$EDGESIZE
        umount "$DIRMOUNT"

	# Si existe dispositivo de loop lo borro.
	[ $LOOPDEVICE ] && losetup -d $LOOPDEVICE

	# Corto el archivo al tamaño del sistema de ficheros.
        echo "      truncate --size=\"$EXTSIZE\"k $IMGFILE  "
        truncate --size="$EXTSIZE"k "$IMGFILE"

	# Desmonto y desbloqueo la imagen
	rm -f "$IMGFILE.lock"
	rmdir "$DIRMOUNT"

else
	[ -z $REPOIP ] && REPOIP=$(ogGetRepoIp)
        echo "      hose $REPOIP 2009 --out sh -c echo -ne REDUCE_IMAGE \"$2\" ${IMGEXT#*\.}"
        hose $REPOIP 2009 --out sh -c "echo -ne REDUCE_IMAGE \"$2\" ${IMGEXT#*\.}"
fi


}



function ogIsSyncImage () {
#/**
# ogIsSyncImage
#@brief   Comprueba si la imagen es sincronizable
#@param 1 Repositorio  [ REPO | CACHE ]
#@param 2 Nombre Imagen
#@param 3 Tipo Imagen [ img |diff ]
#@return 
#@exception OG_ERR_FORMAT     # 1 formato incorrecto.
#@exception OG_ERR_NOTFOUND        # 2 Fichero o dispositivo no encontrado.
#*/
local IMGEXT IMGDIR IMGFILE 

if [ "$*" == "help" ]; then
    ogHelp "$FUNCNAME": "$MSG_HELP_ogIsSyncImage" \
		"$FUNCNAME [ REPO|CACHE ] image_name [ extension ]" \
		"$FUNCNAME REPO Ubuntu12" \
		"$FUNCNAME CACHE Windows7 diff"
    return
fi

if [ $# -lt 2 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME [ REPO|CACHE ] image_name [ extension ]"
    return $?
fi

[ "$3" == "" -o "$3" == "img" ] && IMGEXT="img" || IMGEXT="img.diff"
IMGDIR="$(ogGetParentPath "$1" "/$2")"
IMGFILE="${IMGDIR}"/$(basename "/$2").$IMGEXT

file "$IMGFILE" | grep -i -e " BTRFS Filesystem " -e " ext4 filesystem " >/dev/null
[ $? -eq 0 ] && return 0 || return $OG_ERR_DONTSYNC_IMAGE 


}

#/**
# ogCheckSyncImage
#@brief   Muestra el contenido de la imagen para comprobarla.
#@param 1 Repositorio  [ REPO | CACHE ]
#@param 2 Nombre Imagen
#@param 3 Tipo Imagen [ img |diff ]
#@return 
#@exception OG_ERR_FORMAT     # 1 formato incorrecto.
#@exception OG_ERR_NOTFOUND        # 2 Fichero o dispositivo no encontrado.
#*/
ogCheckSyncImage (){
local IMGEXT  IMGDIR  IMGFILE DIRMOUNT

if [ "$*" == "help" ]; then
    ogHelp "$FUNCNAME":  "$MSG_HELP_ogCheckSyncImage" \
		"$FUNCNAME [ REPO|CACHE ] image_name [ extension ]" \
		"$FUNCNAME REPO Ubuntu12" \
		"$FUNCNAME CACHE Windows7 diff"
    return
fi

if [ $# -lt 2 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $FUNCNAME [ REPO|CACHE ] image_name [ extension ]"
    return $?
fi

[ "$3" == "" -o "$3" == "img" ] && IMGEXT="img" || IMGEXT="img.diff"
IMGDIR="$(ogGetParentPath "$1" "/$2")"
IMGFILE="${IMGDIR}/$(basename "/$2").$IMGEXT"

ogIsSyncImage $1 "$2" "${IMGEXT#*\.}" || return $(ogRaiseError $OG_ERR_DONTSYNC_IMAGE "$3 $4"; echo $?)

DIRMOUNT="/tmp/ogCheckImage$$"
mkdir "$DIRMOUNT"
# FS de la imagen segun la version del kernel: < 3.7 ext4, si >=  btrfs 
if [ $(uname -r|cut -d. -f2) -lt 7 ]; then
	mount -t ext4 -o loop "$IMGFILE" "$DIRMOUNT" 2>&1 | tee -a $OGLOGCOMMAND
	RETVAL=${PIPESTATUS[0]}
else
	mount -o compress=lzo  "$IMGFILE" "$DIRMOUNT" 1>/dev/null | tee -a $OGLOGCOMMAND
	RETVAL=${PIPESTATUS[0]}
fi
ls -C "$DIRMOUNT" | tee -a $OGLOGCOMMAND
umount "$DIRMOUNT" 

rmdir "$DIRMOUNT"
return $RETVAL
}