pintos-os.org Git - pintos-anon/blob - src/filesys/free-map.c

   1 #include "filesys/free-map.h"
   2 #include <bitmap.h>
   3 #include <debug.h>
   4 #include "filesys/file.h"
   5 #include "filesys/filesys.h"
   6 #include "filesys/inode.h"
   7
   8 static struct file *free_map_file;   /* Free map file. */
   9 static struct bitmap *free_map;      /* Free map, one bit per sector. */
  10
  11 /* Initializes the free map. */
  12 void
  13 free_map_init (void)
  14 {
  15   free_map = bitmap_create (block_size (fs_device));
  16   if (free_map == NULL)
  17     PANIC ("bitmap creation failed--file system device is too large");
  18   bitmap_mark (free_map, FREE_MAP_SECTOR);
  19   bitmap_mark (free_map, ROOT_DIR_SECTOR);
  20 }
  21
  22 /* Allocates CNT consecutive sectors from the free map and stores
  23    the first into *SECTORP.
  24    Returns true if successful, false if all sectors were
  25    available. */
  26 bool
  27 free_map_allocate (size_t cnt, block_sector_t *sectorp)
  28 {
  29   block_sector_t sector = bitmap_scan_and_flip (free_map, 0, cnt, false);
  30   if (sector != BITMAP_ERROR
  31       && free_map_file != NULL
  32       && !bitmap_write (free_map, free_map_file))
  33     {
  34       bitmap_set_multiple (free_map, sector, cnt, false);
  35       sector = BITMAP_ERROR;
  36     }
  37   if (sector != BITMAP_ERROR)
  38     *sectorp = sector;
  39   return sector != BITMAP_ERROR;
  40 }
  41
  42 /* Makes CNT sectors starting at SECTOR available for use. */
  43 void
  44 free_map_release (block_sector_t sector, size_t cnt)
  45 {
  46   ASSERT (bitmap_all (free_map, sector, cnt));
  47   bitmap_set_multiple (free_map, sector, cnt, false);
  48   bitmap_write (free_map, free_map_file);
  49 }
  50
  51 /* Opens the free map file and reads it from disk. */
  52 void
  53 free_map_open (void)
  54 {
  55   free_map_file = file_open (inode_open (FREE_MAP_SECTOR));
  56   if (free_map_file == NULL)
  57     PANIC ("can't open free map");
  58   if (!bitmap_read (free_map, free_map_file))
  59     PANIC ("can't read free map");
  60 }
  61
  62 /* Writes the free map to disk and closes the free map file. */
  63 void
  64 free_map_close (void)
  65 {
  66   file_close (free_map_file);
  67 }
  68
  69 /* Creates a new free map file on disk and writes the free map to
  70    it. */
  71 void
  72 free_map_create (void)
  73 {
  74   /* Create inode. */
  75   if (!inode_create (FREE_MAP_SECTOR, bitmap_file_size (free_map)))
  76     PANIC ("free map creation failed");
  77
  78   /* Write bitmap to file. */
  79   free_map_file = file_open (inode_open (FREE_MAP_SECTOR));
  80   if (free_map_file == NULL)
  81     PANIC ("can't open free map");
  82   if (!bitmap_write (free_map, free_map_file))
  83     PANIC ("can't write free map");
  84 }