buutil.c

一套客户/服务器模式的备份系统代码,跨平台,支持linux,AIX, IRIX, FreeBSD, Digital Unix (OSF1), Solaris and HP-UX.

/****************** Start of $RCSfile: buutil.c,v $  ****************
*
* $Source: /home/alb/afbackup/afbackup-3.3.8.1/RCS/buutil.c,v $
* $Id: buutil.c,v 1.5 2005/01/15 08:49:21 alb Exp alb $
* $Date: 2005/01/15 08:49:21 $
* $Author: alb $
*
*
******* description ***********************************************
*
*
*
*******************************************************************/

#include <conf.h>
#include <version.h>

  static char * fileversion = "$RCSfile: buutil.c,v $ $Source: /home/alb/afbackup/afbackup-3.3.8.1/RCS/buutil.c,v $ $Id: buutil.c,v 1.5 2005/01/15 08:49:21 alb Exp alb $ " PACKAGE " " VERSION_STRING;

#include <stdio.h>
#include <string.h>
#include <fcntl.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdarg.h>
#include <errno.h>
#ifdef	HAVE_SYS_TIME_H
#include <sys/time.h>
#endif
#ifdef	HAVE_SYS_WAIT_H
#include <sys/wait.h>
#endif
#ifdef	HAVE_SYS_RESOURCE_H
#include <sys/resource.h>
#endif
#include <sys/types.h>
#include <signal.h>
#ifdef	HAVE_SYS_TIME_H
#ifdef	TIME_WITH_SYS_TIME
#include <time.h>
#endif
#endif
#include <x_types.h>
#include <fileutil.h>
#include <sysutils.h>
#include <netutils.h>
#include <genutils.h>
#include <x_regex.h>

#define DEBUGFILE	"/tmp/afserverlog"

#define	CLEANUP		{ goto cleanup; }
#define	GETOUT		{ goto getout; }
#define	CLEANUPR(rval)	{ r = rval; goto cleanup; }
#define	GETOUTR(rval)	{ r = rval; goto getout; }

#include "backup.h"

int
connect_afbu_server(
  UChar		*servername,
  UChar		*servname,
  Int32		portnum,
  Int32		timeout)
{
  int		sockfd;
  time_t	starttime;
  Flag		uxsock = NO;

  if(servname)
    if(FN_ISPATH(servname))
	uxsock = YES;

  starttime = time(NULL);

  forever{
    if(uxsock)
	sockfd = open_uxsock_conn(servname);
    else
	sockfd = open_tcpip_conn(servername, servname, portnum);
    if(sockfd >= 0)
	break;

    if(sockfd < 0 && time(NULL) - starttime > timeout)
	return(sockfd);

    ms_sleep(drandom() * 400 + 100);
  }

  set_ip_throughput(sockfd);

  set_socket_keepalive(sockfd);

  return(sockfd);
}

