Advertisement
Guest User

Untitled

a guest
Nov 13th, 2019
78
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C 3.64 KB | None | 0 0
  1. #include <stdlib.h>
  2. #include <stdio.h>
  3. #include <sys/stat.h>
  4. #include <sys/types.h>
  5. #include <dirent.h>
  6. #include <unistd.h>
  7. #include <fcntl.h>
  8. #include <time.h>
  9. #include <string.h>
  10. #include <stdint.h>
  11. #include <pwd.h>
  12. #include <grp.h>
  13. #include <sys/types.h>
  14. #include <errno.h>
  15. #include <stdbool.h>
  16.  
  17. void print_rwx(int a) {
  18.     for (int i = 0; i < 9; i += 1) {
  19.         if (a & (1 << (8 - i))) {
  20.             if (i % 3 == 0) {
  21.                 printf("r");
  22.             }
  23.             else if (i % 3 == 1) {
  24.                 printf("w");
  25.             }
  26.             else {
  27.                 printf("x");
  28.             }
  29.         }
  30.         else {
  31.             printf("-");
  32.         }
  33.     }
  34. }
  35.  
  36. void print_info(struct stat *st, char * name) {
  37.     if (S_ISBLK(st -> st_mode)){
  38.         printf("b");
  39.     }
  40.     else if (S_ISCHR(st -> st_mode)) {
  41.         printf("c");
  42.     }
  43.     else if (S_ISDIR(st -> st_mode)) {
  44.         printf("d");
  45.     }
  46.     else if (S_ISFIFO(st -> st_mode)) {
  47.         printf("p");
  48.     }
  49.     else if (S_ISLNK(st -> st_mode)) {
  50.         printf("l");
  51.     }
  52.     else if (S_ISSOCK(st -> st_mode)) {
  53.         printf("s");
  54.     }
  55.     else if (S_ISREG(st -> st_mode)) {
  56.         printf("-");
  57.     }
  58.    
  59.     print_rwx(st -> st_mode % 512);
  60.  
  61.     printf(" %ld ", st -> st_nlink);
  62.  
  63.     struct passwd *user = getpwuid(st -> st_uid);
  64.     if (user == NULL) {
  65.         printf("%d ", st -> st_uid);
  66.     }
  67.     else {
  68.         printf("%s ", user -> pw_name);
  69.     }
  70.  
  71.     struct group *group = getgrgid(st -> st_gid);
  72.     if (group == NULL) {
  73.         printf("%d ", st -> st_gid);
  74.     }
  75.     else {
  76.          printf("%s ", group -> gr_name);
  77.     }
  78.  
  79.     printf("%ld ", st -> st_size);
  80. }
  81.  
  82.  
  83. int main(int argc, char *argv[]) {
  84.     struct stat st;
  85.     struct dirent *entry;
  86.  
  87.     if (lstat(argv[1], &st) == -1) {
  88.         fprintf(stderr, "%s\n", strerror(errno));
  89.         exit(1);
  90.     }
  91.  
  92.     if (!S_ISDIR(st.st_mode)) {
  93.         print_info(&st, argv[1]);
  94.         char *name = argv[1];
  95.         if (S_ISLNK(st.st_mode)) {
  96.             char *res = malloc(PATH_MAX * sizeof(char));
  97.             ssize_t ret;
  98.             ret = readlink(argv[1], res, PATH_MAX);
  99.             if (ret == -1) {
  100.                 fprintf(stderr, "%s\n", strerror(errno));
  101.                 exit(1);
  102.             }
  103.             *(res + ret) = '\0';
  104.             printf("%s -> %s\n", name, res);
  105.             free(res);
  106.         }
  107.         else {
  108.             printf("%s\n", name);
  109.         }
  110.         return 0;
  111.     }
  112.  
  113.     int dirfd = open(argv[1], O_RDONLY | O_DIRECTORY);
  114.     if (dirfd == -1) {
  115.         fprintf(stderr, "%s\n", strerror(errno));
  116.         exit(1);
  117.     }
  118.  
  119.     DIR *current_dir = fdopendir(dirfd);
  120.     if (current_dir == NULL) {
  121.         fprintf(stderr, "%s\n", strerror(errno));
  122.         exit(1);
  123.     }
  124.  
  125.     while (entry = readdir(current_dir)) {
  126.         if (!strncmp(entry -> d_name, ".", 1))
  127.             continue;
  128.         if (fstatat(dirfd, entry -> d_name, &st, AT_SYMLINK_NOFOLLOW) == -1) {
  129.             fprintf(stderr, "%s\n", strerror(errno));
  130.             exit(1);
  131.         }
  132.         print_info(&st, entry -> d_name);
  133.         if (S_ISLNK(st.st_mode)) {
  134.             char *res = malloc(PATH_MAX * sizeof(char));
  135.             ssize_t ret = readlinkat(dirfd, entry -> d_name, res, PATH_MAX);
  136.             if (ret == -1) {
  137.                 fprintf(stderr, "%s\n", strerror(errno));
  138.                 exit(1);
  139.             }
  140.             *(res + ret) = '\0';
  141.             printf("%s -> %s\n", entry -> d_name, res);
  142.             free(res);
  143.         }
  144.         else {
  145.             printf("%s\n", entry -> d_name);
  146.         }
  147.     }
  148.  
  149.     close(dirfd);
  150.  
  151.     return 0;
  152. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement