… | |
… | |
993 | } |
993 | } |
994 | |
994 | |
995 | static signed char |
995 | static signed char |
996 | eio_dent_cmp (const eio_dirent *a, const eio_dirent *b) |
996 | eio_dent_cmp (const eio_dirent *a, const eio_dirent *b) |
997 | { |
997 | { |
998 | return b->score - a->score ? b->score - a->score /* works because our signed char is always 0..100 */ |
998 | return a->score - b->score ? a->score - b->score /* works because our signed char is always 0..100 */ |
999 | : a->inode < b->inode ? -1 : a->inode > b->inode ? 1 : 0; |
999 | : a->inode < b->inode ? -1 : a->inode > b->inode ? 1 : 0; |
1000 | } |
1000 | } |
1001 | |
1001 | |
1002 | #define EIO_DENT_CMP(i,op,j) eio_dent_cmp (&i, &j) op 0 |
1002 | #define EIO_DENT_CMP(i,op,j) eio_dent_cmp (&i, &j) op 0 |
1003 | |
1003 | |
1004 | #define EIO_QSORT_CUTOFF 30 /* quite high, but performs well on many filesystems */ |
1004 | #define EIO_SORT_CUTOFF 30 /* quite high, but performs well on many filesystems */ |
1005 | #define EIO_QSORT_SKIP 60 /* when to skip qsort completely */ |
1005 | #define EIO_SORT_FAST 60 /* when to only use insertion sort */ |
1006 | |
1006 | |
1007 | static void |
1007 | static void |
1008 | eio_dent_sort (eio_dirent *dents, int size) |
1008 | eio_dent_radix_sort (eio_dirent *dents, int size, signed char score_bits, ino_t inode_bits) |
1009 | { |
1009 | { |
|
|
1010 | unsigned char bits [9 + sizeof (ino_t) * 8]; |
|
|
1011 | unsigned char *bit = bits; |
|
|
1012 | |
|
|
1013 | assert (CHAR_BIT == 8); |
|
|
1014 | assert (sizeof (eio_dirent) * 8 < 256); |
|
|
1015 | assert (offsetof (eio_dirent, inode)); /* we use 0 as sentinel */ |
|
|
1016 | assert (offsetof (eio_dirent, score)); /* we use 0 as sentinel */ |
|
|
1017 | |
|
|
1018 | if (size <= EIO_SORT_FAST) |
|
|
1019 | return; |
|
|
1020 | |
|
|
1021 | /* first prepare an array of bits to test in our radix sort */ |
|
|
1022 | /* try to take endianness into account, as well as differences in ino_t sizes */ |
|
|
1023 | /* inode_bits must contain all inodes ORed together */ |
|
|
1024 | /* which is used to skip bits that are 0 everywhere, which is very common */ |
|
|
1025 | { |
|
|
1026 | ino_t endianness; |
1010 | int i, j; |
1027 | int i, j; |
1011 | |
1028 | |
|
|
1029 | /* we store the byte offset of byte n into byte n of "endianness" */ |
|
|
1030 | for (i = 0; i < sizeof (ino_t); ++i) |
|
|
1031 | ((unsigned char *)&endianness)[i] = i; |
|
|
1032 | |
|
|
1033 | *bit++ = 0; |
|
|
1034 | |
|
|
1035 | for (i = 0; i < sizeof (ino_t); ++i) |
|
|
1036 | { |
|
|
1037 | /* shifting off the byte offsets out of "endianness" */ |
|
|
1038 | int offs = (offsetof (eio_dirent, inode) + (endianness & 0xff)) * 8; |
|
|
1039 | endianness >>= 8; |
|
|
1040 | |
|
|
1041 | for (j = 0; j < 8; ++j) |
|
|
1042 | if (inode_bits & (((ino_t)1) << (i * 8 + j))) |
|
|
1043 | *bit++ = offs + j; |
|
|
1044 | } |
|
|
1045 | |
|
|
1046 | for (j = 0; j < 8; ++j) |
|
|
1047 | if (score_bits & (1 << j)) |
|
|
1048 | *bit++ = offsetof (eio_dirent, score) * 8 + j; |
|
|
1049 | } |
|
|
1050 | |
|
|
1051 | /* now actually do the sorting (a variant of MSD radix sort) */ |
|
|
1052 | { |
|
|
1053 | eio_dirent *base_stk [9 + sizeof (ino_t) * 8], *base; |
|
|
1054 | eio_dirent *end_stk [9 + sizeof (ino_t) * 8], *end; |
|
|
1055 | unsigned char *bit_stk [9 + sizeof (ino_t) * 8]; |
|
|
1056 | int stk_idx = 0; |
|
|
1057 | |
|
|
1058 | base_stk [stk_idx] = dents; |
|
|
1059 | end_stk [stk_idx] = dents + size; |
|
|
1060 | bit_stk [stk_idx] = bit - 1; |
|
|
1061 | |
|
|
1062 | do |
|
|
1063 | { |
|
|
1064 | base = base_stk [stk_idx]; |
|
|
1065 | end = end_stk [stk_idx]; |
|
|
1066 | bit = bit_stk [stk_idx]; |
|
|
1067 | |
|
|
1068 | for (;;) |
|
|
1069 | { |
|
|
1070 | unsigned char O = *bit >> 3; |
|
|
1071 | unsigned char M = 1 << (*bit & 7); |
|
|
1072 | |
|
|
1073 | eio_dirent *a = base; |
|
|
1074 | eio_dirent *b = end; |
|
|
1075 | |
|
|
1076 | if (b - a < EIO_SORT_CUTOFF) |
|
|
1077 | break; |
|
|
1078 | |
|
|
1079 | /* now bit-partition the array on the bit */ |
|
|
1080 | /* this ugly asymmetric loop seems to perform much better than typical */ |
|
|
1081 | /* partition algos found in the literature */ |
|
|
1082 | do |
|
|
1083 | if (!(((unsigned char *)a)[O] & M)) |
|
|
1084 | ++a; |
|
|
1085 | else if (!(((unsigned char *)--b)[O] & M)) |
|
|
1086 | { |
|
|
1087 | eio_dirent tmp = *a; *a = *b; *b = tmp; |
|
|
1088 | ++a; |
|
|
1089 | } |
|
|
1090 | while (b > a); |
|
|
1091 | |
|
|
1092 | /* next bit, or stop, if no bits left in this path */ |
|
|
1093 | if (!*--bit) |
|
|
1094 | break; |
|
|
1095 | |
|
|
1096 | base_stk [stk_idx] = a; |
|
|
1097 | end_stk [stk_idx] = end; |
|
|
1098 | bit_stk [stk_idx] = bit; |
|
|
1099 | ++stk_idx; |
|
|
1100 | |
|
|
1101 | end = a; |
|
|
1102 | } |
|
|
1103 | } |
|
|
1104 | while (stk_idx--); |
|
|
1105 | } |
|
|
1106 | } |
|
|
1107 | |
|
|
1108 | static void |
|
|
1109 | eio_dent_insertion_sort (eio_dirent *dents, int size) |
|
|
1110 | { |
|
|
1111 | /* first move the smallest element to the front, to act as a sentinel */ |
|
|
1112 | { |
|
|
1113 | int i; |
|
|
1114 | eio_dirent *min = dents; |
|
|
1115 | |
|
|
1116 | /* the radix pre-pass ensures that the minimum element is in the first EIO_SORT_CUTOFF + 1 elements */ |
|
|
1117 | for (i = size > EIO_SORT_FAST ? EIO_SORT_CUTOFF + 1 : size; --i; ) |
|
|
1118 | if (EIO_DENT_CMP (dents [i], <, *min)) |
|
|
1119 | min = &dents [i]; |
|
|
1120 | |
|
|
1121 | /* swap elements 0 and j (minimum) */ |
|
|
1122 | { |
|
|
1123 | eio_dirent tmp = *dents; *dents = *min; *min = tmp; |
|
|
1124 | } |
|
|
1125 | } |
|
|
1126 | |
|
|
1127 | /* then do standard insertion sort, assuming that all elements are >= dents [0] */ |
|
|
1128 | { |
|
|
1129 | eio_dirent *i, *j; |
|
|
1130 | |
|
|
1131 | for (i = dents + 1; i < dents + size; ++i) |
|
|
1132 | { |
|
|
1133 | eio_dirent value = *i; |
|
|
1134 | |
|
|
1135 | for (j = i - 1; EIO_DENT_CMP (*j, >, value); --j) |
|
|
1136 | j [1] = j [0]; |
|
|
1137 | |
|
|
1138 | j [1] = value; |
|
|
1139 | } |
|
|
1140 | } |
|
|
1141 | } |
|
|
1142 | |
|
|
1143 | static void |
|
|
1144 | eio_dent_sort (eio_dirent *dents, int size, signed char score_bits, ino_t inode_bits) |
|
|
1145 | { |
1012 | if (size <= 1) |
1146 | if (size <= 1) |
1013 | return; /* our insertion sort relies on size > 0 */ |
1147 | return; /* our insertion sort relies on size > 0 */ |
1014 | |
1148 | |
1015 | if (size > EIO_QSORT_SKIP) /* skip quicksort for small directories */ |
1149 | /* first we use a radix sort, but only for dirs >= EIO_SORT_FAST */ |
1016 | { |
1150 | /* and stop sorting when the partitions are <= EIO_SORT_CUTOFF */ |
1017 | /* first, use quicksort */ |
1151 | eio_dent_radix_sort (dents, size, score_bits, inode_bits); |
1018 | /* should be good for 2**31 entries */ |
|
|
1019 | struct rng { int l, r; } rng [32]; |
|
|
1020 | |
1152 | |
1021 | i = 0; |
|
|
1022 | rng[0].l = 0; |
|
|
1023 | rng[0].r = size; |
|
|
1024 | |
|
|
1025 | while (expect_true (i >= 0)) |
|
|
1026 | { |
|
|
1027 | int L = rng [i].l; |
|
|
1028 | int R = rng [i].r - 1; |
|
|
1029 | |
|
|
1030 | if (expect_false (L + EIO_QSORT_CUTOFF < R)) |
|
|
1031 | { |
|
|
1032 | eio_dirent piv = dents [L]; |
|
|
1033 | |
|
|
1034 | while (L < R) |
|
|
1035 | { |
|
|
1036 | while (EIO_DENT_CMP (dents [R], >=, piv) && L < R) |
|
|
1037 | --R; |
|
|
1038 | |
|
|
1039 | if (L < R) |
|
|
1040 | dents [L++] = dents [R]; |
|
|
1041 | |
|
|
1042 | while (EIO_DENT_CMP (dents [L], <=, piv) && L < R) |
|
|
1043 | ++L; |
|
|
1044 | |
|
|
1045 | if (L < R) |
|
|
1046 | dents [R--] = dents [L]; |
|
|
1047 | } |
|
|
1048 | |
|
|
1049 | dents [L] = piv; |
|
|
1050 | |
|
|
1051 | ++i; |
|
|
1052 | rng [i].l = L + 1; |
|
|
1053 | rng [i].r = rng [i - 1].r; |
|
|
1054 | rng [i - 1].r = L; |
|
|
1055 | |
|
|
1056 | if (rng [i].r - rng [i].l > rng [i - 1].r - rng [i - 1].l) |
|
|
1057 | { |
|
|
1058 | struct rng t; |
|
|
1059 | |
|
|
1060 | t = rng [i]; rng [i] = rng [i - 1]; rng [i - 1] = t; |
|
|
1061 | } |
|
|
1062 | } |
|
|
1063 | else |
|
|
1064 | --i; |
|
|
1065 | } |
|
|
1066 | } |
|
|
1067 | |
|
|
1068 | /* use an insertion sort after qsort, or for small arrays */ |
1153 | /* use an insertion sort at the end, or for small arrays, */ |
1069 | /* first move the smallest element to the front, to act as a sentinel */ |
1154 | /* as insertion sort is more efficient for small partitions */ |
1070 | { |
1155 | eio_dent_insertion_sort (dents, size); |
1071 | int min = 0; |
|
|
1072 | |
|
|
1073 | for (i = size > EIO_QSORT_SKIP ? EIO_QSORT_CUTOFF : size; --i; ) |
|
|
1074 | if (EIO_DENT_CMP (dents [i], <, dents [min])) |
|
|
1075 | min = i; |
|
|
1076 | |
|
|
1077 | /* swap elements 0 and j (minimum) */ |
|
|
1078 | { |
|
|
1079 | eio_dirent tmp = dents [0]; dents [0] = dents [min]; dents [min] = tmp; |
|
|
1080 | } |
|
|
1081 | } |
|
|
1082 | |
|
|
1083 | /* then do standard insertion sort */ |
|
|
1084 | for (i = 1; i < size; ++i) |
|
|
1085 | { |
|
|
1086 | eio_dirent value = dents [i]; |
|
|
1087 | |
|
|
1088 | for (j = i - 1; EIO_DENT_CMP (dents [j], >, value); --j) |
|
|
1089 | dents [j + 1] = dents [j]; |
|
|
1090 | |
|
|
1091 | dents [j + 1] = value; |
|
|
1092 | } |
|
|
1093 | } |
1156 | } |
1094 | |
1157 | |
1095 | /* read a full directory */ |
1158 | /* read a full directory */ |
1096 | static void |
1159 | static void |
1097 | eio__scandir (eio_req *req, etp_worker *self) |
1160 | eio__scandir (eio_req *req, etp_worker *self) |
… | |
… | |
1103 | int namesoffs = 0; |
1166 | int namesoffs = 0; |
1104 | int flags = req->int1; |
1167 | int flags = req->int1; |
1105 | eio_dirent *dents = 0; |
1168 | eio_dirent *dents = 0; |
1106 | int dentalloc = 128; |
1169 | int dentalloc = 128; |
1107 | int dentoffs = 0; |
1170 | int dentoffs = 0; |
|
|
1171 | ino_t inode_bits = 0; |
1108 | |
1172 | |
1109 | req->result = -1; |
1173 | req->result = -1; |
1110 | |
1174 | |
1111 | if (!(flags & EIO_READDIR_DENTS)) |
1175 | if (!(flags & EIO_READDIR_DENTS)) |
1112 | flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER); |
1176 | flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER); |
1113 | |
1177 | |
1114 | X_LOCK (wrklock); |
1178 | X_LOCK (wrklock); |
1115 | /* the corresponding closedir is in ETP_WORKER_CLEAR */ |
1179 | /* the corresponding closedir is in ETP_WORKER_CLEAR */ |
1116 | self->dirp = dirp = opendir (req->ptr1); |
1180 | self->dirp = dirp = opendir (req->ptr1); |
1117 | req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE; |
1181 | req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE; |
1118 | req->ptr1 = names = malloc (namesalloc); |
|
|
1119 | req->ptr2 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0; |
1182 | req->ptr1 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0; |
|
|
1183 | req->ptr2 = names = malloc (namesalloc); |
1120 | X_UNLOCK (wrklock); |
1184 | X_UNLOCK (wrklock); |
1121 | |
1185 | |
1122 | if (dirp && names && (!flags || dents)) |
1186 | if (dirp && names && (!flags || dents)) |
1123 | for (;;) |
1187 | for (;;) |
1124 | { |
1188 | { |
… | |
… | |
1132 | |
1196 | |
1133 | /* sort etc. */ |
1197 | /* sort etc. */ |
1134 | req->int1 = flags; |
1198 | req->int1 = flags; |
1135 | req->result = dentoffs; |
1199 | req->result = dentoffs; |
1136 | |
1200 | |
1137 | if (dents) |
|
|
1138 | { |
|
|
1139 | eio_dirent *ent = dents + dentoffs; |
|
|
1140 | |
|
|
1141 | while (ent > dents) |
|
|
1142 | (--ent)->name = names + (size_t)ent->name; |
|
|
1143 | } |
|
|
1144 | |
|
|
1145 | if (flags & EIO_READDIR_STAT_ORDER |
1201 | if (flags & EIO_READDIR_STAT_ORDER) |
1146 | || !(~flags & (EIO_READDIR_DIRS_FIRST | EIO_READDIR_FOUND_UNKNOWN))) |
1202 | eio_dent_sort (dents, dentoffs, 0, inode_bits); /* sort by inode exclusively */ |
1147 | eio_dent_sort (dents, dentoffs); /* score depends of DIRS_FIRST */ |
|
|
1148 | else if (flags & EIO_READDIR_DIRS_FIRST) |
1203 | else if (flags & EIO_READDIR_DIRS_FIRST) |
|
|
1204 | if (flags & EIO_READDIR_FOUND_UNKNOWN) |
|
|
1205 | eio_dent_sort (dents, dentoffs, 7, inode_bits); /* sort by score and inode */ |
|
|
1206 | else |
1149 | { |
1207 | { |
1150 | /* in this case, all is known, and we just put dirs first and sort them */ |
1208 | /* in this case, all is known, and we just put dirs first and sort them */ |
1151 | eio_dirent *ent = dents + dentoffs; |
1209 | eio_dirent *oth = dents + dentoffs; |
1152 | eio_dirent *dir = dents; |
1210 | eio_dirent *dir = dents; |
1153 | |
1211 | |
1154 | /* now move dirs to the front, and non-dirs to the back */ |
1212 | /* now partition dirs to the front, and non-dirs to the back */ |
1155 | /* by walking from both sides and swapping if necessary */ |
1213 | /* by walking from both sides and swapping if necessary */ |
|
|
1214 | /* also clear score, so it doesn't influence sorting */ |
1156 | while (ent > dir) |
1215 | while (oth > dir) |
1157 | { |
1216 | { |
1158 | if (dir->type == DT_DIR) |
1217 | if (dir->type == EIO_DT_DIR) |
1159 | ++dir; |
1218 | ++dir; |
1160 | else |
|
|
1161 | { |
|
|
1162 | --ent; |
|
|
1163 | |
|
|
1164 | if (ent->type == DT_DIR) |
1219 | else if ((--oth)->type == EIO_DT_DIR) |
1165 | { |
1220 | { |
1166 | eio_dirent tmp = *dir; |
1221 | eio_dirent tmp = *dir; *dir = *oth; *oth = tmp; |
1167 | *dir = *ent; |
|
|
1168 | *ent = tmp; |
|
|
1169 | |
1222 | |
1170 | ++dir; |
1223 | ++dir; |
1171 | } |
1224 | } |
1172 | } |
1225 | } |
1173 | } |
|
|
1174 | |
1226 | |
1175 | /* now sort the dirs only */ |
1227 | /* now sort the dirs only */ |
1176 | eio_dent_sort (dents, dir - dents); |
1228 | eio_dent_sort (dents, dir - dents, 0, inode_bits); |
1177 | } |
1229 | } |
1178 | |
|
|
1179 | /* only provide the names array unless DENTS is specified */ |
|
|
1180 | if (!(flags & EIO_READDIR_DENTS)) |
|
|
1181 | { |
|
|
1182 | X_LOCK (wrklock); |
|
|
1183 | assert (!dents); |
|
|
1184 | req->ptr1 = 0; |
|
|
1185 | req->ptr2 = names; |
|
|
1186 | X_UNLOCK (wrklock); |
|
|
1187 | } |
|
|
1188 | |
1230 | |
1189 | break; |
1231 | break; |
1190 | } |
1232 | } |
1191 | |
1233 | |
1192 | /* now add the entry to our list(s) */ |
1234 | /* now add the entry to our list(s) */ |
… | |
… | |
1199 | |
1241 | |
1200 | while (expect_false (namesoffs + len > namesalloc)) |
1242 | while (expect_false (namesoffs + len > namesalloc)) |
1201 | { |
1243 | { |
1202 | namesalloc *= 2; |
1244 | namesalloc *= 2; |
1203 | X_LOCK (wrklock); |
1245 | X_LOCK (wrklock); |
1204 | req->ptr1 = names = realloc (names, namesalloc); |
1246 | req->ptr2 = names = realloc (names, namesalloc); |
1205 | X_UNLOCK (wrklock); |
1247 | X_UNLOCK (wrklock); |
1206 | |
1248 | |
1207 | if (!names) |
1249 | if (!names) |
1208 | break; |
1250 | break; |
1209 | } |
1251 | } |
… | |
… | |
1216 | |
1258 | |
1217 | if (expect_false (dentoffs == dentalloc)) |
1259 | if (expect_false (dentoffs == dentalloc)) |
1218 | { |
1260 | { |
1219 | dentalloc *= 2; |
1261 | dentalloc *= 2; |
1220 | X_LOCK (wrklock); |
1262 | X_LOCK (wrklock); |
1221 | req->ptr2 = dents = realloc (dents, dentalloc * sizeof (eio_dirent)); |
1263 | req->ptr1 = dents = realloc (dents, dentalloc * sizeof (eio_dirent)); |
1222 | X_UNLOCK (wrklock); |
1264 | X_UNLOCK (wrklock); |
1223 | |
1265 | |
1224 | if (!dents) |
1266 | if (!dents) |
1225 | break; |
1267 | break; |
1226 | } |
1268 | } |
1227 | |
1269 | |
1228 | ent = dents + dentoffs; |
1270 | ent = dents + dentoffs; |
1229 | |
1271 | |
1230 | ent->name = (char *)(size_t)namesoffs; /* rather dirtily we store the offset in the pointer */ |
1272 | ent->nameofs = namesoffs; /* rather dirtily we store the offset in the pointer */ |
1231 | ent->namelen = len - 1; |
1273 | ent->namelen = len - 1; |
1232 | ent->inode = D_INO (entp); |
1274 | ent->inode = D_INO (entp); |
|
|
1275 | |
|
|
1276 | inode_bits |= ent->inode; |
1233 | |
1277 | |
1234 | switch (D_TYPE (entp)) |
1278 | switch (D_TYPE (entp)) |
1235 | { |
1279 | { |
1236 | default: |
1280 | default: |
1237 | ent->type = EIO_DT_UNKNOWN; |
1281 | ent->type = EIO_DT_UNKNOWN; |
… | |
… | |
1280 | #ifdef DT_WHT |
1324 | #ifdef DT_WHT |
1281 | case DT_WHT: ent->type = EIO_DT_WHT; break; |
1325 | case DT_WHT: ent->type = EIO_DT_WHT; break; |
1282 | #endif |
1326 | #endif |
1283 | } |
1327 | } |
1284 | |
1328 | |
1285 | ent->score = 0; |
1329 | ent->score = 7; |
1286 | |
1330 | |
1287 | if (flags & EIO_READDIR_DIRS_FIRST) |
1331 | if (flags & EIO_READDIR_DIRS_FIRST) |
1288 | { |
1332 | { |
1289 | if (ent->type == EIO_DT_UNKNOWN) |
1333 | if (ent->type == EIO_DT_UNKNOWN) |
1290 | { |
1334 | { |
1291 | if (*name == '.') /* leading dots are likely directories, and, in any case, rare */ |
1335 | if (*name == '.') /* leading dots are likely directories, and, in any case, rare */ |
1292 | ent->score = 98; |
1336 | ent->score = 1; |
1293 | else if (!strchr (name, '.')) /* absense of dots indicate likely dirs */ |
1337 | else if (!strchr (name, '.')) /* absense of dots indicate likely dirs */ |
1294 | ent->score = len <= 2 ? len + 6 : len <= 4 ? 5 : len <= 7 ? 4 : 1; /* shorter == more likely dir, but avoid too many classes */ |
1338 | ent->score = len <= 2 ? 4 - len : len <= 4 ? 4 : len <= 7 ? 5 : 6; /* shorter == more likely dir, but avoid too many classes */ |
1295 | } |
1339 | } |
1296 | else if (ent->type == EIO_DT_DIR) |
1340 | else if (ent->type == EIO_DT_DIR) |
1297 | ent->score = 100; |
1341 | ent->score = 0; |
1298 | } |
1342 | } |
1299 | } |
1343 | } |
1300 | |
1344 | |
1301 | namesoffs += len; |
1345 | namesoffs += len; |
1302 | ++dentoffs; |
1346 | ++dentoffs; |