Rewrite to use getopt and return instead of exit, a seperate function (fs_link) does the actual work.
This commit is contained in:
Glenn L McGrath
parent
b520e083e0
commit
aa3908d1b7
152
coreutils/ln.c
152
coreutils/ln.c
@ -30,96 +30,82 @@
|
|||||||
#include <dirent.h>
|
#include <dirent.h>
|
||||||
#include <errno.h>
|
#include <errno.h>
|
||||||
|
|
||||||
static int symlinkFlag = FALSE;
|
#define LN_SYMLINK 1
|
||||||
static int removeoldFlag = FALSE;
|
#define LN_FORCE 2
|
||||||
static int followLinks = TRUE;
|
#define LN_NODEREFERENCE 4
|
||||||
|
|
||||||
|
/*
|
||||||
|
* linkDestName is where the link points to,
|
||||||
|
* linkSrcName is the name of the link to be created.
|
||||||
|
*/
|
||||||
|
static int fs_link(const char *link_DestName, const char *link_SrcName, const int flag)
|
||||||
|
{
|
||||||
|
int status;
|
||||||
|
int srcIsDir;
|
||||||
|
char *srcName;
|
||||||
|
|
||||||
|
if (link_DestName==NULL)
|
||||||
|
return(FALSE);
|
||||||
|
|
||||||
|
srcName = (char *) malloc(strlen(link_SrcName)+strlen(link_DestName)+1);
|
||||||
|
|
||||||
|
if (link_SrcName==NULL)
|
||||||
|
strcpy(srcName, link_DestName);
|
||||||
|
else
|
||||||
|
strcpy(srcName, link_SrcName);
|
||||||
|
|
||||||
|
if (flag&LN_NODEREFERENCE)
|
||||||
|
srcIsDir = isDirectory(srcName, TRUE, NULL);
|
||||||
|
else
|
||||||
|
srcIsDir = isDirectory(srcName, FALSE, NULL);
|
||||||
|
|
||||||
|
if ((srcIsDir==TRUE)&&((flag&LN_NODEREFERENCE)==0)) {
|
||||||
|
strcat(srcName, "/");
|
||||||
|
strcat(srcName, link_DestName);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (flag&LN_FORCE)
|
||||||
|
unlink(srcName);
|
||||||
|
|
||||||
|
if (flag&LN_SYMLINK)
|
||||||
|
status = symlink(link_DestName, srcName);
|
||||||
|
else
|
||||||
|
status = link(link_DestName, srcName);
|
||||||
|
|
||||||
|
if (status != 0) {
|
||||||
|
perror(srcName);
|
||||||
|
return(FALSE);
|
||||||
|
}
|
||||||
|
return(TRUE);
|
||||||
|
}
|
||||||
|
|
||||||
extern int ln_main(int argc, char **argv)
|
extern int ln_main(int argc, char **argv)
|
||||||
{
|
{
|
||||||
char *linkName, *dirName=NULL;
|
int flag = 0;
|
||||||
int linkIntoDirFlag;
|
int opt;
|
||||||
int stopIt = FALSE;
|
|
||||||
|
|
||||||
argc--;
|
|
||||||
argv++;
|
|
||||||
|
|
||||||
/* Parse any options */
|
/* Parse any options */
|
||||||
while (argc > 0 && stopIt == FALSE) {
|
while ((opt=getopt(argc, argv, "sfn")) != -1) {
|
||||||
if (**argv == '-') {
|
switch(opt) {
|
||||||
while (*++(*argv))
|
case 's':
|
||||||
switch (**argv) {
|
flag |= LN_SYMLINK;
|
||||||
case 's':
|
break;
|
||||||
symlinkFlag = TRUE;
|
case 'f':
|
||||||
break;
|
flag |= LN_FORCE;
|
||||||
case 'f':
|
break;
|
||||||
removeoldFlag = TRUE;
|
case 'n':
|
||||||
break;
|
flag |= LN_NODEREFERENCE;
|
||||||
case 'n':
|
break;
|
||||||
followLinks = FALSE;
|
default:
|
||||||
break;
|
usage(ln_usage);
|
||||||
case '-':
|
|
||||||
stopIt = TRUE;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
usage(ln_usage);
|
|
||||||
}
|
|
||||||
argc--;
|
|
||||||
argv++;
|
|
||||||
}
|
}
|
||||||
else
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
|
while(optind<(argc-1)) {
|
||||||
if (argc < 2) {
|
if (fs_link(argv[optind], argv[argc-1], flag)==FALSE)
|
||||||
usage(ln_usage);
|
return(FALSE);
|
||||||
|
optind++;
|
||||||
}
|
}
|
||||||
|
return(TRUE);
|
||||||
linkName = argv[argc - 1];
|
|
||||||
|
|
||||||
linkIntoDirFlag = isDirectory(linkName, followLinks, NULL);
|
|
||||||
if ((argc >= 3) && linkIntoDirFlag == FALSE) {
|
|
||||||
errorMsg(not_a_directory, linkName);
|
|
||||||
exit FALSE;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (linkIntoDirFlag == TRUE)
|
|
||||||
dirName = linkName;
|
|
||||||
|
|
||||||
while (argc-- >= 2) {
|
|
||||||
int status;
|
|
||||||
|
|
||||||
if (linkIntoDirFlag == TRUE) {
|
|
||||||
char *baseName = get_last_path_component(*argv);
|
|
||||||
linkName = (char *)xmalloc(strlen(dirName)+strlen(baseName)+2);
|
|
||||||
strcpy(linkName, dirName);
|
|
||||||
if(dirName[strlen(dirName)-1] != '/')
|
|
||||||
strcat(linkName, "/");
|
|
||||||
strcat(linkName,baseName);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (removeoldFlag == TRUE) {
|
|
||||||
status = (unlink(linkName) && errno != ENOENT);
|
|
||||||
if (status != 0) {
|
|
||||||
perror(linkName);
|
|
||||||
exit FALSE;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (symlinkFlag == TRUE)
|
|
||||||
status = symlink(*argv, linkName);
|
|
||||||
else
|
|
||||||
status = link(*argv, linkName);
|
|
||||||
if (status != 0) {
|
|
||||||
perror(linkName);
|
|
||||||
exit FALSE;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (linkIntoDirFlag == TRUE)
|
|
||||||
free(linkName);
|
|
||||||
|
|
||||||
argv++;
|
|
||||||
}
|
|
||||||
return( TRUE);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
|
|||||||
152
ln.c
152
ln.c
@ -30,96 +30,82 @@
|
|||||||
#include <dirent.h>
|
#include <dirent.h>
|
||||||
#include <errno.h>
|
#include <errno.h>
|
||||||
|
|
||||||
static int symlinkFlag = FALSE;
|
#define LN_SYMLINK 1
|
||||||
static int removeoldFlag = FALSE;
|
#define LN_FORCE 2
|
||||||
static int followLinks = TRUE;
|
#define LN_NODEREFERENCE 4
|
||||||
|
|
||||||
|
/*
|
||||||
|
* linkDestName is where the link points to,
|
||||||
|
* linkSrcName is the name of the link to be created.
|
||||||
|
*/
|
||||||
|
static int fs_link(const char *link_DestName, const char *link_SrcName, const int flag)
|
||||||
|
{
|
||||||
|
int status;
|
||||||
|
int srcIsDir;
|
||||||
|
char *srcName;
|
||||||
|
|
||||||
|
if (link_DestName==NULL)
|
||||||
|
return(FALSE);
|
||||||
|
|
||||||
|
srcName = (char *) malloc(strlen(link_SrcName)+strlen(link_DestName)+1);
|
||||||
|
|
||||||
|
if (link_SrcName==NULL)
|
||||||
|
strcpy(srcName, link_DestName);
|
||||||
|
else
|
||||||
|
strcpy(srcName, link_SrcName);
|
||||||
|
|
||||||
|
if (flag&LN_NODEREFERENCE)
|
||||||
|
srcIsDir = isDirectory(srcName, TRUE, NULL);
|
||||||
|
else
|
||||||
|
srcIsDir = isDirectory(srcName, FALSE, NULL);
|
||||||
|
|
||||||
|
if ((srcIsDir==TRUE)&&((flag&LN_NODEREFERENCE)==0)) {
|
||||||
|
strcat(srcName, "/");
|
||||||
|
strcat(srcName, link_DestName);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (flag&LN_FORCE)
|
||||||
|
unlink(srcName);
|
||||||
|
|
||||||
|
if (flag&LN_SYMLINK)
|
||||||
|
status = symlink(link_DestName, srcName);
|
||||||
|
else
|
||||||
|
status = link(link_DestName, srcName);
|
||||||
|
|
||||||
|
if (status != 0) {
|
||||||
|
perror(srcName);
|
||||||
|
return(FALSE);
|
||||||
|
}
|
||||||
|
return(TRUE);
|
||||||
|
}
|
||||||
|
|
||||||
extern int ln_main(int argc, char **argv)
|
extern int ln_main(int argc, char **argv)
|
||||||
{
|
{
|
||||||
char *linkName, *dirName=NULL;
|
int flag = 0;
|
||||||
int linkIntoDirFlag;
|
int opt;
|
||||||
int stopIt = FALSE;
|
|
||||||
|
|
||||||
argc--;
|
|
||||||
argv++;
|
|
||||||
|
|
||||||
/* Parse any options */
|
/* Parse any options */
|
||||||
while (argc > 0 && stopIt == FALSE) {
|
while ((opt=getopt(argc, argv, "sfn")) != -1) {
|
||||||
if (**argv == '-') {
|
switch(opt) {
|
||||||
while (*++(*argv))
|
case 's':
|
||||||
switch (**argv) {
|
flag |= LN_SYMLINK;
|
||||||
case 's':
|
break;
|
||||||
symlinkFlag = TRUE;
|
case 'f':
|
||||||
break;
|
flag |= LN_FORCE;
|
||||||
case 'f':
|
break;
|
||||||
removeoldFlag = TRUE;
|
case 'n':
|
||||||
break;
|
flag |= LN_NODEREFERENCE;
|
||||||
case 'n':
|
break;
|
||||||
followLinks = FALSE;
|
default:
|
||||||
break;
|
usage(ln_usage);
|
||||||
case '-':
|
|
||||||
stopIt = TRUE;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
usage(ln_usage);
|
|
||||||
}
|
|
||||||
argc--;
|
|
||||||
argv++;
|
|
||||||
}
|
}
|
||||||
else
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
|
while(optind<(argc-1)) {
|
||||||
if (argc < 2) {
|
if (fs_link(argv[optind], argv[argc-1], flag)==FALSE)
|
||||||
usage(ln_usage);
|
return(FALSE);
|
||||||
|
optind++;
|
||||||
}
|
}
|
||||||
|
return(TRUE);
|
||||||
linkName = argv[argc - 1];
|
|
||||||
|
|
||||||
linkIntoDirFlag = isDirectory(linkName, followLinks, NULL);
|
|
||||||
if ((argc >= 3) && linkIntoDirFlag == FALSE) {
|
|
||||||
errorMsg(not_a_directory, linkName);
|
|
||||||
exit FALSE;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (linkIntoDirFlag == TRUE)
|
|
||||||
dirName = linkName;
|
|
||||||
|
|
||||||
while (argc-- >= 2) {
|
|
||||||
int status;
|
|
||||||
|
|
||||||
if (linkIntoDirFlag == TRUE) {
|
|
||||||
char *baseName = get_last_path_component(*argv);
|
|
||||||
linkName = (char *)xmalloc(strlen(dirName)+strlen(baseName)+2);
|
|
||||||
strcpy(linkName, dirName);
|
|
||||||
if(dirName[strlen(dirName)-1] != '/')
|
|
||||||
strcat(linkName, "/");
|
|
||||||
strcat(linkName,baseName);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (removeoldFlag == TRUE) {
|
|
||||||
status = (unlink(linkName) && errno != ENOENT);
|
|
||||||
if (status != 0) {
|
|
||||||
perror(linkName);
|
|
||||||
exit FALSE;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (symlinkFlag == TRUE)
|
|
||||||
status = symlink(*argv, linkName);
|
|
||||||
else
|
|
||||||
status = link(*argv, linkName);
|
|
||||||
if (status != 0) {
|
|
||||||
perror(linkName);
|
|
||||||
exit FALSE;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (linkIntoDirFlag == TRUE)
|
|
||||||
free(linkName);
|
|
||||||
|
|
||||||
argv++;
|
|
||||||
}
|
|
||||||
return( TRUE);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user