LCOV - code coverage report
Current view: top level - extmod - vfs_fat.c (source / functions) Hit Total Coverage
Test: unix_coverage_v1.22.0-343-g7b050b366.info Lines: 181 188 96.3 %
Date: 2024-04-25 17:00:48 Functions: 17 17 100.0 %
Branches: 56 66 84.8 %

           Branch data     Line data    Source code
       1                 :            : /*
       2                 :            :  * This file is part of the MicroPython project, http://micropython.org/
       3                 :            :  *
       4                 :            :  * The MIT License (MIT)
       5                 :            :  *
       6                 :            :  * Copyright (c) 2014 Damien P. George
       7                 :            :  * Copyright (c) 2016 Paul Sokolovsky
       8                 :            :  *
       9                 :            :  * Permission is hereby granted, free of charge, to any person obtaining a copy
      10                 :            :  * of this software and associated documentation files (the "Software"), to deal
      11                 :            :  * in the Software without restriction, including without limitation the rights
      12                 :            :  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
      13                 :            :  * copies of the Software, and to permit persons to whom the Software is
      14                 :            :  * furnished to do so, subject to the following conditions:
      15                 :            :  *
      16                 :            :  * The above copyright notice and this permission notice shall be included in
      17                 :            :  * all copies or substantial portions of the Software.
      18                 :            :  *
      19                 :            :  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
      20                 :            :  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
      21                 :            :  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
      22                 :            :  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
      23                 :            :  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
      24                 :            :  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
      25                 :            :  * THE SOFTWARE.
      26                 :            :  */
      27                 :            : 
      28                 :            : #include "py/mpconfig.h"
      29                 :            : #if MICROPY_VFS_FAT
      30                 :            : 
      31                 :            : #if !MICROPY_ENABLE_FINALISER
      32                 :            : #error "MICROPY_VFS_FAT requires MICROPY_ENABLE_FINALISER"
      33                 :            : #endif
      34                 :            : 
      35                 :            : #if !MICROPY_VFS
      36                 :            : #error "with MICROPY_VFS_FAT enabled, must also enable MICROPY_VFS"
      37                 :            : #endif
      38                 :            : 
      39                 :            : #include <string.h>
      40                 :            : #include "py/runtime.h"
      41                 :            : #include "py/mperrno.h"
      42                 :            : #include "lib/oofatfs/ff.h"
      43                 :            : #include "extmod/vfs_fat.h"
      44                 :            : #include "shared/timeutils/timeutils.h"
      45                 :            : 
      46                 :            : #if FF_MAX_SS == FF_MIN_SS
      47                 :            : #define SECSIZE(fs) (FF_MIN_SS)
      48                 :            : #else
      49                 :            : #define SECSIZE(fs) ((fs)->ssize)
      50                 :            : #endif
      51                 :            : 
      52                 :            : #define mp_obj_fat_vfs_t fs_user_mount_t
      53                 :            : 
      54                 :          4 : static mp_import_stat_t fat_vfs_import_stat(void *vfs_in, const char *path) {
      55                 :          4 :     fs_user_mount_t *vfs = vfs_in;
      56                 :          4 :     FILINFO fno;
      57         [ -  + ]:          4 :     assert(vfs != NULL);
      58                 :          4 :     FRESULT res = f_stat(&vfs->fatfs, path, &fno);
      59         [ +  + ]:          4 :     if (res == FR_OK) {
      60         [ +  - ]:          2 :         if ((fno.fattrib & AM_DIR) != 0) {
      61                 :            :             return MP_IMPORT_STAT_DIR;
      62                 :            :         } else {
      63                 :          2 :             return MP_IMPORT_STAT_FILE;
      64                 :            :         }
      65                 :            :     }
      66                 :            :     return MP_IMPORT_STAT_NO_EXIST;
      67                 :            : }
      68                 :            : 
      69                 :         50 : static mp_obj_t fat_vfs_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
      70                 :         50 :     mp_arg_check_num(n_args, n_kw, 1, 1, false);
      71                 :            : 
      72                 :            :     // create new object
      73                 :         50 :     fs_user_mount_t *vfs = mp_obj_malloc(fs_user_mount_t, type);
      74                 :         50 :     vfs->fatfs.drv = vfs;
      75                 :            : 
      76                 :            :     // Initialise underlying block device
      77                 :         50 :     vfs->blockdev.flags = MP_BLOCKDEV_FLAG_FREE_OBJ;
      78                 :         50 :     vfs->blockdev.block_size = FF_MIN_SS; // default, will be populated by call to MP_BLOCKDEV_IOCTL_BLOCK_SIZE
      79                 :         50 :     mp_vfs_blockdev_init(&vfs->blockdev, args[0]);
      80                 :            : 
      81                 :            :     // mount the block device so the VFS methods can be used
      82                 :         50 :     FRESULT res = f_mount(&vfs->fatfs);
      83         [ +  + ]:         50 :     if (res == FR_NO_FILESYSTEM) {
      84                 :            :         // don't error out if no filesystem, to let mkfs()/mount() create one if wanted
      85                 :         26 :         vfs->blockdev.flags |= MP_BLOCKDEV_FLAG_NO_FILESYSTEM;
      86         [ -  + ]:         24 :     } else if (res != FR_OK) {
      87                 :          0 :         mp_raise_OSError(fresult_to_errno_table[res]);
      88                 :            :     }
      89                 :            : 
      90                 :         50 :     return MP_OBJ_FROM_PTR(vfs);
      91                 :            : }
      92                 :            : 
      93                 :            : #if _FS_REENTRANT
      94                 :            : static mp_obj_t fat_vfs_del(mp_obj_t self_in) {
      95                 :            :     mp_obj_fat_vfs_t *self = MP_OBJ_TO_PTR(self_in);
      96                 :            :     // f_umount only needs to be called to release the sync object
      97                 :            :     f_umount(&self->fatfs);
      98                 :            :     return mp_const_none;
      99                 :            : }
     100                 :            : static MP_DEFINE_CONST_FUN_OBJ_1(fat_vfs_del_obj, fat_vfs_del);
     101                 :            : #endif
     102                 :            : 
     103                 :         22 : static mp_obj_t fat_vfs_mkfs(mp_obj_t bdev_in) {
     104                 :            :     // create new object
     105                 :         22 :     fs_user_mount_t *vfs = MP_OBJ_TO_PTR(fat_vfs_make_new(&mp_fat_vfs_type, 1, 0, &bdev_in));
     106                 :            : 
     107                 :            :     // make the filesystem
     108                 :         22 :     uint8_t working_buf[FF_MAX_SS];
     109                 :         22 :     FRESULT res = f_mkfs(&vfs->fatfs, FM_FAT | FM_SFD, 0, working_buf, sizeof(working_buf));
     110         [ +  + ]:         22 :     if (res == FR_MKFS_ABORTED) { // Probably doesn't support FAT16
     111                 :          2 :         res = f_mkfs(&vfs->fatfs, FM_FAT32, 0, working_buf, sizeof(working_buf));
     112                 :            :     }
     113         [ -  + ]:         22 :     if (res != FR_OK) {
     114                 :          0 :         mp_raise_OSError(fresult_to_errno_table[res]);
     115                 :            :     }
     116                 :            : 
     117                 :         22 :     return mp_const_none;
     118                 :            : }
     119                 :            : static MP_DEFINE_CONST_FUN_OBJ_1(fat_vfs_mkfs_fun_obj, fat_vfs_mkfs);
     120                 :            : static MP_DEFINE_CONST_STATICMETHOD_OBJ(fat_vfs_mkfs_obj, MP_ROM_PTR(&fat_vfs_mkfs_fun_obj));
     121                 :            : 
     122                 :            : typedef struct _mp_vfs_fat_ilistdir_it_t {
     123                 :            :     mp_obj_base_t base;
     124                 :            :     mp_fun_1_t iternext;
     125                 :            :     mp_fun_1_t finaliser;
     126                 :            :     bool is_str;
     127                 :            :     FF_DIR dir;
     128                 :            : } mp_vfs_fat_ilistdir_it_t;
     129                 :            : 
     130                 :        266 : static mp_obj_t mp_vfs_fat_ilistdir_it_iternext(mp_obj_t self_in) {
     131                 :        266 :     mp_vfs_fat_ilistdir_it_t *self = MP_OBJ_TO_PTR(self_in);
     132                 :            : 
     133                 :        266 :     for (;;) {
     134                 :        266 :         FILINFO fno;
     135                 :        266 :         FRESULT res = f_readdir(&self->dir, &fno);
     136                 :        266 :         char *fn = fno.fname;
     137   [ +  +  +  + ]:        266 :         if (res != FR_OK || fn[0] == 0) {
     138                 :            :             // stop on error or end of dir
     139                 :            :             break;
     140                 :            :         }
     141                 :            : 
     142                 :            :         // Note that FatFS already filters . and .., so we don't need to
     143                 :            : 
     144                 :            :         // make 4-tuple with info about this entry
     145                 :        176 :         mp_obj_tuple_t *t = MP_OBJ_TO_PTR(mp_obj_new_tuple(4, NULL));
     146         [ +  + ]:        176 :         if (self->is_str) {
     147                 :        172 :             t->items[0] = mp_obj_new_str(fn, strlen(fn));
     148                 :            :         } else {
     149                 :          4 :             t->items[0] = mp_obj_new_bytes((const byte *)fn, strlen(fn));
     150                 :            :         }
     151         [ +  + ]:        176 :         if (fno.fattrib & AM_DIR) {
     152                 :            :             // dir
     153                 :        122 :             t->items[1] = MP_OBJ_NEW_SMALL_INT(MP_S_IFDIR);
     154                 :            :         } else {
     155                 :            :             // file
     156                 :         54 :             t->items[1] = MP_OBJ_NEW_SMALL_INT(MP_S_IFREG);
     157                 :            :         }
     158                 :        176 :         t->items[2] = MP_OBJ_NEW_SMALL_INT(0); // no inode number
     159                 :        176 :         t->items[3] = mp_obj_new_int_from_uint(fno.fsize);
     160                 :            : 
     161                 :        176 :         return MP_OBJ_FROM_PTR(t);
     162                 :            :     }
     163                 :            : 
     164                 :            :     // ignore error because we may be closing a second time
     165                 :         90 :     f_closedir(&self->dir);
     166                 :            : 
     167                 :         90 :     return MP_OBJ_STOP_ITERATION;
     168                 :            : }
     169                 :            : 
     170                 :        112 : static mp_obj_t mp_vfs_fat_ilistdir_it_del(mp_obj_t self_in) {
     171                 :        112 :     mp_vfs_fat_ilistdir_it_t *self = MP_OBJ_TO_PTR(self_in);
     172                 :            :     // ignore result / error because we may be closing a second time.
     173                 :        112 :     f_closedir(&self->dir);
     174                 :        112 :     return mp_const_none;
     175                 :            : }
     176                 :            : 
     177                 :        112 : static mp_obj_t fat_vfs_ilistdir_func(size_t n_args, const mp_obj_t *args) {
     178                 :        112 :     mp_obj_fat_vfs_t *self = MP_OBJ_TO_PTR(args[0]);
     179                 :        112 :     bool is_str_type = true;
     180                 :        112 :     const char *path;
     181         [ +  + ]:        112 :     if (n_args == 2) {
     182         [ +  + ]:         94 :         if (mp_obj_get_type(args[1]) == &mp_type_bytes) {
     183                 :          4 :             is_str_type = false;
     184                 :            :         }
     185                 :         94 :         path = mp_obj_str_get_str(args[1]);
     186                 :            :     } else {
     187                 :            :         path = "";
     188                 :            :     }
     189                 :            : 
     190                 :            :     // Create a new iterator object to list the dir
     191                 :        112 :     mp_vfs_fat_ilistdir_it_t *iter = mp_obj_malloc_with_finaliser(mp_vfs_fat_ilistdir_it_t, &mp_type_polymorph_iter_with_finaliser);
     192                 :        112 :     iter->iternext = mp_vfs_fat_ilistdir_it_iternext;
     193                 :        112 :     iter->finaliser = mp_vfs_fat_ilistdir_it_del;
     194                 :        112 :     iter->is_str = is_str_type;
     195                 :        112 :     FRESULT res = f_opendir(&self->fatfs, &iter->dir, path);
     196         [ +  + ]:        112 :     if (res != FR_OK) {
     197                 :          2 :         mp_raise_OSError(fresult_to_errno_table[res]);
     198                 :            :     }
     199                 :            : 
     200                 :        110 :     return MP_OBJ_FROM_PTR(iter);
     201                 :            : }
     202                 :            : static MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(fat_vfs_ilistdir_obj, 1, 2, fat_vfs_ilistdir_func);
     203                 :            : 
     204                 :         42 : static mp_obj_t fat_vfs_remove_internal(mp_obj_t vfs_in, mp_obj_t path_in, mp_int_t attr) {
     205                 :         42 :     mp_obj_fat_vfs_t *self = MP_OBJ_TO_PTR(vfs_in);
     206                 :         42 :     const char *path = mp_obj_str_get_str(path_in);
     207                 :            : 
     208                 :         42 :     FILINFO fno;
     209                 :         42 :     FRESULT res = f_stat(&self->fatfs, path, &fno);
     210                 :            : 
     211         [ +  + ]:         42 :     if (res != FR_OK) {
     212                 :          2 :         mp_raise_OSError(fresult_to_errno_table[res]);
     213                 :            :     }
     214                 :            : 
     215                 :            :     // check if path is a file or directory
     216         [ +  + ]:         40 :     if ((fno.fattrib & AM_DIR) == attr) {
     217                 :         36 :         res = f_unlink(&self->fatfs, path);
     218                 :            : 
     219         [ +  + ]:         36 :         if (res != FR_OK) {
     220                 :          2 :             mp_raise_OSError(fresult_to_errno_table[res]);
     221                 :            :         }
     222                 :         34 :         return mp_const_none;
     223                 :            :     } else {
     224         [ +  + ]:          4 :         mp_raise_OSError(attr ? MP_ENOTDIR : MP_EISDIR);
     225                 :            :     }
     226                 :            : }
     227                 :            : 
     228                 :         12 : static mp_obj_t fat_vfs_remove(mp_obj_t vfs_in, mp_obj_t path_in) {
     229                 :         12 :     return fat_vfs_remove_internal(vfs_in, path_in, 0); // 0 == file attribute
     230                 :            : }
     231                 :            : static MP_DEFINE_CONST_FUN_OBJ_2(fat_vfs_remove_obj, fat_vfs_remove);
     232                 :            : 
     233                 :         28 : static mp_obj_t fat_vfs_rmdir(mp_obj_t vfs_in, mp_obj_t path_in) {
     234                 :         28 :     return fat_vfs_remove_internal(vfs_in, path_in, AM_DIR);
     235                 :            : }
     236                 :            : static MP_DEFINE_CONST_FUN_OBJ_2(fat_vfs_rmdir_obj, fat_vfs_rmdir);
     237                 :            : 
     238                 :         16 : static mp_obj_t fat_vfs_rename(mp_obj_t vfs_in, mp_obj_t path_in, mp_obj_t path_out) {
     239                 :         16 :     mp_obj_fat_vfs_t *self = MP_OBJ_TO_PTR(vfs_in);
     240                 :         16 :     const char *old_path = mp_obj_str_get_str(path_in);
     241                 :         16 :     const char *new_path = mp_obj_str_get_str(path_out);
     242                 :         16 :     FRESULT res = f_rename(&self->fatfs, old_path, new_path);
     243         [ +  + ]:         16 :     if (res == FR_EXIST) {
     244                 :            :         // if new_path exists then try removing it (but only if it's a file)
     245                 :          2 :         fat_vfs_remove_internal(vfs_in, path_out, 0); // 0 == file attribute
     246                 :            :         // try to rename again
     247                 :          2 :         res = f_rename(&self->fatfs, old_path, new_path);
     248                 :            :     }
     249         [ +  + ]:         16 :     if (res == FR_OK) {
     250                 :         14 :         return mp_const_none;
     251                 :            :     } else {
     252                 :          2 :         mp_raise_OSError(fresult_to_errno_table[res]);
     253                 :            :     }
     254                 :            : 
     255                 :            : }
     256                 :            : static MP_DEFINE_CONST_FUN_OBJ_3(fat_vfs_rename_obj, fat_vfs_rename);
     257                 :            : 
     258                 :         44 : static mp_obj_t fat_vfs_mkdir(mp_obj_t vfs_in, mp_obj_t path_o) {
     259                 :         44 :     mp_obj_fat_vfs_t *self = MP_OBJ_TO_PTR(vfs_in);
     260                 :         44 :     const char *path = mp_obj_str_get_str(path_o);
     261                 :         44 :     FRESULT res = f_mkdir(&self->fatfs, path);
     262         [ +  + ]:         44 :     if (res == FR_OK) {
     263                 :         38 :         return mp_const_none;
     264                 :            :     } else {
     265                 :          6 :         mp_raise_OSError(fresult_to_errno_table[res]);
     266                 :            :     }
     267                 :            : }
     268                 :            : static MP_DEFINE_CONST_FUN_OBJ_2(fat_vfs_mkdir_obj, fat_vfs_mkdir);
     269                 :            : 
     270                 :            : // Change current directory.
     271                 :         12 : static mp_obj_t fat_vfs_chdir(mp_obj_t vfs_in, mp_obj_t path_in) {
     272                 :         12 :     mp_obj_fat_vfs_t *self = MP_OBJ_TO_PTR(vfs_in);
     273                 :         12 :     const char *path;
     274                 :         12 :     path = mp_obj_str_get_str(path_in);
     275                 :            : 
     276                 :         12 :     FRESULT res = f_chdir(&self->fatfs, path);
     277                 :            : 
     278         [ +  + ]:         12 :     if (res != FR_OK) {
     279                 :          2 :         mp_raise_OSError(fresult_to_errno_table[res]);
     280                 :            :     }
     281                 :            : 
     282                 :         10 :     return mp_const_none;
     283                 :            : }
     284                 :            : static MP_DEFINE_CONST_FUN_OBJ_2(fat_vfs_chdir_obj, fat_vfs_chdir);
     285                 :            : 
     286                 :            : // Get the current directory.
     287                 :          6 : static mp_obj_t fat_vfs_getcwd(mp_obj_t vfs_in) {
     288                 :          6 :     mp_obj_fat_vfs_t *self = MP_OBJ_TO_PTR(vfs_in);
     289                 :          6 :     char buf[MICROPY_ALLOC_PATH_MAX + 1];
     290                 :          6 :     FRESULT res = f_getcwd(&self->fatfs, buf, sizeof(buf));
     291         [ -  + ]:          6 :     if (res != FR_OK) {
     292                 :          0 :         mp_raise_OSError(fresult_to_errno_table[res]);
     293                 :            :     }
     294                 :          6 :     return mp_obj_new_str(buf, strlen(buf));
     295                 :            : }
     296                 :            : static MP_DEFINE_CONST_FUN_OBJ_1(fat_vfs_getcwd_obj, fat_vfs_getcwd);
     297                 :            : 
     298                 :            : // Get the status of a file or directory.
     299                 :         16 : static mp_obj_t fat_vfs_stat(mp_obj_t vfs_in, mp_obj_t path_in) {
     300                 :         16 :     mp_obj_fat_vfs_t *self = MP_OBJ_TO_PTR(vfs_in);
     301                 :         16 :     const char *path = mp_obj_str_get_str(path_in);
     302                 :            : 
     303                 :         16 :     FILINFO fno;
     304   [ +  -  +  +  :         16 :     if (path[0] == 0 || (path[0] == '/' && path[1] == 0)) {
                   +  + ]
     305                 :            :         // stat root directory
     306                 :          2 :         fno.fsize = 0;
     307                 :          2 :         fno.fdate = 0x2821; // Jan 1, 2000
     308                 :          2 :         fno.ftime = 0;
     309                 :          2 :         fno.fattrib = AM_DIR;
     310                 :            :     } else {
     311                 :         14 :         FRESULT res = f_stat(&self->fatfs, path, &fno);
     312         [ +  + ]:         14 :         if (res != FR_OK) {
     313                 :          2 :             mp_raise_OSError(fresult_to_errno_table[res]);
     314                 :            :         }
     315                 :            :     }
     316                 :            : 
     317                 :         14 :     mp_obj_tuple_t *t = MP_OBJ_TO_PTR(mp_obj_new_tuple(10, NULL));
     318                 :         14 :     mp_int_t mode = 0;
     319         [ +  + ]:         14 :     if (fno.fattrib & AM_DIR) {
     320                 :            :         mode |= MP_S_IFDIR;
     321                 :            :     } else {
     322                 :         12 :         mode |= MP_S_IFREG;
     323                 :            :     }
     324                 :         28 :     mp_int_t seconds = timeutils_seconds_since_epoch(
     325                 :         14 :         1980 + ((fno.fdate >> 9) & 0x7f),
     326                 :         14 :         (fno.fdate >> 5) & 0x0f,
     327                 :         14 :         fno.fdate & 0x1f,
     328                 :         14 :         (fno.ftime >> 11) & 0x1f,
     329                 :         14 :         (fno.ftime >> 5) & 0x3f,
     330                 :         14 :         2 * (fno.ftime & 0x1f)
     331                 :            :         );
     332                 :         14 :     t->items[0] = MP_OBJ_NEW_SMALL_INT(mode); // st_mode
     333                 :         14 :     t->items[1] = MP_OBJ_NEW_SMALL_INT(0); // st_ino
     334                 :         14 :     t->items[2] = MP_OBJ_NEW_SMALL_INT(0); // st_dev
     335                 :         14 :     t->items[3] = MP_OBJ_NEW_SMALL_INT(0); // st_nlink
     336                 :         14 :     t->items[4] = MP_OBJ_NEW_SMALL_INT(0); // st_uid
     337                 :         14 :     t->items[5] = MP_OBJ_NEW_SMALL_INT(0); // st_gid
     338                 :         14 :     t->items[6] = mp_obj_new_int_from_uint(fno.fsize); // st_size
     339                 :         14 :     t->items[7] = mp_obj_new_int_from_uint(seconds); // st_atime
     340                 :         14 :     t->items[8] = mp_obj_new_int_from_uint(seconds); // st_mtime
     341                 :         14 :     t->items[9] = mp_obj_new_int_from_uint(seconds); // st_ctime
     342                 :            : 
     343                 :         14 :     return MP_OBJ_FROM_PTR(t);
     344                 :            : }
     345                 :            : static MP_DEFINE_CONST_FUN_OBJ_2(fat_vfs_stat_obj, fat_vfs_stat);
     346                 :            : 
     347                 :            : // Get the status of a VFS.
     348                 :         12 : static mp_obj_t fat_vfs_statvfs(mp_obj_t vfs_in, mp_obj_t path_in) {
     349                 :         12 :     mp_obj_fat_vfs_t *self = MP_OBJ_TO_PTR(vfs_in);
     350                 :         12 :     (void)path_in;
     351                 :            : 
     352                 :         12 :     DWORD nclst;
     353                 :         12 :     FATFS *fatfs = &self->fatfs;
     354                 :         12 :     FRESULT res = f_getfree(fatfs, &nclst);
     355         [ -  + ]:         12 :     if (FR_OK != res) {
     356                 :          0 :         mp_raise_OSError(fresult_to_errno_table[res]);
     357                 :            :     }
     358                 :            : 
     359                 :         12 :     mp_obj_tuple_t *t = MP_OBJ_TO_PTR(mp_obj_new_tuple(10, NULL));
     360                 :            : 
     361                 :         12 :     t->items[0] = MP_OBJ_NEW_SMALL_INT(fatfs->csize * SECSIZE(fatfs)); // f_bsize
     362                 :         12 :     t->items[1] = t->items[0]; // f_frsize
     363                 :         12 :     t->items[2] = MP_OBJ_NEW_SMALL_INT((fatfs->n_fatent - 2)); // f_blocks
     364                 :         12 :     t->items[3] = MP_OBJ_NEW_SMALL_INT(nclst); // f_bfree
     365                 :         12 :     t->items[4] = t->items[3]; // f_bavail
     366                 :         12 :     t->items[5] = MP_OBJ_NEW_SMALL_INT(0); // f_files
     367                 :         12 :     t->items[6] = MP_OBJ_NEW_SMALL_INT(0); // f_ffree
     368                 :         12 :     t->items[7] = MP_OBJ_NEW_SMALL_INT(0); // f_favail
     369                 :         12 :     t->items[8] = MP_OBJ_NEW_SMALL_INT(0); // f_flags
     370                 :         12 :     t->items[9] = MP_OBJ_NEW_SMALL_INT(FF_MAX_LFN); // f_namemax
     371                 :            : 
     372                 :         12 :     return MP_OBJ_FROM_PTR(t);
     373                 :            : }
     374                 :            : static MP_DEFINE_CONST_FUN_OBJ_2(fat_vfs_statvfs_obj, fat_vfs_statvfs);
     375                 :            : 
     376                 :         18 : static mp_obj_t vfs_fat_mount(mp_obj_t self_in, mp_obj_t readonly, mp_obj_t mkfs) {
     377                 :         18 :     fs_user_mount_t *self = MP_OBJ_TO_PTR(self_in);
     378                 :            : 
     379                 :            :     // Read-only device indicated by writeblocks[0] == MP_OBJ_NULL.
     380                 :            :     // User can specify read-only device by:
     381                 :            :     //  1. readonly=True keyword argument
     382                 :            :     //  2. nonexistent writeblocks method (then writeblocks[0] == MP_OBJ_NULL already)
     383         [ -  + ]:         18 :     if (mp_obj_is_true(readonly)) {
     384                 :          0 :         self->blockdev.writeblocks[0] = MP_OBJ_NULL;
     385                 :            :     }
     386                 :            : 
     387                 :            :     // check if we need to make the filesystem
     388         [ +  + ]:         18 :     FRESULT res = (self->blockdev.flags & MP_BLOCKDEV_FLAG_NO_FILESYSTEM) ? FR_NO_FILESYSTEM : FR_OK;
     389         [ -  + ]:          4 :     if (res == FR_NO_FILESYSTEM && mp_obj_is_true(mkfs)) {
     390                 :          0 :         uint8_t working_buf[FF_MAX_SS];
     391                 :          0 :         res = f_mkfs(&self->fatfs, FM_FAT | FM_SFD, 0, working_buf, sizeof(working_buf));
     392                 :            :     }
     393         [ +  - ]:          4 :     if (res != FR_OK) {
     394                 :          4 :         mp_raise_OSError(fresult_to_errno_table[res]);
     395                 :            :     }
     396                 :         14 :     self->blockdev.flags &= ~MP_BLOCKDEV_FLAG_NO_FILESYSTEM;
     397                 :            : 
     398                 :         14 :     return mp_const_none;
     399                 :            : }
     400                 :            : static MP_DEFINE_CONST_FUN_OBJ_3(vfs_fat_mount_obj, vfs_fat_mount);
     401                 :            : 
     402                 :          8 : static mp_obj_t vfs_fat_umount(mp_obj_t self_in) {
     403                 :          8 :     (void)self_in;
     404                 :            :     // keep the FAT filesystem mounted internally so the VFS methods can still be used
     405                 :          8 :     return mp_const_none;
     406                 :            : }
     407                 :            : static MP_DEFINE_CONST_FUN_OBJ_1(fat_vfs_umount_obj, vfs_fat_umount);
     408                 :            : 
     409                 :            : static const mp_rom_map_elem_t fat_vfs_locals_dict_table[] = {
     410                 :            :     #if _FS_REENTRANT
     411                 :            :     { MP_ROM_QSTR(MP_QSTR___del__), MP_ROM_PTR(&fat_vfs_del_obj) },
     412                 :            :     #endif
     413                 :            :     { MP_ROM_QSTR(MP_QSTR_mkfs), MP_ROM_PTR(&fat_vfs_mkfs_obj) },
     414                 :            :     { MP_ROM_QSTR(MP_QSTR_open), MP_ROM_PTR(&fat_vfs_open_obj) },
     415                 :            :     { MP_ROM_QSTR(MP_QSTR_ilistdir), MP_ROM_PTR(&fat_vfs_ilistdir_obj) },
     416                 :            :     { MP_ROM_QSTR(MP_QSTR_mkdir), MP_ROM_PTR(&fat_vfs_mkdir_obj) },
     417                 :            :     { MP_ROM_QSTR(MP_QSTR_rmdir), MP_ROM_PTR(&fat_vfs_rmdir_obj) },
     418                 :            :     { MP_ROM_QSTR(MP_QSTR_chdir), MP_ROM_PTR(&fat_vfs_chdir_obj) },
     419                 :            :     { MP_ROM_QSTR(MP_QSTR_getcwd), MP_ROM_PTR(&fat_vfs_getcwd_obj) },
     420                 :            :     { MP_ROM_QSTR(MP_QSTR_remove), MP_ROM_PTR(&fat_vfs_remove_obj) },
     421                 :            :     { MP_ROM_QSTR(MP_QSTR_rename), MP_ROM_PTR(&fat_vfs_rename_obj) },
     422                 :            :     { MP_ROM_QSTR(MP_QSTR_stat), MP_ROM_PTR(&fat_vfs_stat_obj) },
     423                 :            :     { MP_ROM_QSTR(MP_QSTR_statvfs), MP_ROM_PTR(&fat_vfs_statvfs_obj) },
     424                 :            :     { MP_ROM_QSTR(MP_QSTR_mount), MP_ROM_PTR(&vfs_fat_mount_obj) },
     425                 :            :     { MP_ROM_QSTR(MP_QSTR_umount), MP_ROM_PTR(&fat_vfs_umount_obj) },
     426                 :            : };
     427                 :            : static MP_DEFINE_CONST_DICT(fat_vfs_locals_dict, fat_vfs_locals_dict_table);
     428                 :            : 
     429                 :            : static const mp_vfs_proto_t fat_vfs_proto = {
     430                 :            :     .import_stat = fat_vfs_import_stat,
     431                 :            : };
     432                 :            : 
     433                 :            : MP_DEFINE_CONST_OBJ_TYPE(
     434                 :            :     mp_fat_vfs_type,
     435                 :            :     MP_QSTR_VfsFat,
     436                 :            :     MP_TYPE_FLAG_NONE,
     437                 :            :     make_new, fat_vfs_make_new,
     438                 :            :     protocol, &fat_vfs_proto,
     439                 :            :     locals_dict, &fat_vfs_locals_dict
     440                 :            :     );
     441                 :            : 
     442                 :            : #endif // MICROPY_VFS_FAT

Generated by: LCOV version 1.15-5-g462f71d