KernelLand/Modules/Storage/LVM/volumes.c

   1 /*
   2  * Acess2 Logical Volume Manager
   3  * - By John Hodge (thePowersGang)
   4  *
   5  * volumes.c
   6  * - Volume management
   7  */
   8 #define DEBUG   1
   9 #include "lvm_int.h"
  10
  11 // === PROTOTYPES ===
  12  int    LVM_int_VFSReadEmul(void *Arg, Uint64 BlockStart, size_t BlockCount, void *Dest);
  13  int    LVM_int_VFSWriteEmul(void *Arg, Uint64 BlockStart, size_t BlockCount, const void *Source);
  14
  15 // === CODE ===
  16 // --------------------------------------------------------------------
  17 // Managment / Initialisation
  18 // --------------------------------------------------------------------
  19 int LVM_AddVolumeVFS(const char *Name, int FD)
  20 {
  21         // Assuming 512-byte blocks, not a good idea
  22 //      return LVM_AddVolume(Name, (void*)(Uint)FD, 512, LVM_int_VFSReadEmul, LVM_int_VFSWriteEmul);
  23         return 0;
  24 }
  25
  26 void *LVM_AddVolume(const tLVM_VolType *Type, const char *Name, void *Ptr, size_t BlockSize, size_t BlockCount)
  27 {
  28         tLVM_Vol        dummy_vol;
  29         tLVM_Format     *fmt;
  30
  31         if( BlockCount == 0 || BlockSize == 0 ) {
  32                 Log_Error("LVM", "BlockSize(0x%x)/BlockCount(0x%x) invalid in LVM_AddVolume",
  33                         BlockSize, BlockCount);
  34                 return NULL;
  35         }
  36
  37         dummy_vol.Type = Type;
  38         dummy_vol.Ptr = Ptr;
  39         dummy_vol.BlockCount = BlockCount;
  40         dummy_vol.BlockSize = BlockSize;
  41
  42         // Read the first block of the volume
  43         void *first_block = malloc(BlockSize);
  44         if( !first_block ) {
  45                 Log_Error("LVM", "LVM_AddVolume - malloc error on %i bytes", BlockSize);
  46                 return NULL;
  47         }
  48         Type->Read(Ptr, 0, 1, first_block);
  49
  50         // Determine Format
  51         // TODO: Determine format
  52         fmt = &gLVM_MBRType;
  53
  54         // Type->CountSubvolumes
  55         dummy_vol.nSubVolumes = fmt->CountSubvolumes(&dummy_vol, first_block);
  56
  57         // Create real volume descriptor
  58         // TODO: If this needs to be rescanned later, having the subvolume list separate might be an idea
  59         size_t  allocsize = sizeof(tLVM_Vol) + strlen(Name) + 1 + sizeof(tLVM_SubVolume*) * dummy_vol.nSubVolumes;
  60         tLVM_Vol        *real_vol = malloc( allocsize );
  61         if( !real_vol ) {
  62                 Log_Error("LVM", "LVM_AddVolume - malloc error on %i bytes", allocsize);
  63                 free(first_block);
  64                 return NULL;
  65         }
  66         real_vol->Next = NULL;
  67         real_vol->Type = Type;
  68         real_vol->Ptr = Ptr;
  69         real_vol->BlockSize = BlockSize;
  70         real_vol->BlockCount = BlockCount;
  71         real_vol->nSubVolumes = dummy_vol.nSubVolumes;
  72         real_vol->SubVolumes = (void*)( real_vol->Name + strlen(Name) + 1 );
  73         real_vol->BlockSize = BlockSize;
  74         strcpy(real_vol->Name, Name);
  75         memset(real_vol->SubVolumes, 0, sizeof(tLVM_SubVolume*) * real_vol->nSubVolumes);
  76         // - VFS Nodes
  77         memset(&real_vol->DirNode, 0, sizeof(tVFS_Node));
  78         real_vol->DirNode.Type = &gLVM_VolNodeType;
  79         real_vol->DirNode.ImplPtr = real_vol;
  80         real_vol->DirNode.Flags = VFS_FFLAG_DIRECTORY;
  81         real_vol->DirNode.Size = -1;
  82         memset(&real_vol->VolNode, 0, sizeof(tVFS_Node));
  83         real_vol->VolNode.Type = &gLVM_VolNodeType;
  84         real_vol->VolNode.ImplPtr = real_vol;
  85         real_vol->VolNode.Size = BlockCount * BlockSize;
  86
  87         // Type->PopulateSubvolumes
  88         fmt->PopulateSubvolumes(real_vol, first_block);
  89         free(first_block);
  90
  91         // Add to volume list
  92         gpLVM_LastVolume->Next = real_vol;
  93         gpLVM_LastVolume = real_vol;
  94
  95         return real_vol;
  96 }
  97
  98 void LVM_int_SetSubvolume_Anon(tLVM_Vol *Volume, int Index, Uint64 FirstBlock, Uint64 BlockCount)
  99 {
 100         tLVM_SubVolume  *sv;
 101          int    namelen;
 102
 103         if( Index < 0 || Index >= Volume->nSubVolumes ) {
 104                 Log_Warning("LVM", "SV ID is out of range (0 < %i < %i)",
 105                         Index, Volume->nSubVolumes);
 106                 return ;
 107         }
 108
 109         if( Volume->SubVolumes[Index] ) {
 110                 Log_Warning("LVM", "Attempt to set SV %i of %p twice", Index, Volume);
 111                 return ;
 112         }
 113
 114         namelen = snprintf(NULL, 0, "%i", Index);
 115
 116         sv = malloc( sizeof(tLVM_SubVolume) + namelen + 1 );
 117         if(!sv) {
 118                 // Oh, f*ck
 119                 return ;
 120         }
 121         Volume->SubVolumes[Index] = sv;
 122
 123         sv->Vol = Volume;
 124         sprintf(sv->Name, "%i", Index);
 125         sv->FirstBlock = FirstBlock;
 126         sv->BlockCount = BlockCount;
 127         memset(&sv->Node, 0, sizeof(tVFS_Node));
 128
 129         sv->Node.ImplPtr = sv;
 130         sv->Node.Type = &gLVM_SubVolNodeType;
 131         sv->Node.Size = BlockCount * Volume->BlockSize;
 132
 133         Log_Log("LVM", "Partition %s/%s - 0x%llx+0x%llx blocks",
 134                 Volume->Name, sv->Name, FirstBlock, BlockCount);
 135 }
 136
 137 // --------------------------------------------------------------------
 138 // IO
 139 // --------------------------------------------------------------------
 140 size_t LVM_int_ReadVolume(tLVM_Vol *Volume, Uint64 BlockNum, size_t BlockCount, void *Dest)
 141 {
 142         return Volume->Type->Read(Volume->Ptr, BlockNum, BlockCount, Dest);
 143 }
 144
 145 size_t LVM_int_WriteVolume(tLVM_Vol *Volume, Uint64 BlockNum, size_t BlockCount, const void *Src)
 146 {
 147         return Volume->Type->Write(Volume->Ptr, BlockNum, BlockCount, Src);
 148 }
 149
 150 int LVM_int_VFSReadEmul(void *Arg, Uint64 BlockStart, size_t BlockCount, void *Dest)
 151 {
 152         size_t  blocksize;
 153         size_t  rv;
 154
 155         blocksize = 512;        // TODO: Don't assume
 156
 157         rv = VFS_ReadAt( (int)(Uint)Arg, BlockStart * blocksize, BlockCount * blocksize, Dest );
 158         rv /= blocksize;
 159         return rv;
 160 }
 161
 162 int LVM_int_VFSWriteEmul(void *Arg, Uint64 BlockStart, size_t BlockCount, const void *Source)
 163 {
 164         size_t  blocksize;
 165         size_t  rv;
 166
 167         blocksize = 512;        // TODO: Don't assume
 168
 169         rv = VFS_WriteAt( (int)(Uint)Arg, BlockStart * blocksize, BlockCount * blocksize, Source );
 170         rv /= blocksize;
 171         return rv;
 172 }
 173