#include "../..//vfs.h" #include "ext2_structs.h" #include #include #include #include #include #include #include ext2_superblock* supblks; uint32_t* blk_size; blk_grp** blk_grps; uint32_t* blk_grp_num; char** mnts; char** devs; int num_mnts; int max_mnts; typedef struct { uint32_t inode; char type; char* name; } dir_entry; typedef struct { int num; inode* inode; char is_cont_valid; char* contents; } file_info; static char get_bmap_bit(uint32_t index,uint8_t* bitmap) { uint32_t byte=index/8; uint32_t bit=index%8; char entry=bitmap[byte]; return (entry&(1<0; } static void set_bmap_bit(uint32_t index,uint8_t* bitmap) { uint32_t byte=index/8; uint32_t bit=index%8; bitmap[byte]=bitmap[byte]|(1<i_size; if (supblks[num].s_feature_rw_compat&EXT2_FEATURE_RW_COMPAT_LARGE_FILE) { size=size|(((uint64_t)node->i_ext_size_or_dir_acl)<<32); } return size; } void set_sz(inode* node,uint64_t sz,int num) { if (supblks[num].s_feature_rw_compat&EXT2_FEATURE_RW_COMPAT_LARGE_FILE) { node->i_size=(uint32_t)(sz&0xFFFFFFFF); node->i_ext_size_or_dir_acl=(uint32_t)((sz&0xFFFFFFFF00000000)>>32); } else { node->i_size=(uint32_t)sz; } } void inc_sz(inode* node,int num) { if (supblks[num].s_feature_rw_compat&EXT2_FEATURE_RW_COMPAT_LARGE_FILE) { node->i_size+=1; if (node->i_size==0) { node->i_ext_size_or_dir_acl+=1; } } else { node->i_size+=1; } } int read_char(inode* node,FILE* f,int pos,int num) { if (node->i_block[0]==0) { return -1; } int block=pos/blk_size[num]; pos=pos%blk_size[num]; if (block<12) { if (node->i_block[block]==0) { return -1; } else { return ((char*)read_blk(node->i_block[block],f,num))[pos]; } } else if (block<268) { uint32_t* blocks=read_blk(node->i_block[12],f,num); if (blocks[block-12]==0) { return -1; } else { return ((char*)read_blk(blocks[block],f,num))[pos]; } } return -1; } void append_char(inode* node,uint32_t inode_num,uint8_t c,FILE* f,int num) { uint64_t size=get_sz(node,num); uint32_t blk_idx=size/blk_size[num]; uint32_t offset=size%blk_size[num]; if (blk_idx>12||(node->i_block[blk_idx]==0)) return; uint32_t blk=node->i_block[blk_idx]; uint8_t* block=read_blk(blk,f,num); block[offset]=c; write_blk(blk,block,f,num); inc_sz(node,num); write_inode(inode_num,node,f,num); } int write_char(inode* node,uint8_t c,uint64_t pos,FILE* f,int num) { uint64_t size=get_sz(node,num); uint32_t blk_idx=pos/blk_size[num]; if (pos>size) { if (blk_idx>12) { return EFBIG; } else { if (node->i_block[blk_idx]==0) { for (uint32_t i=0;i<=blk_idx;i++) { // node.i_block[blk_idx]=reserve_block(f,num); return 0; } } else { set_sz(node,size++,num); } } } uint32_t offset=pos%blk_size[num]; if (blk_idx>12||(node->i_block[blk_idx]==0)) return 0; uint32_t blk=node->i_block[blk_idx]; uint8_t* block=read_blk(blk,f,num); block[offset]=c; write_blk(blk,block,f,num); return 0; } void* read_inode_contents(inode* node,FILE* f,int num) { if (node->i_block[0]==0) { return NULL; } uint64_t size=get_sz(node,num); char* data=malloc(sizeof(char)*ceil(size/blk_size[num])); for (int i=0;i<12;i++) { if (node->i_block[i]==0) { break; } memcpy(&data[i*blk_size[num]],read_blk(node->i_block[i],f,num),blk_size[num]); } if (node->i_block[12]!=0) { uint32_t* blocks=read_blk(node->i_block[12],f,num); for (int i=0;i<256;i++) { if (blocks[i]==0) { break; } memcpy(&data[(i+12)*blk_size[num]],read_blk(blocks[i],f,num),blk_size[num]); } } return (void*)data; } dir_entry* get_dir_listing(uint32_t inode_num,FILE* f,int num) { dir_entry* entries=malloc(sizeof(dir_entry)*100); int num_entries_used=0; int max_len=100; inode* dir_inode=read_inode(inode_num,f,num); uint32_t size=dir_inode->i_size; uint32_t tot_size=0; ext2_dir_entry* dir=read_inode_contents(dir_inode,f,num); ext2_dir_entry* current_entry=dir; for(int i=0;tot_sizefile_type==0) { break; } if(num_entries_used==max_len) { max_len+=100; entries=realloc(entries,sizeof(dir_entry)*max_len); } entries[num_entries_used].inode=current_entry->inode; entries[num_entries_used].type=current_entry->file_type; // entries[num_entries_used].name=malloc(sizeof(char)*((current_entry->name_len)+1)); entries[num_entries_used].name=current_entry->file_name; // strcpy(entries[num_entries_used].name,current_entry->file_name); entries[num_entries_used].name[(int)current_entry->name_len]='\0'; num_entries_used++; tot_size+=current_entry->rec_len; current_entry=(ext2_dir_entry*)(((char*)current_entry)+current_entry->rec_len); } if(num_entries_used==max_len) { max_len+=1; entries=realloc(entries,sizeof(dir_entry)*max_len); } entries[num_entries_used].inode=0; return entries; } void free_dir_listing(char** names) { // for(int i=0;names[i]!=NULL;i++) { // free(names[i]); // } // free(names); } ext2_dir_entry* read_dir_entry(uint32_t inode_num,uint32_t dir_entry_num,FILE* f,int num) { inode* dir_inode=read_inode(inode_num,f,num); uint32_t size=dir_inode->i_size; uint32_t tot_size=0; uint32_t ent_num=0; ext2_dir_entry* dir=read_inode_contents(dir_inode,f,num); ext2_dir_entry* current_entry=dir; for(int i=0;tot_sizefile_type==0) { break; } if(ent_num==dir_entry_num) { return current_entry; } ent_num++; tot_size+=current_entry->rec_len; current_entry=(ext2_dir_entry*)(((char*)current_entry)+current_entry->rec_len); } return NULL; } uint32_t inode_for_fname(uint32_t dir_inode_num, char* name, char* got_inode,FILE* f,int num) { *got_inode=0; dir_entry* entries=get_dir_listing(dir_inode_num,f,num); for(int i=0;entries[i].inode!=0;i++) { if (strcmp(entries[i].name,name)==0) { *got_inode=1; return entries[i].inode; } } return 0; } char* fname_for_inode(uint32_t dir_inode_num, uint32_t inode_num,FILE* f,int num) { for(int i=0;;i++) { ext2_dir_entry* entry=read_dir_entry(dir_inode_num,i,f,num); if (entry) { if (entry->inode==inode_num) { char* name=malloc(strlen(entry->file_name)+1); strcpy(name,entry->file_name); return name; } } else { break; } } return NULL; } static char drv(fs_op op,FILE* stream,void* data1,void* data2) { if (op==FSOP_MOUNT) { char* dev=(char*)data2; if (max_mnts==num_mnts) { supblks=realloc(supblks,sizeof(ext2_superblock)*(max_mnts+32)); blk_size=realloc(blk_size,sizeof(uint32_t)*(max_mnts+32)); blk_grps=realloc(blk_grps,sizeof(blk_grp*)*(max_mnts+32)); blk_grp_num=realloc(blk_grp_num,sizeof(uint32_t)*(max_mnts+32)); mnts=realloc(mnts,sizeof(char*)*(max_mnts+32)); devs=realloc(devs,sizeof(char*)*(max_mnts+32)); max_mnts+=32; } FILE* f=fopen(dev,"r"); fseek(f,1024,SEEK_SET); ext2_superblock* supblk=(ext2_superblock*)malloc(sizeof(char)*1024); fread(supblk,1024,1,f); supblks[num_mnts]=*supblk; blk_size[num_mnts]=1024<<(supblk->s_log_blk_size); double num_blk_grps_dbl=supblk->s_blocks_count/(double)supblk->s_blocks_per_group; uint32_t num_blk_grps=ceil(num_blk_grps_dbl); uint32_t num_blks=((sizeof(blk_grp)*num_blk_grps)/blk_size[num_mnts])+1; blk_grps[num_mnts]=malloc(sizeof(uint8_t)*num_blks*1024); blk_grp_num[num_mnts]=num_blk_grps; char* data_ptr=(char*)(blk_grps[num_mnts]); for (uint32_t i=0;imntpnt)==0) { stream->data=malloc(sizeof(file_info)); data=stream->data; data->num=i; data->is_cont_valid=0; break; } } if (data) { FILE* f=fopen(devs[data->num],"r+"); uint32_t inode_num=2; inode* node; for (char* tok=strtok(stream->path,"/");tok!=NULL;tok=strtok(NULL,"/")) { char got_inode; uint32_t temp_num=inode_for_fname(inode_num,tok,&got_inode,f,data->num); if (got_inode) { inode_num=temp_num; node=read_inode(inode_num,f,data->num); if ((node->i_mode&EXT2_S_IFDIR)==0) { char* next_tok=strtok(NULL,"/"); if (next_tok) { errno=ENOTDIR; fclose(f); return 0; } else { break; } } } else { if (stream->wr) { char* next_tok=strtok(NULL,"/"); if (next_tok) { errno=ENOENT; fclose(f); return 0; } else { klog("INFO","Creating new file"); klog("INFO","Parent directory inode:%d",inode_num); inode_num=reserve_inode(f,data->num); klog("INFO","Inode %d is free",inode_num); inode node; node.i_mode=EXT2_S_IFREG; node.i_uid=0; node.i_size=0; node.i_atime=0; node.i_ctime=0; node.i_mtime=0; node.i_dtime=0; node.i_gid=0; node.i_links_count=1; node.i_blocks=0; node.i_flags=0; node.i_osd1=0; for (int i=0;i<15;i++) { node.i_block[i]=0; } node.i_generation=0; node.i_file_acl=0; node.i_ext_size_or_dir_acl=0; node.i_faddr=0; node.i_osd2=0; write_inode(inode_num,&node,f,data->num); errno=ENOENT; fclose(f); return 0; } } errno=ENOENT; fclose(f); return 0; } } data->inode=node; fclose(f); return 1; } else { return 0; } } if (op==FSOP_GETC) { file_info* data=stream->data; //char* contents; // if (data->is_cont_valid) { // contents=data->contents; // } else { // FILE* f=fopen(devs[data->num],"r"); // contents=read_inode_contents(data->inode,f,data->num); // fclose(f); // data->is_cont_valid=1; // data->contents=contents; // } if (stream->pos>get_sz(data->inode,data->num)) { *((int*)data1)=EOF; stream->eof=1; return 1; } FILE* f=fopen(devs[data->num],"r"); *((int*)data1)=read_char(data->inode,f,stream->pos,data->num); fclose(f); stream->pos++; return 1; } if (op==FSOP_PUTC) { file_info* data=stream->data; FILE* f=fopen(devs[data->num],"r+"); write_char(data->inode,*((uint8_t*)data1),stream->pos,f,data->num); return 1; } if (op==FSOP_CLOSE) { return 1; } return 0; } void init_ext2() { supblks=malloc(sizeof(ext2_superblock)*32); blk_size=malloc(sizeof(uint32_t)*32); blk_grps=malloc(sizeof(blk_grp*)*32); blk_grp_num=malloc(sizeof(uint32_t)*32); mnts=malloc(sizeof(char*)*32); devs=malloc(sizeof(char*)*32); num_mnts=0; max_mnts=32; register_fs(drv,"ext2"); }