Int32
rewrite_tape_label(
  UChar		*label,
  Int32		new_cartnum,
  UChar		*oldserver,
  Int32		oldport)
{
  UChar		new_buffer[1000];	/* label is <= 512 bytes */
  UChar		*cptr, *cptr2;
  Int32		n;
  int		org_cartno;

  if(!strstr(label, INFOHEADER))
    return(1);

  cptr = strstr(label, CARTNOTEXT);
  if(!cptr)
    return(1);

  if(sscanf(cptr + strlen(CARTNOTEXT), "%d", &org_cartno) < 1)
    return(1);
  strncpy(new_buffer, label, n = (cptr - label) + strlen(CARTNOTEXT));
  sprintf(new_buffer + n, "%d\n", (int) new_cartnum);
  cptr2 = strchr(cptr, '\n');
  if(cptr2)
    strcpy(new_buffer + strlen(new_buffer), cptr2 + 1);
  else
    return(1);

  strcpy(label, new_buffer);

  if( !(cptr = strstr(label, CART2NOTEXT)) ){
    cptr = strstr(label, CARTNOTEXT);
    cptr2 = strchr(cptr, '\n');
    if(cptr2){
	strncpy(new_buffer, label, n = cptr2 - label);
	new_buffer[n] = '\0';
    }
    else{
	strcpy(new_buffer, label);
	strcat(new_buffer, "\n\n");
    }
    sprintf(new_buffer + strlen(new_buffer), "%s%d\n\n%s%s%s%d\n",
				CART2NOTEXT, org_cartno, COPYFROMTEXT,
				oldserver, PORTSEP, (int) oldport);
    if(cptr2)
	strcpy(new_buffer + strlen(new_buffer), cptr2 + 1);
  }
#if 0 /* if there is a secondary label, leave it untouched (recursive copies) */
  else{
    n = (cptr - label) + strlen(CART2NOTEXT);
    strncpy(new_buffer, label, n);
    sprintf(new_buffer + n, "%d\n\n%s%s%s%d\n", org_cartno, COPYFROMTEXT,
					oldserver, PORTSEP, (int) oldport);
    n = strlen(new_buffer);
    if( (cptr2 = strchr(cptr, '\n')) )
	strcpy(new_buffer + n, cptr2 + 1);
  }
#endif

  strcpy(label, new_buffer);

  if( (cptr = strstr(label, CDATETEXT)) ){
    n = (cptr - label) + strlen(CDATETEXT);
    strncpy(new_buffer, label, n);
    sprintf(new_buffer + n, "%s\n", actimestr());
    n = strlen(new_buffer);
    if( (cptr2 = strchr(cptr, '\n')) )
	strcpy(new_buffer + n, cptr2 + 1);
  }
  else if( (cptr = strstr(label, DATETEXT)) ){
    cptr2 = strchr(cptr, '\n');
    if(cptr2){
	strncpy(new_buffer, label, n = cptr2 - label);
	new_buffer[n + 1] = '\0';
    }
    else{
	strcpy(new_buffer, label);
	strcat(new_buffer, "\n");
    }
    sprintf(new_buffer + strlen(new_buffer), "\n%s%s\n",
				CDATETEXT, actimestr());
    if(cptr2)
	strcpy(new_buffer + strlen(new_buffer), cptr2 + 1);
  }
  else{
    cptr = strstr(label, INFOHEADER);
    strncpy(new_buffer, label, n = (cptr - label) + strlen(INFOHEADER));
    sprintf(new_buffer + n, "%s%s\n\n%s%s\n\n",
			DATETEXT, actimestr(), CDATETEXT, actimestr());
    strcat(new_buffer, cptr + strlen(INFOHEADER));
  }

  strcpy(label, new_buffer);

  return(0);
}

Int32
save_bytes_per_tape(
  UChar		*filen,
  Int32		cart,
  Real64	numbytes,
  Int32		numfiles,
  Flag		tape_full,
  time_t	the_time)
{
  UChar		dbuf[32], tbuf[32], vbuf[128], kbuf[32];

  Real64_to_intstr(numbytes, dbuf);
  time_t_to_intstr(the_time, tbuf);

  sprintf(vbuf, "%s %d %d %s", dbuf, (int) numfiles,
				tape_full ? 1 : 0, tbuf);
  sprintf(kbuf, "%d:", (int) cart);

  return(kfile_insert(filen, kbuf, vbuf, KFILE_SORTN | KFILE_LOCKED));
}

Int32
get_bytes_per_tape(
  UChar		*filen,
  Int32		cart,
  Real64	*numbytes,
  Int32		*numfiles,
  Flag		*tape_full,
  time_t	*last_wrtime)
{
  UChar		*cptr, kbuf[32];
  int		i1, i2, n;
  double	d1;
  time_t	t;
  Flag		got_entry = NO;

  sprintf(kbuf, "%d:", (int) cart);
  cptr = kfile_get(filen, kbuf, KFILE_LOCKED);
  if(cptr){
    i1 = i2 = -1;
    if(sscanf(cptr, "%lf%d%d%n", &d1, &i1, &i2, &n) >= 1){
	if(numbytes)
	  *numbytes = (Real64) d1;
	if(i1 != -1 && numfiles)
	  *numfiles = (Int32) i1;
	if(i2 != -1){
	  if(tape_full)
	    *tape_full = (i2 ? YES : NO);
	  if(last_wrtime){
	    *last_wrtime = (time_t) 0;
	    if((t = strint2time(cptr + n)) != UNSPECIFIED_TIME)
		*last_wrtime = t;
	  }
	}

	got_entry = YES;
    }
    free(cptr);
  }

  if(!got_entry){
    if(numbytes)
	*numbytes = 0.0;
    if(numfiles)
	*numfiles = 0;
    if(tape_full)
	*tape_full = NO;
    if(last_wrtime)
	*last_wrtime = (time_t) 0;
  }

  return(0);
}

