kexec-tools/kcp.c

183 lines
3.8 KiB
C
Raw Normal View History

2006-07-20 03:36:18 +00:00
/* Cheap program to make a variable directory first before copying contents.
* Main use is for kdump because nash does not support variables, hence a
* command sequence like the following does not work:
* date=`date ...`
* mkdir -p /x/y/$date
* cp foo /x/y/$date/bar
*
* Don Zickus (dzickus@redhat.com)
*
* Copyright 2006 Red Hat Software
*
* This software may be freely redistributed under the terms of the GNU
* General Public License, version 2.
*
* 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., 675 Mass Ave, Cambridge, MA 02139, USA.
*
*/
#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <strings.h>
#include <string.h>
/* grab the local time and replace the %DATE var with it */
char * xlate_time(const char *dst)
{
struct tm *lclnow;
time_t now;
char *new_dst,*top;
int x;
//get the time
if ((top=(char *)malloc(256)) == NULL)
return NULL;
if ((now = time(NULL)) < 1)
return NULL;
if ((lclnow = localtime(&now)) == NULL)
return NULL;
//copy the easy stuff
new_dst=top;
while (*dst && (*dst != '%')) *new_dst++ = *dst++;
//translate the date part
//we output Year-Month-Day-Hour:Minute
if (*dst == '%'){
x = sprintf(new_dst,"%d-%02d-%02d-%02d:%02d", lclnow->tm_year+1900,
lclnow->tm_mon+1, lclnow->tm_mday, lclnow->tm_hour,
lclnow->tm_min);
new_dst += x;
//finish the copy
dst += 5; //skip over %DATE
while (*dst) *new_dst++ = *dst++;
}
*new_dst='\0';
return top;
}
void usage(int rc)
{
printf("usage: kcp source dest\n");
printf(" kcp --ssh src user@host:/dst\n");
printf(" kcp --local src dst\n");
2006-07-20 03:36:18 +00:00
printf("Will translate any %%DATE command properly\n");
printf("in the 'dest' variable\n");
exit(rc);
}
int main(int argc, char *argv[])
{
char *src, *dst, *new_dst, *ptr;
char *path;
2006-07-20 03:36:18 +00:00
int using_ssh=0;
char *login;
int status;
pid_t child;
2006-07-20 03:36:18 +00:00
if (argc < 4)
2006-07-20 03:36:18 +00:00
usage(1);
src = argv[2];
dst = argv[3];
2006-07-20 03:36:18 +00:00
if ((new_dst=xlate_time(dst)) == NULL){
printf("Failed to translate time\n");
exit(1);
}
if (!strcmp(argv[1], "--ssh"))
using_ssh =1;
2006-07-20 03:36:18 +00:00
/*
* Now that we have called xlate_time, new_dst
* holds the expanded ssh destination
*/
if (using_ssh) {
login=strdup(new_dst);
ptr=index(login, ':');
*ptr++='\0';
path = ptr;
} else {
login = NULL;
path = new_dst;
}
2006-07-20 03:36:18 +00:00
/*
*this makes our target directory
*/
if ((child = fork()) == 0) {
/*
* child
*/
if (using_ssh) {
if (execlp("ssh", "ssh", "-q", "-o", "BatchMode=yes", "-o",
"StrictHostKeyChecking=no", login, "mkdir", "-p",
path, NULL) < 0) {
perror("Failed to run ssh");
exit(1);
}
} else {
if (execlp("mkdir", "mkdir", "-p", path, NULL) < 0) {
perror("Failed to run mkdir");
exit(1);
}
2006-07-20 03:36:18 +00:00
}
} else {
/*
* parent
*/
if (child < 0) {
perror("Could not fork");
2006-07-20 03:36:18 +00:00
exit(1);
}
wait(&status);
if (WEXITSTATUS(status) != 0) {
printf ("%s exited abnormally: error = %d\n",
using_ssh ? "ssh":"mkdir", WEXITSTATUS(status));
exit(1);
2006-07-20 03:36:18 +00:00
}
}
/*
* now that we have our directory, lets copy everything over
* Note that scp can be used for local copies as well
*/
if ((child = fork()) == 0) {
/*need to include login info if scp to remote host*/
if (using_ssh)
path=new_dst;
src, path;
if (execlp("scp", "scp", "-q", "-o", "BatchMode=yes", "-o",
"StrictHostKeyChecking=no", src, path, NULL) < 0) {
perror("Failed to run scp\n");
2006-07-20 03:36:18 +00:00
exit(1);
}
} else {
if (child < 0) {
perror("Could not fork");
exit(1);
}
wait(&status);
if (WEXITSTATUS(status) != 0) {
printf("scp exited abnormally: error = %d\n",
WEXITSTATUS(status));
2006-07-20 03:36:18 +00:00
exit(1);
}
}
exit(0);
2006-07-20 03:36:18 +00:00
}