1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
| #include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>
#define SZ_ALLOC (10)
int myScandir(char *name, struct dirent ***res) {
// Ouverture dossier
DIR *dir;
if ((dir=opendir(name)) == NULL) return -1;
// Traitement du dossier
size_t sz=0;
size_t nb=0;
struct dirent **tmp;
struct dirent *current;
// Lecture du dossier ouvert
(*res)=NULL;
while ((current=readdir(dir)) != NULL) {
// Besoin d'espace ?
if (nb == sz) {
// Agrandissement espace (pour le premier, comme "*res" est NULL, realloc se comporte comme un simple malloc)
sz+=SZ_ALLOC;
tmp=realloc(*res, sz * sizeof(*tmp));
// Souci alloc ?
if (tmp == NULL) goto nettoyage;
// Allocation réussie
(*res)=tmp;
}
// Allocation pour dossier en cours
(*res)[nb]=malloc(sizeof(*current));
if ((*res)[nb] == NULL) goto nettoyage;
// Récupération dossier en cours (qui est en mémoire statique)
memcpy((*res)[nb], current, sizeof(*current));
// On a un dossier de plus
nb++;
}
// Tout a été traité
closedir(dir);
return nb;
nettoyage:
// Ici gestion de la merde - Il faut nettoyer ce qui a été alloué
// Attention, il n'y a eu que (au max) "nb" allocations réussies (le dernier (*res)[nb] pouvant être NULL ce qui ne dérangera pas le free())
// La variable "sz" ne servant plus à rien servira d'indice de boucle
for (sz=0, tmp=(*res); sz < nb; sz++, tmp++) free(*tmp);
free(*res);
return -1;
}
int main(void) {
struct dirent **namelist;
int n;
n = myScandir(".", &namelist);
printf("nb dir: %d\n", n);
while (n--) {
printf("%s (0x%04x)\n", namelist[n]->d_name, namelist[n]->d_type);
free (namelist[n]);
}
free (namelist);
} |
Partager