1 | /* |
1 | /* |
2 | * libeio implementation |
2 | * libeio implementation |
3 | * |
3 | * |
4 | * Copyright (c) 2007,2008 Marc Alexander Lehmann <libeio@schmorp.de> |
4 | * Copyright (c) 2007,2008,2009 Marc Alexander Lehmann <libeio@schmorp.de> |
5 | * All rights reserved. |
5 | * All rights reserved. |
6 | * |
6 | * |
7 | * Redistribution and use in source and binary forms, with or without modifica- |
7 | * Redistribution and use in source and binary forms, with or without modifica- |
8 | * tion, are permitted provided that the following conditions are met: |
8 | * tion, are permitted provided that the following conditions are met: |
9 | * |
9 | * |
… | |
… | |
68 | #endif |
68 | #endif |
69 | |
69 | |
70 | #ifdef _WIN32 |
70 | #ifdef _WIN32 |
71 | |
71 | |
72 | /*doh*/ |
72 | /*doh*/ |
73 | |
|
|
74 | #else |
73 | #else |
75 | |
74 | |
76 | # include "config.h" |
75 | # include "config.h" |
77 | # include <sys/time.h> |
76 | # include <sys/time.h> |
78 | # include <sys/select.h> |
77 | # include <sys/select.h> |
79 | # include <unistd.h> |
78 | # include <unistd.h> |
80 | # include <utime.h> |
79 | # include <utime.h> |
81 | # include <signal.h> |
80 | # include <signal.h> |
82 | # include <dirent.h> |
81 | # include <dirent.h> |
|
|
82 | |
|
|
83 | /* POSIX_SOURCE is useless on bsd's, and XOPEN_SOURCE is unreliable there, too */ |
|
|
84 | # if defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__) |
|
|
85 | # define D_INO(de) (de)->d_fileno |
|
|
86 | # define _DIRENT_HAVE_D_TYPE /* sigh */ |
|
|
87 | # elif defined(__linux) || defined(d_ino) || _XOPEN_SOURCE >= 600 |
|
|
88 | # define D_INO(de) (de)->d_ino |
|
|
89 | # endif |
|
|
90 | |
|
|
91 | # ifdef _DIRENT_HAVE_D_TYPE |
|
|
92 | # define D_TYPE(de) (de)->d_type |
|
|
93 | # endif |
83 | |
94 | |
84 | # ifndef EIO_STRUCT_DIRENT |
95 | # ifndef EIO_STRUCT_DIRENT |
85 | # define EIO_STRUCT_DIRENT struct dirent |
96 | # define EIO_STRUCT_DIRENT struct dirent |
86 | # endif |
97 | # endif |
87 | |
98 | |
… | |
… | |
100 | # else |
111 | # else |
101 | # error sendfile support requested but not available |
112 | # error sendfile support requested but not available |
102 | # endif |
113 | # endif |
103 | #endif |
114 | #endif |
104 | |
115 | |
|
|
116 | #ifndef D_TYPE |
|
|
117 | # define D_TYPE(de) 0 |
|
|
118 | #endif |
|
|
119 | |
|
|
120 | #ifndef D_INO |
|
|
121 | # define D_INO(de) 0 |
|
|
122 | #endif |
|
|
123 | |
105 | /* number of seconds after which an idle threads exit */ |
124 | /* number of seconds after which an idle threads exit */ |
106 | #define IDLE_TIMEOUT 10 |
125 | #define IDLE_TIMEOUT 10 |
107 | |
126 | |
108 | /* used for struct dirent, AIX doesn't provide it */ |
127 | /* used for struct dirent, AIX doesn't provide it */ |
109 | #ifndef NAME_MAX |
128 | #ifndef NAME_MAX |
… | |
… | |
961 | } |
980 | } |
962 | |
981 | |
963 | return res; |
982 | return res; |
964 | } |
983 | } |
965 | |
984 | |
|
|
985 | static int |
|
|
986 | eio_dent_cmp (const void *a_, const void *b_) |
|
|
987 | { |
|
|
988 | const eio_dirent *a = (const eio_dirent *)a_; |
|
|
989 | const eio_dirent *b = (const eio_dirent *)b_; |
|
|
990 | |
|
|
991 | return (int)b->score - (int)a->score ? (int)b->score - (int)a->score |
|
|
992 | : a->inode < b->inode ? -1 : a->inode > b->inode ? 1 : 0; /* int might be < ino_t */ |
|
|
993 | } |
|
|
994 | |
966 | /* read a full directory */ |
995 | /* read a full directory */ |
967 | static void |
996 | static void |
968 | eio__scandir (eio_req *req, etp_worker *self) |
997 | eio__scandir (eio_req *req, etp_worker *self) |
969 | { |
998 | { |
970 | DIR *dirp; |
999 | DIR *dirp; |
971 | EIO_STRUCT_DIRENT *entp; |
1000 | EIO_STRUCT_DIRENT *entp; |
972 | char *name, *names; |
1001 | unsigned char *name, *names; |
973 | int memlen = 4096; |
1002 | int namesalloc = 4096; |
974 | int memofs = 0; |
1003 | int namesoffs = 0; |
|
|
1004 | int flags = req->int1; |
|
|
1005 | eio_dirent *dents = 0; |
|
|
1006 | int dentalloc = 128; |
975 | int res = 0; |
1007 | int dentoffs = 0; |
|
|
1008 | |
|
|
1009 | req->result = -1; |
|
|
1010 | |
|
|
1011 | if (!(flags & EIO_READDIR_DENTS)) |
|
|
1012 | flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER); |
976 | |
1013 | |
977 | X_LOCK (wrklock); |
1014 | X_LOCK (wrklock); |
|
|
1015 | |
978 | /* the corresponding closedir is in ETP_WORKER_CLEAR */ |
1016 | /* the corresponding closedir is in ETP_WORKER_CLEAR */ |
979 | self->dirp = dirp = opendir (req->ptr1); |
1017 | self->dirp = dirp = opendir (req->ptr1); |
980 | req->flags |= EIO_FLAG_PTR2_FREE; |
1018 | req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE; |
981 | req->ptr2 = names = malloc (memlen); |
1019 | req->ptr1 = names = malloc (namesalloc); |
|
|
1020 | req->ptr2 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0; |
|
|
1021 | |
982 | X_UNLOCK (wrklock); |
1022 | X_UNLOCK (wrklock); |
983 | |
1023 | |
984 | if (dirp && names) |
1024 | if (dirp && names && (!flags || dents)) |
985 | for (;;) |
1025 | for (;;) |
986 | { |
1026 | { |
987 | errno = 0; |
1027 | errno = 0; |
988 | entp = readdir (dirp); |
1028 | entp = readdir (dirp); |
989 | |
1029 | |
990 | if (!entp) |
1030 | if (!entp) |
|
|
1031 | { |
|
|
1032 | if (errno) |
|
|
1033 | break; |
|
|
1034 | |
|
|
1035 | /* sort etc. */ |
|
|
1036 | req->int1 = flags; |
|
|
1037 | req->result = dentoffs; |
|
|
1038 | |
|
|
1039 | if (flags & EIO_READDIR_STAT_ORDER || !(~flags & (EIO_READDIR_DIRS_FIRST | EIO_READDIR_FOUND_UNKNOWN)) |
|
|
1040 | { |
|
|
1041 | /* pray your qsort doesn't use quicksort */ |
|
|
1042 | qsort (dents, dentoffs, sizeof (*dents), eio_dent_cmp); /* score depends of DIRS_FIRST */ |
|
|
1043 | } |
|
|
1044 | else if (flags & EIO_READDIR_DIRS_FIRST && !(flags & EIO_READDIR_FOUND_UNKNOWN)) |
|
|
1045 | { |
|
|
1046 | /* in this case, all is known, and we just put dirs first and sort them */ |
|
|
1047 | eio_dirent *ent = dents + dentoffs; |
|
|
1048 | eio_dirent *dir = dents; |
|
|
1049 | |
|
|
1050 | while (ent > dir) |
|
|
1051 | { |
|
|
1052 | if (dir->type == DT_DIR) |
|
|
1053 | ++dir; |
|
|
1054 | else |
|
|
1055 | { |
|
|
1056 | --ent; |
|
|
1057 | |
|
|
1058 | if (ent->type == DT_DIR) |
|
|
1059 | { |
|
|
1060 | eio_dirent tmp = *dir; |
|
|
1061 | *dir = *ent; |
|
|
1062 | *ent = tmp; |
|
|
1063 | |
|
|
1064 | ++dir; |
|
|
1065 | } |
|
|
1066 | } |
|
|
1067 | } |
|
|
1068 | |
|
|
1069 | /* now sort the dirs only */ |
|
|
1070 | qsort (dents, dir - dents, sizeof (*dents), eio_dent_cmp); |
|
|
1071 | } |
|
|
1072 | |
|
|
1073 | {int i; for(i=0;i<dentoffs;++i){eio_dirent *e=dents+i; printf ("%9ld %3d %s\n", e->inode,e->score,e->name);}}//D |
|
|
1074 | |
991 | break; |
1075 | break; |
|
|
1076 | } |
992 | |
1077 | |
993 | name = entp->d_name; |
1078 | name = entp->d_name; |
994 | |
1079 | |
995 | if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2]))) |
1080 | if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2]))) |
996 | { |
1081 | { |
997 | int len = strlen (name) + 1; |
1082 | int len = strlen (name) + 1; |
998 | |
1083 | |
999 | res++; |
1084 | while (expect_false (namesoffs + len > namesalloc)) |
1000 | |
|
|
1001 | while (memofs + len > memlen) |
|
|
1002 | { |
1085 | { |
1003 | memlen *= 2; |
1086 | namesalloc *= 2; |
1004 | X_LOCK (wrklock); |
1087 | X_LOCK (wrklock); |
1005 | req->ptr2 = names = realloc (names, memlen); |
1088 | req->ptr1 = names = realloc (names, namesalloc); |
1006 | X_UNLOCK (wrklock); |
1089 | X_UNLOCK (wrklock); |
1007 | |
1090 | |
1008 | if (!names) |
1091 | if (!names) |
1009 | break; |
1092 | break; |
1010 | } |
1093 | } |
1011 | |
1094 | |
1012 | memcpy (names + memofs, name, len); |
1095 | memcpy (names + namesoffs, name, len); |
|
|
1096 | |
|
|
1097 | if (dents) |
|
|
1098 | { |
|
|
1099 | struct eio_dirent *ent; |
|
|
1100 | |
|
|
1101 | if (expect_false (dentoffs == dentalloc)) |
|
|
1102 | { |
|
|
1103 | dentalloc *= 2; |
|
|
1104 | X_LOCK (wrklock); |
|
|
1105 | req->ptr2 = dents = realloc (dents, dentalloc * sizeof (eio_dirent)); |
|
|
1106 | X_UNLOCK (wrklock); |
|
|
1107 | |
|
|
1108 | if (!dents) |
|
|
1109 | break; |
|
|
1110 | } |
|
|
1111 | |
|
|
1112 | ent = dents + dentoffs; |
|
|
1113 | |
|
|
1114 | ent->name = names + namesoffs; |
|
|
1115 | ent->namelen = len - 1; |
|
|
1116 | ent->inode = D_INO (entp); |
|
|
1117 | |
|
|
1118 | switch (D_TYPE (entp)) |
|
|
1119 | { |
|
|
1120 | default: |
|
|
1121 | ent->type = EIO_DT_UNKNOWN; |
|
|
1122 | flags |= EIO_READDIR_FOUND_UNKNOWN; |
|
|
1123 | break; |
|
|
1124 | |
|
|
1125 | #ifdef DT_FIFO |
|
|
1126 | case DT_FIFO: ent->type = EIO_DT_FIFO; break; |
|
|
1127 | #endif |
|
|
1128 | #ifdef DT_CHR |
|
|
1129 | case DT_CHR: ent->type = EIO_DT_CHR; break; |
|
|
1130 | #endif |
|
|
1131 | #ifdef DT_DIR |
|
|
1132 | case DT_DIR: ent->type = EIO_DT_DIR; break; |
|
|
1133 | #endif |
|
|
1134 | #ifdef DT_BLK |
|
|
1135 | case DT_BLK: ent->type = EIO_DT_BLK; break; |
|
|
1136 | #endif |
|
|
1137 | #ifdef DT_REG |
|
|
1138 | case DT_REG: ent->type = EIO_DT_REG; break; |
|
|
1139 | #endif |
|
|
1140 | #ifdef DT_LNK |
|
|
1141 | case DT_LNK: ent->type = EIO_DT_LNK; break; |
|
|
1142 | #endif |
|
|
1143 | #ifdef DT_SOCK |
|
|
1144 | case DT_SOCK: ent->type = EIO_DT_SOCK; break; |
|
|
1145 | #endif |
|
|
1146 | #ifdef DT_WHT |
|
|
1147 | case DT_WHT: ent->type = EIO_DT_WHT; break; |
|
|
1148 | #endif |
|
|
1149 | } |
|
|
1150 | |
|
|
1151 | ent->score = 0; |
|
|
1152 | |
|
|
1153 | if (flags & EIO_READDIR_DIRS_FIRST) |
|
|
1154 | { |
|
|
1155 | if (ent->type == EIO_DT_UNKNOWN) |
|
|
1156 | { |
|
|
1157 | if (*name == '.') /* leading dots are likely directories, and, in any case, rare */ |
|
|
1158 | ent->score = 98; |
|
|
1159 | else if (!strchr (name, '.')) /* absense of dots indicate likely dirs */ |
|
|
1160 | ent->score = len <= 4 ? 5 : len <= 7 ? 4 : 1; /* shorter == more likely dir, but avoid too many classes */ |
|
|
1161 | } |
|
|
1162 | else if (ent->type == DT_DIR) |
|
|
1163 | ent->score = 100; |
|
|
1164 | } |
|
|
1165 | } |
|
|
1166 | |
1013 | memofs += len; |
1167 | namesoffs += len; |
|
|
1168 | ++dentoffs; |
1014 | } |
1169 | } |
1015 | } |
1170 | } |
1016 | |
1171 | else |
1017 | if (errno) |
|
|
1018 | res = -1; |
|
|
1019 | |
|
|
1020 | req->result = res; |
1172 | req->result = -1; |
|
|
1173 | |
|
|
1174 | /* if user doesn't want the dents, do not provide it */ |
|
|
1175 | if (!(flags & EIO_READDIR_DENTS)) |
|
|
1176 | { |
|
|
1177 | X_LOCK (wrklock); |
|
|
1178 | free (dents); |
|
|
1179 | req->ptr2 = req->ptr1; |
|
|
1180 | req->ptr1 = 0; |
|
|
1181 | X_UNLOCK (wrklock); |
|
|
1182 | } |
1021 | } |
1183 | } |
1022 | |
1184 | |
1023 | #if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO) |
1185 | #if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO) |
1024 | # undef msync |
1186 | # undef msync |
1025 | # define msync(a,b,c) ((errno = ENOSYS), -1) |
1187 | # define msync(a,b,c) ((errno = ENOSYS), -1) |
… | |
… | |
1445 | eio_req *eio_rmdir (const char *path, int pri, eio_cb cb, void *data) |
1607 | eio_req *eio_rmdir (const char *path, int pri, eio_cb cb, void *data) |
1446 | { |
1608 | { |
1447 | return eio__1path (EIO_RMDIR, path, pri, cb, data); |
1609 | return eio__1path (EIO_RMDIR, path, pri, cb, data); |
1448 | } |
1610 | } |
1449 | |
1611 | |
1450 | eio_req *eio_readdir (const char *path, int pri, eio_cb cb, void *data) |
1612 | eio_req *eio_readdir (const char *path, int flags, int pri, eio_cb cb, void *data) |
1451 | { |
1613 | { |
1452 | return eio__1path (EIO_READDIR, path, pri, cb, data); |
1614 | REQ (EIO_READDIR); PATH; req->int1 = flags; SEND; |
1453 | } |
1615 | } |
1454 | |
1616 | |
1455 | eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data) |
1617 | eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data) |
1456 | { |
1618 | { |
1457 | REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->int3 = (long)dev; SEND; |
1619 | REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->int3 = (long)dev; SEND; |