public static List <BTCS> init_btom(FileStream stream, UInt64 block_num)
{
List <UInt64> btin_block_num = new List <UInt64>();
Table header = Table.get_table_header(stream, block_num);
List <BTCS> btom = new List <BTCS>();
TableType[] table_info = Table.save_record(stream, block_num, header);
Footer footer = Table.save_footer(stream, block_num, header);
if (header.table_type == 5 || header.table_type == 7)
{
if (footer.TotalIndexNode > 1) //btin
{
for (int i = 0; i < header.record_num; i++)
{
BTCS b = new BTCS();
b.NodeID = table_info[i].NodeID;
b.Checkpoint = table_info[i].StructureID;
b.BlockNum = table_info[i].BlockNum;
btin_block_num.Add(b.BlockNum);
}
btom = init_btin_all(stream, btin_block_num);
}
else //btrn, btln
{
for (int i = 0; i < header.record_num; i++)
{
BTCS b = new BTCS();
b.NodeID = table_info[i].NodeID;
b.Checkpoint = table_info[i].StructureID;
b.BlockNum = table_info[i].BlockNum;
btom.Add(b);
}
}
}
else
{
return(null);
}
return(btom);
}
public static List <BTCS> init_btin(FileStream stream, UInt64 block_num)
{
Table header = Table.get_table_header(stream, block_num);
List <BTCS> btin = new List <BTCS>();
TableType[] table_info = Table.save_record(stream, block_num, header);
// Console.WriteLine("--------Block Num {0}, # of record {1}------", block_num, header.record_num) ;
if (header.table_type == 6)
{
for (int i = 0; i < header.record_num; i++)
{
BTCS b = new BTCS();
b.NodeID = table_info[i].NodeID;
b.Checkpoint = table_info[i].StructureID;
b.BlockNum = table_info[i].BlockNum;
btin.Add(b);
}
}
return(btin);
}
public static bool init_APFS(FileStream fs, CSB csb)
{
Console.WriteLine("*************** Make File Structure at Checkpoint : {0} *************", csb.CSB_Checkpoint);
RECORD.parent_node_dic.Clear();
RECORD.parent_node_dic = new Dictionary <UInt64, List <UInt64> >();
RECORD.ffr_dict.Clear();
RECORD.ffr_dict = new Dictionary <UInt64, FileFolderRecord>();
RECORD.er_dic.Clear();
RECORD.er_dic = new Dictionary <UInt64, ExtentRecord>();
////volume structure
UInt64 VRB_addr = csb.VolumesIndexblock;
Console.WriteLine();
Console.WriteLine("VRB address : {0} {1}", VRB_addr, Utility.get_address(VRB_addr));
UInt64 VB_addr = Table.get_block_address(fs, VRB_addr, "0x30");
Console.WriteLine("VB address : {0}", VB_addr);
Console.WriteLine("first csb address : {0}, blocksize : {1}, totalLength : {2}", CSB.first_csb_address, CSB.BlockSize, CSB.TotalSize);
Table VB_h;
VB_h = Table.get_table_header(fs, VB_addr);
if (VB_h == null)
{
throw new Exception("This csb is invalid.");
}
//VB record
Console.WriteLine();
TableType[] VBrecords = Table.save_record(fs, VB_addr, VB_h);
//VCSB
for (int i = 0; i < VB_h.record_num; i++)
{
Console.WriteLine();
UInt64 VCSB_addr;
VCSB_addr = VBrecords[i].BlockNum;
Console.WriteLine("VCSB_addr : {0}", VCSB_addr);
//각 VCSB
Console.WriteLine();
VCSB vcsb = VCSB.init_vcsb(fs, VCSB_addr);
//BTCS
Console.WriteLine();
UInt64 BTCS_addr = vcsb.BTCS;
Console.WriteLine("BTCS address : {0}", BTCS_addr);
UInt64 BTOM_addr = Table.get_block_address(fs, BTCS_addr, "0x30");
Console.WriteLine("BTOM address : {0}", BTOM_addr);
int n = 0;
List <BTCS> btrn_btln = BTCS.init_btom(fs, BTOM_addr);
foreach (BTCS b in btrn_btln)
{
n++;
Console.WriteLine("btrn_btln : {0} ", n);
Console.WriteLine("node id : {0}", b.NodeID);
Console.WriteLine("Checkpoint : {0}", b.Checkpoint);
Console.WriteLine("block num: {0}, {1}\n", b.BlockNum, Utility.get_address(b.BlockNum));
if (n > 1)
{
RECORD.init_btln(fs, b.BlockNum);
}
}
// Queue 만들기
Queue <UInt64> q = new Queue <UInt64>();
q.Enqueue(2); // NodeID of root node
ulong root = 2;
if (!RECORD.parent_node_dic.ContainsKey(root))
{
return(true);
}
RECORD.ffr_dict[root].path = @restore_to_path;
while (q.Count > 0)
{
UInt64 parent_node_id = q.Dequeue();
String parent_path = RECORD.ffr_dict[parent_node_id].path;
Console.WriteLine("ParentID = {0}", parent_node_id);
if (!RECORD.parent_node_dic.ContainsKey(parent_node_id))
{
continue;
}
foreach (UInt64 child_node_Id in RECORD.parent_node_dic[parent_node_id])
{
String fname = new string(RECORD.ffr_dict[child_node_Id].FileName, 0, RECORD.ffr_dict[child_node_Id].FileName.Length - 1);
String new_path = Path.Combine(parent_path, fname);
RECORD.ffr_dict[child_node_Id].path = new_path;
Console.WriteLine(" NodeID = {0}, path : {1}", child_node_Id, new_path);
if (RECORD.ffr_dict[child_node_Id].Flag[0] == '8')
{
if (RECORD.er_dic.ContainsKey(child_node_Id))
{
Extent new_extent = Extent.read_extent(fs, (long)Utility.get_address(RECORD.er_dic[child_node_Id].ExtentStartBlockNum), (long)RECORD.er_dic[child_node_Id].ExtentLength);
Extent.write_extent(child_node_Id, new_extent.buf, new_extent.Count, new_path);
}
else
{
if (!File.Exists(new_path))
{
File.Create(new_path);
}
}
}
else if (RECORD.ffr_dict[child_node_Id].Flag[0] == '4')
{
q.Enqueue(child_node_Id);
Directory.CreateDirectory(new_path);
}
}
}
Console.WriteLine();
}
Console.WriteLine("FIN");
return(true);
}
public static bool restore_folder(FileStream fs, CSB csb, string search_name)
{
RECORD.parent_node_dic.Clear();
RECORD.parent_node_dic = new Dictionary <UInt64, List <UInt64> >();
RECORD.ffr_dict.Clear();
RECORD.ffr_dict = new Dictionary <UInt64, FileFolderRecord>();
RECORD.er_dic.Clear();
RECORD.er_dic = new Dictionary <UInt64, ExtentRecord>();
bool restore = false;
////volume structure
UInt64 VRB_addr = csb.VolumesIndexblock;
UInt64 VB_addr = Table.get_block_address(fs, VRB_addr, "0x30");
Table VB_h = Table.get_table_header(fs, VB_addr);
if (VB_h == null)
{
throw new Exception("This csb is invalid.\n Cannot restore folder");
}
//VB record
Console.WriteLine();
TableType[] VBrecords = Table.save_record(fs, VB_addr, VB_h);
//VCSB
for (int i = 0; i < VB_h.record_num; i++)
{
Console.WriteLine();
UInt64 VCSB_addr;
VCSB_addr = VBrecords[i].BlockNum;
//각 VCSB
Console.WriteLine();
VCSB vcsb = VCSB.init_vcsb(fs, VCSB_addr);
//BTCS
Console.WriteLine();
UInt64 BTCS_addr = vcsb.BTCS;
UInt64 BTOM_addr = Table.get_block_address(fs, BTCS_addr, "0x30");
int n = 0;
List <BTCS> btrn_btln = BTCS.init_btom(fs, BTOM_addr);
foreach (BTCS b in btrn_btln)
{
n++;
if (n > 1)
{
RECORD.init_btln(fs, b.BlockNum);
}
}
// Queue 만들기
Queue <UInt64> q = new Queue <UInt64>();
q.Enqueue(2); // NodeID of root node
ulong root = 2;
if (!RECORD.parent_node_dic.ContainsKey(root))
{
return(false);
}
RECORD.ffr_dict[root].path = @restore_to_path;
restore = false;
while (q.Count > 0)
{
UInt64 parent_node_id = q.Dequeue();
String parent_path = RECORD.ffr_dict[parent_node_id].path;
Console.WriteLine("ParentID = {0}", parent_node_id);
if (!RECORD.parent_node_dic.ContainsKey(parent_node_id))
{
continue;
}
foreach (UInt64 child_node_Id in RECORD.parent_node_dic[parent_node_id])
{
String fname = new string(RECORD.ffr_dict[child_node_Id].FileName, 0, RECORD.ffr_dict[child_node_Id].FileName.Length - 1);
if (restore)
{
RECORD.ffr_dict[child_node_Id].path = Path.Combine(parent_path, fname);
}
else
{
RECORD.ffr_dict[child_node_Id].path = parent_path;
}
Console.WriteLine(" NodeID = {0}, {1}", child_node_Id, fname);
String new_path = RECORD.ffr_dict[child_node_Id].path;
if (restore && RECORD.ffr_dict[child_node_Id].Flag[0] == '8')
{
if (RECORD.er_dic.ContainsKey(child_node_Id))
{
Extent new_extent = Extent.read_extent(fs, (long)Utility.get_address(RECORD.er_dic[child_node_Id].ExtentStartBlockNum), (long)RECORD.er_dic[child_node_Id].ExtentLength);
Extent.write_extent(child_node_Id, new_extent.buf, new_extent.Count, new_path);
}
else if (!File.Exists(new_path))
{
File.Create(new_path);
}
}
if (RECORD.ffr_dict[child_node_Id].Flag[0] == '4')
{
if (!restore)
{
if (fname.Equals(search_name))
{
Console.WriteLine("---------Find Folder-------");
RECORD.ffr_dict[child_node_Id].path = Path.Combine(RECORD.ffr_dict[child_node_Id].path, fname);
Directory.CreateDirectory(RECORD.ffr_dict[child_node_Id].path);
q.Clear();
q.Enqueue(child_node_Id);
restore = true;
break;
}
else
{
q.Enqueue(child_node_Id);
}
}
else
{
Directory.CreateDirectory(new_path);
q.Enqueue(child_node_Id);
}
}
}
}
Console.WriteLine();
}
Console.WriteLine("FIN");
return(restore);
}