Int32
get_tape_usages(
  UChar		*filen,
  TapeUsage	**usagesp,
  Int32		*num_usages)
{
  UChar		*cptr;
  int		i1, i2, n;
  double	d1;
  Int32		i, r = 0, num_specs, num_entries = 0;
  TapeUsage	*usages;
  time_t	t;
  KeyValuePair	*allentries = NULL;

  allentries = kfile_getall(filen, &num_entries, KFILE_LOCKED);
  if(!allentries && !access(filen, F_OK) && access(filen, R_OK))
    return(-1);

  usages = NEWP(TapeUsage, num_entries + 1);
  if(! usages)
	GETOUTR(FATAL_ERROR);

  memset(usages, 0, sizeof(usages[0]) * (num_entries + 1));
  *usagesp = usages;
  num_specs = 0;

  if(num_entries > 0){
    for(i = 0; i < num_entries; i++){
      if(sscanf(allentries[i].key, "%d%n", &i1, &n) > 0){
	cptr = first_nospace(allentries[i].key + n);
	if(*cptr != ':')
	  continue;

	usages[num_specs].tape_num = i1;

	i1 = i2 = -1;
	if(sscanf(allentries[i].value, "%lf%d%d%n", &d1, &i1, &i2, &n) > 0){
	  usages[num_specs].bytes_on_tape = (Real64) d1;

	  if(i1 != -1)
	    usages[num_specs].files_on_tape = (Int32) i1;

	  usages[num_specs].tape_full = DONT_KNOW;
	  if(i2 != -1){
	    usages[num_specs].tape_full = (i2 ? YES : NO);

	    usages[num_specs].last_wrtime = (time_t) 0;
	    if((t = strint2time(allentries[i].value + n)) != UNSPECIFIED_TIME)
		usages[num_specs].last_wrtime = t;
	  }
	}

	num_specs++;
      }
    }
  }

  if(num_usages)
    *num_usages = num_specs;

  memset(usages + num_specs, 0, sizeof(usages[0]));

 cleanup:
  kfile_freeall(allentries, num_entries);

  return(r);

 getout:
  CLEANUP;
}

Int32
incr_tape_use_count(UChar * filen, Int32 cart)
{
  UChar		vbuf[32], kbuf[32], *cptr;
  Int32		n = 0;

  sprintf(kbuf, "%d:", (int) cart);
  cptr = kfile_get(filen, kbuf, KFILE_LOCKED);
  if(cptr){
    strint(cptr, &n);
    free(cptr);
  }

  sprintf(vbuf, "%d", (int) n + 1);

  return(kfile_insert(filen, kbuf, vbuf, KFILE_SORTN | KFILE_LOCKED));
}

Int32
host_in_list(UChar * host, UChar * hostlist)
{
  Int32	r = 0;
  UChar	*line = NULL;
  FILE	*fp;

  if(!hostlist)
    return(1);
  if(!hostlist[0])
    return(1);

  if(FN_ISABSPATH(hostlist)){
    fp = fopen(hostlist, "r");
    if(!fp)
	return(-2);

    while(!feof(fp)){
	ZFREE(line);
	line = fget_alloc_str(fp);
	if(!line)
	  continue;

	massage_string(line);

	if(same_host(line, host) > 0){
	  r = 1;
	  break;
	}
    }
    ZFREE(line);
    fclose(fp);
  }
  else if(hostlist[0] == '|'){
    int	pid, pst, inp[2];

    line = repl_substring(hostlist + 1, "%H", host);
    if(!line)
	return(-3);
    if(pipe(inp))
	return(-5);

    pid = fork_forced();

    if(pid == 0){
	dup2(inp[0], 0);
	close(inp[0]);
	close(inp[1]);

	pst = open(NULLFILE, O_WRONLY);
	if(pst >= 0){
	  dup2(pst, 1);
	  dup2(pst, 2);
	  close(pst);
	}

	execl("/bin/sh", "/bin/sh", "-c", line, NULL);
	free(line);
	return(-5);
    }

    close(inp[0]);
    if(pid > 0){
	struct sigaction	siga, osiga;

	SETZERO(siga);
	siga.sa_handler = SIG_IGN;
#ifdef	SA_RESTART
	siga.sa_flags = SA_RESTART;
#endif
	sigaction(SIGPIPE, &siga, &osiga);
	
	write_forced(inp[1], host, strlen(host));
	write_forced(inp[1], "\n", 1);

	sigaction(SIGPIPE, &osiga, NULL);
    }
    close(inp[1]);

    free(line);

    if(pid < 0)
	return(-4);

    waitpid_forced(pid, &pst, 0);

    r = (WEXITSTATUS(pst) ? 0 : 1);
  }
  else{
    forever{
	line = strchr(hostlist, ',');
	if(!line)
	  line = strchr(hostlist, ';');
	if(!line)
	  line = strchr(hostlist, ':');

	if(line)
	  *line = '\0';

	if(same_host(host, hostlist) > 0)
	  r = 1;

	if(line){
	  *line = ',';
	  hostlist = line + 1;
	}
	else
	  break;
    }
  }

  return(r);
}