Esempio n. 1
0
        public void addChunk(WriteFileCommand writeChunkCommand)
        {
            if (writeChunkCommand is SparseWriteCommand)
            {
                PackingType = PackingType.Sparse;
            }
            else if (writeChunkCommand is UnlzoCommand)
            {
                PackingType = PackingType.Lzo;
            }

            FileChunksTotalSize += writeChunkCommand.Size;
            Chunks.Add(writeChunkCommand);
        }
Esempio n. 2
0
        public List <WriteFileCommand> pack(string workDirectory, FileStream outputStream, IMessageLogger messageLogger)
        {
            long currentOffset             = outputStream.Position;
            List <WriteFileCommand> chunks = new List <WriteFileCommand>();

            if (Chunks.Count > 0)
            {
                messageLogger.logMessage($"Добавление образа {Name}.");
                string imgFilename = Path.Combine(workDirectory, Name + ".img");

                WriteFileCommand command = Chunks[0];

                if (command is WritePiCommand writePiCommand)
                {
                    long size = writeFileToBinary(outputStream, imgFilename);
                    chunks.Add(new WritePiCommand(writePiCommand, currentOffset, size));
                }
                else if (command is WriteBootCommand writeBootCommand)
                {
                    long size = writeFileToBinary(outputStream, imgFilename);
                    chunks.Add(new WriteBootCommand(writeBootCommand, currentOffset, size));
                }
                else if (command is StoreSecureInfoCommand storeSecureInfoCommand)
                {
                    long size = writeFileToBinary(outputStream, imgFilename);
                    chunks.Add(new StoreSecureInfoCommand(storeSecureInfoCommand, currentOffset, size));
                }
                else if (command is StoreNuttxConfigCommand storeNuttxConfigCommand)
                {
                    long size = writeFileToBinary(outputStream, imgFilename);
                    chunks.Add(new StoreNuttxConfigCommand(storeNuttxConfigCommand, currentOffset, size));
                }
                else if (command is SparseWriteCommand writeSparseCommand)
                {
                    messageLogger.logMessage("Упаковка sparse.");
//                    string simgFilename = Path.Combine( workDirectory, Name + ".simg" );
                    Compress.Sparse.Compress(imgFilename, 4096, 157286400);

                    int    index         = 0;
                    string chunkFileName = imgFilename + ".chunk." + index++;
                    while (File.Exists(chunkFileName))
                    {
                        messageLogger.logMessage("Добавление части " + index + ".");
                        currentOffset = outputStream.Position;
                        long size = writeFileToBinary(outputStream, chunkFileName);
                        File.Delete(chunkFileName);
                        chunks.Add(new SparseWriteCommand(writeSparseCommand, currentOffset, size, firstChunk: index == 1));
                        chunkFileName = imgFilename + ".chunk." + index++;
                    }
                }
                else if (command is UnlzoCommand unlzoCommand)
                {
                    const long chunkSize = 100 * 1024 * 1024;

                    messageLogger.logMessage("Упаковка lzo.");

                    string wholeLzoFilename = imgFilename + ".lzo";
                    if (File.Exists(wholeLzoFilename))
                    {
                        File.Delete(wholeLzoFilename);
                    }
                    Compress.Lzo.Compress(imgFilename, wholeLzoFilename);
                    long packedLength = 0;
                    using (FileStream inputStream = new FileStream(wholeLzoFilename, FileMode.Open, FileAccess.Read)) {
                        packedLength = inputStream.Length;
                    }

                    if (packedLength < chunkSize)
                    {
                        currentOffset = outputStream.Position;
                        long size = writeFileToBinary(outputStream, wholeLzoFilename);
                        //long size = (!File.Exists( imgFilename + ".chunk.1" )) ?
                        //    writeFileToBinary( outputStream, wholeLzoFilename ) :
                        //    writeFileToBinary( outputStream, imgFilename + ".chunk.1" );
                        chunks.Add(new UnlzoCommand(unlzoCommand, currentOffset, size, firstChunk: true));
                    }
                    else
                    {
                        using (FileStream inputStream = new FileStream(imgFilename, FileMode.Open, FileAccess.Read))
                        {
                            long imgLength = inputStream.Length;

                            //if (!File.Exists( imgFilename + ".chunk.1" ))
                            {
                                messageLogger.logMessage("Разбиение на чанки.");
                                int chunksCounter = 0;
                                for (long position = 0; position < imgLength; position += chunkSize)
                                {
                                    chunksCounter++;
                                    long remainderLength = imgLength - position;
                                    //readFileFromBinary( inputStream, position, remainderLength >= chunkSize ? chunkSize : remainderLength, imgFilename + $"a{(chunksCounter + 9):X}" );
                                    readFileFromBinary(inputStream, position, remainderLength >= chunkSize ? chunkSize : remainderLength, imgFilename + ".plain_chunk." + chunksCounter);
                                }
                                if (chunksCounter > 1)
                                {
                                    messageLogger.logMessage("Упаковка " + chunksCounter + " частей.");
                                }
                                for (int chunkIndex = 1; chunkIndex <= chunksCounter; chunkIndex++)
                                {
                                    messageLogger.logMessage("Упаковка части " + chunkIndex + ".");

                                    //string plainChunkName = imgFilename + $"a{(chunkIndex + 9):X}";
                                    string plainChunkName = imgFilename + ".plain_chunk." + chunkIndex;
                                    string chunkName      = imgFilename + ".chunk." + chunkIndex;
                                    if (File.Exists(chunkName))
                                    {
                                        File.Delete(chunkName);
                                    }
                                    Compress.Lzo.Compress(plainChunkName, chunkName);
                                    File.Delete(plainChunkName);
                                }
                            }
                            //for (int chunkIndex = 1; chunkIndex <= chunksCounter; chunkIndex++)

                            for (int chunkIndex = 1; File.Exists(imgFilename + ".chunk." + chunkIndex); chunkIndex++)
                            {
                                messageLogger.logMessage("Добавление части " + chunkIndex + ".");
                                currentOffset = outputStream.Position;
                                long size = writeFileToBinary(outputStream, imgFilename + ".chunk." + chunkIndex);
                                File.Delete(imgFilename + ".chunk." + chunkIndex);
                                chunks.Add(new UnlzoCommand(unlzoCommand, currentOffset, size, firstChunk: chunkIndex == 1));
                            }
                        }
                    }
                    File.Delete(wholeLzoFilename);
                }
            }
            return(chunks);
        }
Esempio n. 3
0
 public Partition(WriteFileCommand writeChunkCommand)
 {
     Name = writeChunkCommand.PartitionName;
     Size = writeChunkCommand.Size;
 }