update ls

This commit is contained in:
Freya Murphy 2023-05-16 14:00:22 -04:00
parent 04150a407f
commit 6458a0f9e0
2 changed files with 427 additions and 207 deletions

View file

@ -9,9 +9,11 @@
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <errno.h> #include <errno.h>
#include <sys/stat.h>
#include <unistd.h> #include <unistd.h>
#include <sys/ioctl.h> #include <sys/ioctl.h>
/* quirky colors */
#define FILE_COLOR ANSCII BLACK COLOR #define FILE_COLOR ANSCII BLACK COLOR
#define DIR_COLOR ANSCII BOLD NEXT NORMAL BLUE COLOR #define DIR_COLOR ANSCII BOLD NEXT NORMAL BLUE COLOR
#define DIR_COLOR_EXEC ANSCII BACKGROUND GREEN NEXT NORMAL BLACK COLOR #define DIR_COLOR_EXEC ANSCII BACKGROUND GREEN NEXT NORMAL BLACK COLOR
@ -22,36 +24,49 @@
#define BLK_COLOR ANSCII BOLD NEXT NORMAL YELLOW COLOR #define BLK_COLOR ANSCII BOLD NEXT NORMAL YELLOW COLOR
#define SOCK_COLOR ANSCII BOLD NEXT NORMAL MAGENTA COLOR #define SOCK_COLOR ANSCII BOLD NEXT NORMAL MAGENTA COLOR
/**
* Describes when to do something
*/
enum When { enum When {
YES, YES, /* always */
NO, NO, /* never */
AUTO AUTO /* when not tty */
}; };
/**
* Flags that are to be used with ls
*/
static struct { static struct {
bool hidden; bool hidden; /* -a */
bool hide_dot; bool hide_dot; /* -A */
bool more_info; bool more_info; /* -l */
bool one_column; bool one_column; /* -1 */
bool recurse; bool recurse; /* -R */
enum When colored; bool octets; /* -o */
enum When colored; /* --color=never/no/yes/always/auto */
} flags; } flags;
/**
* Keeps track of the files data
*/
struct FileInfo { struct FileInfo {
struct passwd* usr; struct passwd* usr; /* the files owner user */
struct group* grp; struct group* grp; /* the files owner group */
char* name; char* name; /* the file name */
char date[13]; char date[13]; /* the files romanized date */
char mode[11]; char mode[11]; /* the files mode romanized or octets */
char size[5]; char size[5]; /* the human readable file size */
int links; int links; /* the amount of inode links */
int bytes; int bytes; /* the amount of bytes in the file*/
bool set_uid; bool set_uid; /* if the file is set uid */
bool set_gid; bool set_gid; /* if the file is set gid */
bool exec; bool exec; /* if the binary is executable by the user running ls */
unsigned char type; unsigned char type; /* the type of the file */
}; };
/**
* Keeps track of the biggest data to make the output data look pretty
*/
struct FileListInfo { struct FileListInfo {
int max_link; int max_link;
int max_usr; int max_usr;
@ -62,6 +77,11 @@ struct FileListInfo {
int total_size; int total_size;
}; };
/**
* Gets a directory and fails if it cannot be opened
* @param path the path to the directory
* @returns the directory if found or NULL if not
*/
static DIR* get_directory(char* path) { static DIR* get_directory(char* path) {
DIR* d = opendir(path); DIR* d = opendir(path);
if (d == NULL) { if (d == NULL) {
@ -74,6 +94,94 @@ static DIR* get_directory(char* path) {
return d; return d;
} }
/**
* Takes in a file and parses its mode to be humand readable
* @param info the file infomation to store the mode into
* @param mode the octet form of the mode
* @param type the type of the file
*/
static void parse_file_mode(struct FileInfo* info, mode_t mode, int type) {
switch (type) {
case DT_BLK:
info->mode[0] = 'b'; /* block device */
break;
case DT_CHR:
info->mode[0] = 'c'; /* something i dunno */
break;
case DT_DIR:
info->mode[0] = 'd'; /* directory */
break;
case DT_FIFO:
info->mode[0] = 'f'; /* fifo */
break;
case DT_LNK:
info->mode[0] = 'l'; /* symlink */
break;
case DT_SOCK:
info->mode[0] = 's'; /* socket */
break;
case DT_UNKNOWN:
info->mode[0] = 'u'; /* unkown (eye emoji) */
break;
case DT_WHT:
info->mode[0] = 'w'; /* tf, wheat? idk */
break;
default:
info->mode[0] = '-'; /* regular file */
break;
}
/* parse info for the user bits */
info->mode[1] = (mode & S_IRUSR) ? 'r' : '-';
info->mode[2] = (mode & S_IWUSR) ? 'w' : '-';
if (mode & S_IXUSR && mode & S_ISUID) {
info->mode[3] = 's';
} else if (mode & S_ISUID) {
info->mode[3] = 'S';
} else if (mode & S_IXUSR) {
info->mode[3] = 'x';
} else {
info->mode[3] = '-';
}
/* parse info for the group bits */
info->mode[4] = (mode & S_IRGRP) ? 'r' : '-';
info->mode[5] = (mode & S_IWGRP) ? 'w' : '-';
if (mode & S_IXGRP && mode & S_ISGID) {
info->mode[6] = 's';
} else if (mode & S_ISGID) {
info->mode[6] = 'S';
} else if (mode & S_IXGRP) {
info->mode[6] = 'x';
} else {
info->mode[6] = '-';
}
/* parse info for the other bits */
info->mode[7] = (mode & S_IROTH) ? 'r' : '-';
info->mode[8] = (mode & S_IWOTH) ? 'w' : '-';
if (mode & S_IXOTH && mode & S_ISVTX) {
info->mode[9] = 't';
} else if (mode & S_ISVTX) {
info->mode[9] = 'T';
} else if (mode & S_IXOTH) {
info->mode[9] = 'x';
} else {
info->mode[9] = '-';
}
/* make a null terminated string */
info->mode[10] = '\0';
}
/**
* Gets information about a given file_name in a directory and
* stores it into a fileinfo struct
* @param file_name the name of the file in the directory currently
* stored in the path buffer
* @param info the file info struct to store the data into
* @returns if successfull in reading the file
*/
static bool get_file_info(const char* file_name, struct FileInfo* info) { static bool get_file_info(const char* file_name, struct FileInfo* info) {
uid_t uid; uid_t uid;
@ -85,412 +193,507 @@ static bool get_file_info(const char* file_name, struct FileInfo* info) {
uid = getuid(); uid = getuid();
gid = getgid(); gid = getgid();
/* default to all false */
memset(&s, 0, sizeof(struct stat)); memset(&s, 0, sizeof(struct stat));
/* push the file to the path buffer to stat */
save = push_path_buffer(file_name); save = push_path_buffer(file_name);
/* get file stats */
if (lstat(get_path_buffer(), &s) < 0) { if (lstat(get_path_buffer(), &s) < 0) {
error_s("failed to read file '%s'", get_path_buffer()); error_s("failed to read file '%s'", get_path_buffer());
pop_path_buffer(save); pop_path_buffer(save);
return false; return false;
} }
/* get file type */
ty = (s.st_mode & S_IFMT) >> 12; ty = (s.st_mode & S_IFMT) >> 12;
info->set_uid = false;
info->set_gid = false;
info->exec = false;
switch (ty) { /* parse long list mode as octets or human readable */
case DT_BLK: if (flags.octets) {
info->mode[0] = 'b'; info->mode[0] = '0' + (s.st_mode / (8 * 8 * 8) % 8);
break; info->mode[1] = '0' + (s.st_mode / (8 * 8 * 1) % 8);
case DT_CHR: info->mode[2] = '0' + (s.st_mode / (8 * 1 * 1) % 8);
info->mode[0] = 'c'; info->mode[3] = '0' + (s.st_mode / (1 * 1 * 1) % 8);
break; info->mode[4] = '\0';
case DT_DIR:
info->mode[0] = 'd';
break;
case DT_FIFO:
info->mode[0] = 'f';
break;
case DT_LNK:
info->mode[0] = 'l';
break;
case DT_SOCK:
info->mode[0] = 's';
break;
case DT_UNKNOWN:
info->mode[0] = 'u';
break;
case DT_WHT:
info->mode[0] = 'w';
break;
default:
info->mode[0] = '-';
break;
}
info->mode[1] = (s.st_mode & S_IRUSR) ? 'r' : '-';
info->mode[2] = (s.st_mode & S_IWUSR) ? 'w' : '-';
if (s.st_mode & S_IXUSR && s.st_mode & S_ISUID) {
info->mode[3] = 's';
info->set_uid = true;
if (!info->exec) info->exec = info->usr->pw_uid == uid;
} else if (s.st_mode & S_ISUID) {
info->mode[3] = 'S';
info->set_uid = true;
} else if (s.st_mode & S_IXUSR) {
info->mode[3] = 'x';
if (!info->exec) info->exec = info->usr->pw_uid == uid;
} else { } else {
info->mode[3] = '-'; parse_file_mode(info, s.st_mode, ty);
} }
info->mode[4] = (s.st_mode & S_IRGRP) ? 'r' : '-'; /* check if the file is executable by the user */
info->mode[5] = (s.st_mode & S_IWGRP) ? 'w' : '-'; if (
if (s.st_mode & S_IXGRP && s.st_mode & S_ISGID) { (s.st_mode & S_IXUSR && s.st_uid == uid) ||
info->mode[6] = 's'; (s.st_mode & S_IXGRP && s.st_gid == gid) ||
info->set_gid = true; s.st_mode & S_IXOTH
if (!info->exec) info->exec = info->grp->gr_gid == gid; ) {
} else if (s.st_mode & S_ISGID) {
info->mode[6] = 'S';
info->set_gid = true;
} else if (s.st_mode & S_IXGRP) {
info->mode[6] = 'x';
if (!info->exec) info->exec = info->grp->gr_gid == gid;
} else {
info->mode[6] = '-';
}
info->mode[7] = (s.st_mode & S_IROTH) ? 'r' : '-';
info->mode[8] = (s.st_mode & S_IWOTH) ? 'w' : '-';
if (s.st_mode & S_IXOTH && s.st_mode & S_ISVTX) {
info->mode[9] = 't';
info->set_gid = true;
info->exec = true;
} else if (s.st_mode & S_ISVTX) {
info->mode[9] = 'T';
info->set_gid = true;
} else if (s.st_mode & S_IXOTH) {
info->mode[9] = 'x';
info->exec = true; info->exec = true;
} else { } else {
info->mode[9] = '-'; info->exec = false;
} }
/* get setuid and setgid bits */
info->set_uid = s.st_mode & S_ISUID;
info->set_gid = s.st_mode & S_ISGID;
info->mode[10] = '\0'; /* get user information */
info->usr = getpwuid(s.st_uid); info->usr = getpwuid(s.st_uid);
if (info->usr == NULL) { if (info->usr == NULL) {
error_s("failed to get user from `%s`\n", get_path_buffer()); error_s("failed to get user id: `%s`", get_path_buffer());
pop_path_buffer(save); pop_path_buffer(save);
return false; return false;
} }
/* get group information */
info->grp = getgrgid(s.st_gid); info->grp = getgrgid(s.st_gid);
if (info->grp == NULL) { if (info->grp == NULL) {
error_s("failed to get user from `%s`\n", get_path_buffer()); error_s("failed to get user id: `%s`", get_path_buffer());
pop_path_buffer(save); pop_path_buffer(save);
return false; return false;
} }
/* update inode links and file type */
info->links = s.st_nlink; info->links = s.st_nlink;
info->type = ty; info->type = ty;
/* copy the file name into the info */
file_len = strlen(file_name) + 1; file_len = strlen(file_name) + 1;
info->name = xalloc(file_len); info->name = xalloc(file_len);
memcpy(info->name, file_name, file_len); memcpy(info->name, file_name, file_len);
/* get the date and time for the file */
print_file_size(s.st_size, info->size); print_file_size(s.st_size, info->size);
print_date_time(s.st_mtim.tv_sec + s.st_mtim.tv_nsec / 1000000000, info->date); print_date_time(s.st_mtim.tv_sec + s.st_mtim.tv_nsec / 1000000000, info->date);
/* convert the bytes into human readable */
info->bytes = (s.st_size + s.st_blksize - 1) / s.st_blksize; info->bytes = (s.st_size + s.st_blksize - 1) / s.st_blksize;
/* cleanup */
pop_path_buffer(save); pop_path_buffer(save);
return true; return true;
} }
/**
* Gets the file color from its mode
* @pram info the file info to get the color from
* @return the anscii escape code for the file color
*/
static char* get_file_color(struct FileInfo* info) { static char* get_file_color(struct FileInfo* info) {
char* color; char* color;
if (info->type == DT_DIR) { if (info->type == DT_DIR) {
if (info->mode[8] == 'w') { if (info->mode[8] == 'w') {
color = DIR_COLOR_EXEC; color = DIR_COLOR_EXEC; /* directory is green if anyone can write it */
} else { } else {
color = DIR_COLOR; color = DIR_COLOR; /* other whise as the french say it, bleu */
} }
} else if (info->type == DT_LNK) { } else if (info->type == DT_LNK) {
color = LINK_COLOR; color = LINK_COLOR; /* symlink color */
} else if (info->type == DT_SOCK) { } else if (info->type == DT_SOCK) {
color = SOCK_COLOR; color = SOCK_COLOR; /* sockt color */
} else if ( } else if (
info->type == DT_CHR || info->type == DT_CHR ||
info->type == DT_BLK info->type == DT_BLK
) { ) {
color = BLK_COLOR; color = BLK_COLOR; /* the weird ones are yellow, cool */
} else { } else {
if (info->set_uid) { if (info->set_uid) {
color = SET_UID_COLOR; color = SET_UID_COLOR; /* hightlighted red if setuid */
} else if (info->set_gid) { } else if (info->set_gid) {
color = SET_GID_COLOR; color = SET_GID_COLOR; /* highlighted yellow if setgid and not setuid */
} else if (info->exec) { } else if (info->exec) {
color = EXEC_COLOR; color = EXEC_COLOR; /* green if executable */
} else { } else {
color = FILE_COLOR; color = FILE_COLOR; /* white if basic and boring */
} }
} }
return color; return color;
} }
static void list_files(struct FileInfo* files, int file_len, struct FileListInfo info) { /**
* Lists a file in long info mode
struct winsize w; * @param info the global info for a directory
* @param finfo the file specific info
*/
static void list_file_long(struct FileListInfo* info, struct FileInfo* finfo) {
char* color; char* color;
int column_width, row_count, i;
color = get_file_color(finfo);
printf("%s %*d %*s %*s %*s %s %s%s%s", /* print data */
finfo->mode,
info->max_link,
finfo->links,
info->max_usr,
finfo->usr->pw_name,
info->max_grp,
finfo->grp->gr_name,
info->max_size,
finfo->size,
finfo->date,
flags.colored != NO ? color : "", /* if not colored dont color */
finfo->name,
flags.colored != NO ? "\x1b[0m" : ""
);
/* if it is a symlink, add the path where it points to */
if (finfo->type == DT_LNK) {
int save = push_path_buffer(finfo->name);
/* get the link */
char lnk[PATH_MAX];
ssize_t n;
if ((n = readlink(get_path_buffer(), lnk, PATH_MAX)) != -1) {
printf(" -> %.*s\n", (int)n, lnk); /* add if successfull */
} else {
putchar('\n'); /* dont if not */
}
pop_path_buffer(save); /* cleanup */
} else {
putchar('\n'); /* if not link put newline now */
}
}
/**
* Print one file in column mode
* @param finfo the file info to be printed
*/
static void list_file_column(struct FileInfo* finfo) {
char* color;
color = get_file_color(finfo);
printf("%s%s%s\n", /* print the data */
flags.colored != NO ? color : "", finfo->name, /* if not colored dont color */
flags.colored != NO ? "\x1b[0m" : ""
);
}
/**
* List a file normally (without -l or -1)
* @param info the global folder information
* @param finfo the file specific info
* @param tty_width the width of the tty
* @param index, the index in the direcory that has been printed
* @param column_width the width of the column
* @param row_count the count of rows
*/
static void list_file_normal(
struct FileListInfo* info,
struct FileInfo* finfo,
int tty_width,
int index,
int column_width,
int row_count
) {
char* color;
color = get_file_color(finfo); /* get file color */
if (info->total_len > tty_width) { /* snap to column if the files span multiple lines */
if (index != 0 && index % row_count == 0) putchar('\n'); /* if at the end of the column put a new line */
printf("%s%*s%s", flags.colored != NO ? color : "", -column_width,
finfo->name, flags.colored != NO ? "\x1b[0m" : ""); /* if not colored dont color */
} else { /* otherwise just print next to eachother seperated by two spaces */
printf("%s%s%s ", flags.colored != NO ? color : "", finfo->name,
flags.colored != NO ? "\x1b[0m" : "");
}
}
/**
* Get the width of the tty, set to 0 if not a tty
* @param width the int pointer to store the width in
*/
static void get_window_size(int* width) {
struct winsize w;
ioctl(STDOUT_FILENO, TIOCGWINSZ, &w);
if (!isatty(1)) {
*width = 0;
} else {
*width = w.ws_col;
}
}
/**
* List a set of files
* @param files a array of file info structs containing file data
* @param file_len the amount of files
* @param info the global folder infomation
*/
static void list_files(struct FileInfo* files, int file_len, struct FileListInfo info) {
/* pre define variables to make gcc happy */
int column_width = 0, row_count = 0, i = 0, width = 0;
get_window_size(&width); /* get screen width */
/* if your not a tty, set to column mode and disable color if not forced */
if (width > 0) {
column_width = info.max_name + 1;
row_count = width / column_width;
} else {
flags.one_column = true;
if (flags.colored == AUTO)
flags.colored = NO;
}
/* if set to long list, display folder size */
if (flags.more_info) { if (flags.more_info) {
char total[13]; char total[13];
print_file_size(info.total_size, total); print_file_size(info.total_size, total);
printf("total %s\n", total); printf("total %s\n", total);
} }
ioctl(STDOUT_FILENO, TIOCGWINSZ, &w); /* list files base on list mode */
if (!isatty(1)) {
flags.one_column = true;
if (flags.colored == AUTO)
flags.colored = NO;
}
column_width = info.max_name + 1;
row_count = w.ws_col / column_width;
for (i = 0; i < file_len; i++) { for (i = 0; i < file_len; i++) {
struct FileInfo finfo = files[i]; struct FileInfo finfo = files[i];
color = get_file_color(&finfo); if (flags.more_info) { /* -l long list */
if (flags.more_info) { list_file_long(&info, &finfo);
printf("%s %*d %*s %*s %*s %s %s%s%s", } else if (flags.one_column) { /* -1 column list */
finfo.mode, list_file_column(&finfo);
info.max_link, } else { /* normal default listing */
finfo.links, list_file_normal(&info, &finfo, width, i, column_width, row_count);
info.max_usr,
finfo.usr->pw_name,
info.max_grp,
finfo.grp->gr_name,
info.max_size,
finfo.size,
finfo.date,
flags.colored != NO ? color : "",
finfo.name,
flags.colored != NO ? "\x1b[0m" : ""
);
if (finfo.type == DT_LNK) {
int save = push_path_buffer(finfo.name);
char lnk[PATH_MAX];
ssize_t n;
if ((n = readlink(get_path_buffer(), lnk, PATH_MAX)) != -1) {
printf(" -> %.*s\n", (int)n, lnk);
} else {
putchar('\n');
}
pop_path_buffer(save);
} else {
putchar('\n');
}
} else if (flags.one_column) {
printf("%s%s%s\n", flags.colored != NO ? color : "", finfo.name, flags.colored != NO ? "\x1b[0m" : "");
} else {
if (info.total_len > w.ws_col) {
if (i != 0 && i % row_count == 0) putchar('\n');
printf("%s%*s%s", flags.colored != NO ? color : "", -column_width,
finfo.name, flags.colored != NO ? "\x1b[0m" : "");
} else {
printf("%s%s%s ", flags.colored != NO ? color : "", finfo.name,
flags.colored != NO ? "\x1b[0m" : "");
}
} }
free(finfo.name); free(finfo.name); /* cleanup */
} }
if (!flags.more_info) printf("\n"); if (!flags.more_info) printf("\n"); /* only long long list prints a new line at the end */
} }
/**
* Get the number of places a number fills
* @param n the number to check
* @returns the num of places the number filles
*/
static int num_places (int n) { static int num_places (int n) {
int r = 1; int r = 1;
if (n < 0) n = (n == INT_MIN) ? INT_MAX: -n; if (n < 0) n = (n == INT_MIN) ? INT_MAX: -n;
while (n > 9) { while (n > 9) { /* keep going for each place */
n /= 10; n /= 10;
r++; r++;
} }
return r; return r;
} }
/**
* Read and push a file info the given array
* @param file a pointer to the list of files
* @param info the global folder info to update as read
* @param size the amount of files currently in the array
* @param capacity the capacity of the array
* @param file_path the path of the file
*/
static void push_file( static void push_file(
struct FileInfo** files, struct FileInfo** files,
struct FileListInfo* info, struct FileListInfo* info,
int* size, int* capacity, int* size, int* capacity,
const char* file_path const char* file_path
) { ) {
/* allocate variables on the stack */
struct FileInfo finfo; struct FileInfo finfo;
int user_len, group_len, name_len, size_len, link_len; int user_len, group_len, name_len, size_len, link_len;
/* if you fail to get the file info, abort */
if (!get_file_info(file_path, &finfo)) return; if (!get_file_info(file_path, &finfo)) return;
/* if the array is full realloc to make it bigger */
if (*size == *capacity) { if (*size == *capacity) {
*capacity *= 2; *capacity *= 2;
*files = xrealloc(*files, sizeof(struct FileInfo) * *capacity); *files = xrealloc(*files, sizeof(struct FileInfo) * *capacity);
} }
/* update user name length if its larger than global */
user_len = strlen(finfo.usr->pw_name); user_len = strlen(finfo.usr->pw_name);
if (user_len > info->max_usr) info->max_usr = user_len; if (user_len > info->max_usr) info->max_usr = user_len;
/* update group name length if its larger than global */
group_len = strlen(finfo.grp->gr_name); group_len = strlen(finfo.grp->gr_name);
if (group_len > info->max_grp) info->max_grp = group_len; if (group_len > info->max_grp) info->max_grp = group_len;
/* update file name length if its larger than global */
name_len = strlen(file_path); name_len = strlen(file_path);
if (name_len > info->max_name) info->max_name = name_len; if (name_len > info->max_name) info->max_name = name_len;
/* update file size human readable length if its larger than global */
size_len = strlen(finfo.size); size_len = strlen(finfo.size);
if (size_len > info->max_size) info->max_size = size_len; if (size_len > info->max_size) info->max_size = size_len;
/* update inode link place count length if its larger than global */
link_len = num_places(finfo.links); link_len = num_places(finfo.links);
if (link_len > info->max_link) info->max_link = link_len; if (link_len > info->max_link) info->max_link = link_len;
/* update final info */
info->total_len += name_len + 2; info->total_len += name_len + 2;
info->total_size += finfo.bytes; info->total_size += finfo.bytes;
/* insert file data into array */
(*files)[*size] = finfo; (*files)[*size] = finfo;
(*size)++; (*size)++; /* increment size counter */
} }
/**
* When recursing, read all files and folders, recurses each folder, and prints each file
* @param dir_name the dir_name relative to the current path buffer
*/
static void recurse_directory(char* dir_name) { static void recurse_directory(char* dir_name) {
/* allocate variables on the stack */
DIR* d; DIR* d;
int capacity, size, save; int capacity, size, save;
struct dirent* file; struct dirent* file;
struct FileInfo* files; struct FileInfo* files;
struct FileListInfo info; struct FileListInfo info;
/* push the dir name to the path buffer */
save = push_path_buffer(dir_name); save = push_path_buffer(dir_name);
/* if failed to open dir abort */
d = get_directory(get_path_buffer()); d = get_directory(get_path_buffer());
if (d == NULL) { if (d == NULL) {
return; return;
} }
/* allocate initial array*/
capacity = 8; capacity = 8;
size = 0; size = 0;
files = xalloc(sizeof(struct FileInfo) * capacity); files = xalloc(sizeof(struct FileInfo) * capacity);
memset(&info, 0, sizeof(struct FileListInfo)); memset(&info, 0, sizeof(struct FileListInfo)); /* zero out default directory globals */
/* for each file read, recurse if directory, read file if file */
while((file = readdir(d)) != NULL) { while((file = readdir(d)) != NULL) {
if (!flags.hidden && prefix(".", file->d_name)) continue; if (!flags.hidden && prefix(".", file->d_name)) continue; /* if not set to show hidden, skip */
if (flags.hide_dot && is_dot_dir(file->d_name)) continue; if (flags.hide_dot && is_dot_dir(file->d_name)) continue; /* if its set to hide dot skip */
if (file->d_type == DT_DIR && !is_dot_dir(file->d_name)) { if (file->d_type == DT_DIR && !is_dot_dir(file->d_name)) { /* dont recurse dot files or scary infinite loop */
recurse_directory(file->d_name); recurse_directory(file->d_name); /* recurse if directory */
} else { } else {
push_file(&files, &info, &size, &capacity, file->d_name); push_file(&files, &info, &size, &capacity, file->d_name); /* read file if a file */
} }
} }
/* print dir name */
if (flags.colored == NO) { if (flags.colored == NO) {
printf("\n%s:\n", get_path_buffer()); printf("\n%s:\n", get_path_buffer());
} else { } else {
printf("\n%s%s:%s\n", DIR_COLOR, get_path_buffer(), FILE_COLOR); printf("\n%s%s:%s\n", DIR_COLOR, get_path_buffer(), FILE_COLOR);
} }
/* list files */
list_files(files, size, info); list_files(files, size, info);
free(files); free(files); /* clean up */
if (!flags.more_info) printf("\n");
closedir(d); closedir(d);
pop_path_buffer(save); pop_path_buffer(save);
} }
/**
* List directory non recursivly or recursivly
* @param path the name of the directory to search
*/
static void list_directory(char* path) { static void list_directory(char* path) {
/* allocate variables on the stack */
DIR* d; DIR* d;
int capacity, size, save; int capacity, size, save;
struct FileInfo* files; struct FileInfo* files;
struct FileListInfo info; struct FileListInfo info;
struct dirent* file; struct dirent* file;
/* if set to recurse, run the recurse function */
if (flags.recurse) { if (flags.recurse) {
recurse_directory(path); recurse_directory(path);
return; return;
} }
/* if failed to open directory, abort */
d = get_directory(path); d = get_directory(path);
if (d == NULL) return; if (d == NULL) return;
/* push folder to path buffer */
save = push_path_buffer(path); save = push_path_buffer(path);
/* set array defaults */
capacity = 8; capacity = 8;
size = 0; size = 0;
/* allocate file array */
files = xalloc(sizeof(struct FileInfo) * capacity); files = xalloc(sizeof(struct FileInfo) * capacity);
memset(&info, 0, sizeof(struct FileListInfo)); memset(&info, 0, sizeof(struct FileListInfo)); /* zero out directory globals */
/* add each file regardless if its a directory or file since not recursing */
while ((file = readdir(d)) != NULL) { while ((file = readdir(d)) != NULL) {
if (!flags.hidden && prefix(".", file->d_name)) continue; if (!flags.hidden && prefix(".", file->d_name)) continue; /* if not set to show hidden ship */
if (flags.hide_dot && is_dot_dir(file->d_name)) continue; if (flags.hide_dot && is_dot_dir(file->d_name)) continue; /* if set to hide dot skip */
push_file(&files, &info, &size, &capacity, file->d_name); push_file(&files, &info, &size, &capacity, file->d_name); /* add file to array */
} }
/* only list files if there are any to list */
if (size > 0) list_files(files, size, info); if (size > 0) list_files(files, size, info);
/* cleanup */
free(files); free(files);
pop_path_buffer(save);
closedir(d); closedir(d);
pop_path_buffer(save);
} }
/* return if a path is a directory */
static bool is_dir(const char* path) { static bool is_dir(const char* path) {
struct stat s; struct stat s;
if (stat(path, &s) < 0) return false; if (stat(path, &s) < 0) return false;
return S_ISDIR(s.st_mode); return S_ISDIR(s.st_mode);
} }
/**
* Goes though each argument and prints the argument if it is a file
* @param start the argument to start at
* @param argc the argument count
* @param argv, the argument data
*/
static void list_file_args(int start, int argc, char** argv) { static void list_file_args(int start, int argc, char** argv) {
/* allocate variables on the stack */
int capacity, size, i; int capacity, size, i;
struct FileInfo* files; struct FileInfo* files;
struct FileListInfo info; struct FileListInfo info;
/* array defaults */
capacity = 8; capacity = 8;
size = 0; size = 0;
files = xalloc(sizeof(struct FileInfo) * capacity); files = xalloc(sizeof(struct FileInfo) * capacity); /* allocate array */
memset(&info, 0, sizeof(struct FileListInfo)); memset(&info, 0, sizeof(struct FileListInfo)); /* zero out globals */
for (i = start; i < argc; i++) { for (i = start; i < argc; i++) {
if (is_dir(argv[i])) continue; if (is_dir(argv[i])) continue; /* if the argument is a dir skip */
push_file(&files, &info, &size, &capacity, argv[i]); push_file(&files, &info, &size, &capacity, argv[i]);
} }
/* print file args */
if (size > 0) list_files(files, size, info); if (size > 0) list_files(files, size, info);
/* clean up */
free(files); free(files);
} }
/**
* Prints help message for ls
*/
static void help(void) { static void help(void) {
printf("Usage: ls [FILE]...\n\n"); printf("Usage: ls [FILE]...\n\n");
printf("List directory contents\n\n"); printf("List directory contents\n\n");
printf("\t-1\tOne column output\n"); printf("\t-1\tOne column output\n");
printf("\t-l\tLong format\n"); printf("\t-l\tLong format\n");
printf("\t-o\tList raw octets instead of romanized file mode\n");
printf("\t-a\tInclude names starting with .\n"); printf("\t-a\tInclude names starting with .\n");
printf("\t-A\tLike -a but without . and ..\n"); printf("\t-A\tLike -a but without . and ..\n");
printf("\t-R\tRecurse\n"); printf("\t-R\tRecurse\n");
} }
/**
* Takes in each argument that has a single - and parses it
* @param c the character after the -
* @param next the next argument in argv that hasnt been parsed
* @reutrn if the next arg was used or if the arg was invalid
*/
static int short_arg(char c, char* next) { static int short_arg(char c, char* next) {
UNUSED(next); UNUSED(next); /* next arg unused */
switch (c) { switch (c) {
case 'R': case 'R':
flags.recurse = true; flags.recurse = true;
@ -508,15 +711,24 @@ static int short_arg(char c, char* next) {
case 'l': case 'l':
flags.more_info = true; flags.more_info = true;
break; break;
case 'o':
flags.octets = true;
break;
default: default:
return ARG_INVALID; return ARG_INVALID;
} }
return ARG_UNUSED; return ARG_UNUSED;
} }
/**
* Takes in each long argument (--)
* @param cur the current argument
* @param next the next argument in argv that hasnt been parsed
* @param if the next arg was used or if the arg was invalid
*/
static int long_arg(char* cur, char* next) { static int long_arg(char* cur, char* next) {
UNUSED(next); UNUSED(next); /* next arg unused */
if (prefix("--color=", cur)) { if (prefix("--color=", cur)) { /* parse --color argument */
char* arg = cur + 8; char* arg = cur + 8;
if (streql("yes", arg) || streql("always", arg)) { if (streql("yes", arg) || streql("always", arg)) {
flags.colored = YES; flags.colored = YES;
@ -528,38 +740,47 @@ static int long_arg(char* cur, char* next) {
error("invalid color options: %s", arg); error("invalid color options: %s", arg);
} }
} else { } else {
return ARG_IGNORE; return ARG_IGNORE; /* stop reading arguments */
} }
return ARG_UNUSED; return ARG_UNUSED; /* return */
} }
/**
* Lists a directorys contents
*/
COMMAND(ls_main) { COMMAND(ls_main) {
int start, i; int start, i;
bool titled; bool titled;
/* initalize flag defaults */
flags.hidden = false; flags.hidden = false;
flags.more_info = false; flags.more_info = false;
flags.hide_dot = false; flags.hide_dot = false;
flags.one_column = false; flags.one_column = false;
flags.recurse = false; flags.recurse = false;
flags.colored = NO; flags.colored = NO;
flags.octets = false;
/* parse given arguments */
start = parse_args(argc, argv, help, short_arg, long_arg); start = parse_args(argc, argv, help, short_arg, long_arg);
/* if not arguments are given list current directory */
if (argc - start == 0) { if (argc - start == 0) {
list_directory("."); list_directory(".");
return EXIT_SUCCESS; return EXIT_SUCCESS;
} }
/* list all arguments that are files */
list_file_args(start, argc, argv); list_file_args(start, argc, argv);
/* list each directory in the arguments */
titled = argc - start > 1; titled = argc - start > 1;
for (i = start; i < argc; i++) { for (i = start; i < argc; i++) {
if (!is_dir(argv[i])) continue; if (!is_dir(argv[i])) continue; /* if not directory skip */
if (titled && !flags.recurse) { if (titled && !flags.recurse) { /* list title of title is forced and not set to recurse*/
if (flags.colored != NO) { if (flags.colored != NO) {
printf("\n%s%s:%s\n", DIR_COLOR, argv[i], FILE_COLOR); printf("\n%s%s:%s\n", DIR_COLOR, argv[i], FILE_COLOR);
} else { } else {
@ -567,8 +788,8 @@ COMMAND(ls_main) {
} }
} }
list_directory(argv[i]); list_directory(argv[i]); /* parse and list directory */
} }
return EXIT_SUCCESS; return EXIT_SUCCESS; /* return */
} }

View file

@ -7,7 +7,6 @@
-Isrc -Isrc
-Icommand -Icommand
-Ilib -Ilib
-Ihash
-D_DEFAULT_SOURCE -D_DEFAULT_SOURCE
-DMAJOR=0 -DMAJOR=0
-DMINOR=0 -DMINOR=0