Пример #1
0
        private static CodeCommand PackCommand(CodeType type, List <CodeCommand> cmds)
        {
            Debug.Assert(0 < cmds.Count);

            CodeType packType;
            CodeInfo packInfo;

            switch (type)
            {
            case CodeType.TXTAddLine:
                packType = CodeType.TXTAddLineOp;
                packInfo = new CodeInfo_TXTAddLineOp(cmds);
                break;

            case CodeType.TXTReplace:
                packType = CodeType.TXTReplaceOp;
                packInfo = new CodeInfo_TXTReplaceOp(cmds);
                break;

            case CodeType.TXTDelLine:
                packType = CodeType.TXTDelLineOp;
                packInfo = new CodeInfo_TXTDelLineOp(cmds);
                break;

            case CodeType.IniRead:
                packType = CodeType.IniReadOp;
                packInfo = new CodeInfo_IniReadOp(cmds);
                break;

            case CodeType.IniWrite:
                packType = CodeType.IniWriteOp;
                packInfo = new CodeInfo_IniWriteOp(cmds);
                break;

            case CodeType.IniDelete:
                packType = CodeType.IniDeleteOp;
                packInfo = new CodeInfo_IniDeleteOp(cmds);
                break;

            case CodeType.IniReadSection:
                packType = CodeType.IniReadSectionOp;
                packInfo = new CodeInfo_IniReadSectionOp(cmds);
                break;

            case CodeType.IniAddSection:
                packType = CodeType.IniAddSectionOp;
                packInfo = new CodeInfo_IniAddSectionOp(cmds);
                break;

            case CodeType.IniDeleteSection:
                packType = CodeType.IniDeleteSectionOp;
                packInfo = new CodeInfo_IniDeleteSectionOp(cmds);
                break;

            case CodeType.IniWriteTextLine:
                packType = CodeType.IniWriteTextLineOp;
                packInfo = new CodeInfo_IniWriteTextLineOp(cmds);
                break;

            case CodeType.Visible:
                packType = CodeType.VisibleOp;
                packInfo = new CodeInfo_VisibleOp(cmds);
                break;

            case CodeType.ReadInterface:
                packType = CodeType.ReadInterfaceOp;
                packInfo = new CodeInfo_ReadInterfaceOp(cmds);
                break;

            case CodeType.WriteInterface:
                packType = CodeType.WriteInterfaceOp;
                packInfo = new CodeInfo_WriteInterfaceOp(cmds);
                break;

            case CodeType.WimExtract:
                packType = CodeType.WimExtractOp;
                packInfo = new CodeInfo_WimExtractOp(cmds);
                break;

            case CodeType.WimPathAdd:     // Use WimPathAdd as representative of WimPath*
                packType = CodeType.WimPathOp;
                packInfo = new CodeInfo_WimPathOp(cmds);
                break;

            default:
                throw new InternalException("Internal Logic Error at CodeOptimizer.InternalOptimize");
            }

            return(new CodeCommand(MergeRawCodes(cmds), cmds[0].Section, packType, packInfo, cmds[0].LineIdx));
        }
Пример #2
0
        public static List <LogInfo> IniWriteOp(EngineState s, CodeCommand cmd)
        {
            List <LogInfo> logs = new List <LogInfo>();

            Debug.Assert(cmd.Info.GetType() == typeof(CodeInfo_IniWriteOp));
            CodeInfo_IniWriteOp infoOp = cmd.Info as CodeInfo_IniWriteOp;

            string fileName = StringEscaper.Preprocess(s, infoOp.Infos[0].FileName);

            if (StringEscaper.PathSecurityCheck(fileName, out string errorMsg) == false)
            {
                logs.Add(new LogInfo(LogState.Error, errorMsg));
                return(logs);
            }

            IniKey[] keys = new IniKey[infoOp.Cmds.Count];
            for (int i = 0; i < keys.Length; i++)
            {
                CodeInfo_IniWrite info = infoOp.Infos[i];

                string sectionName = StringEscaper.Preprocess(s, info.Section); // WB082 : 여기 값은 변수 Expand 안한다.
                string key         = StringEscaper.Preprocess(s, info.Key);     // WB082 : 여기 값은 변수 Expand는 안 하나, Escaping은 한다.
                string value       = StringEscaper.Preprocess(s, info.Value);

                if (sectionName.Equals(string.Empty, StringComparison.Ordinal))
                {
                    throw new InvalidCodeCommandException("Section name cannot be empty", cmd);
                }
                if (key.Equals(string.Empty, StringComparison.Ordinal))
                {
                    throw new InvalidCodeCommandException("Key name cannot be empty", cmd);
                }

                keys[i] = new IniKey(sectionName, key, value);
            }

            string dirPath = Path.GetDirectoryName(fileName);

            if (Directory.Exists(dirPath) == false)
            {
                Directory.CreateDirectory(dirPath);
            }

            bool result = Ini.SetKeys(fileName, keys);

            if (result)
            {
                for (int i = 0; i < keys.Length; i++)
                {
                    IniKey kv = keys[i];
                    logs.Add(new LogInfo(LogState.Success, $"Key [{kv.Key}] and its value [{kv.Value}] written", infoOp.Cmds[i]));
                }
                logs.Add(new LogInfo(LogState.Success, $"Wrote [{keys.Length}] values to [{fileName}]", cmd));
            }
            else
            {
                for (int i = 0; i < keys.Length; i++)
                {
                    IniKey kv = keys[i];
                    logs.Add(new LogInfo(LogState.Error, $"Could not write key [{kv.Key}] and its value [{kv.Value}]", infoOp.Cmds[i]));
                }
                logs.Add(new LogInfo(LogState.Error, $"Could not write [{keys.Length}] values to [{fileName}]", cmd));
            }

            return(logs);
        }
Пример #3
0
        public static List <LogInfo> IniWriteOp(EngineState s, CodeCommand cmd)
        {
            List <LogInfo> logs = new List <LogInfo>();

            CodeInfo_IniWriteOp infoOp = cmd.Info.Cast <CodeInfo_IniWriteOp>();

            string fileName = StringEscaper.Preprocess(s, infoOp.Infos[0].FileName);

            Debug.Assert(fileName != null, $"{nameof(fileName)} != null");

            if (!StringEscaper.PathSecurityCheck(fileName, out string errorMsg))
            {
                return(LogInfo.LogErrorMessage(logs, errorMsg));
            }

            IniKey[] keys = new IniKey[infoOp.Cmds.Count];
            for (int i = 0; i < keys.Length; i++)
            {
                CodeInfo_IniWrite info = infoOp.Infos[i];

                string sectionName = StringEscaper.Preprocess(s, info.Section);
                string key         = StringEscaper.Preprocess(s, info.Key);
                string value       = StringEscaper.Preprocess(s, info.Value);

                if (sectionName.Length == 0)
                {
                    return(LogInfo.LogErrorMessage(logs, "Section name cannot be empty"));
                }
                if (key.Length == 0)
                {
                    return(LogInfo.LogErrorMessage(logs, "Key name cannot be empty"));
                }

                keys[i] = new IniKey(sectionName, key, value);
            }

            string dirPath = Path.GetDirectoryName(fileName);

            if (dirPath == null)
            {
                throw new InternalException("Internal Logic Error at IniWriteOp");
            }
            if (!Directory.Exists(dirPath))
            {
                Directory.CreateDirectory(dirPath);
            }

            // If a dest file does not exist, create an empty file to force ANSI encoding as default in IniReadWriter.
            if (!File.Exists(fileName))
            {
                File.Create(fileName).Dispose();
            }

            bool result;

            if (s.CompatAutoCompactIniWriteCommand)
            {
                result = IniReadWriter.WriteCompactKeys(fileName, keys);
            }
            else
            {
                result = IniReadWriter.WriteKeys(fileName, keys);
            }

            if (result)
            {
                for (int i = 0; i < keys.Length; i++)
                {
                    IniKey kv = keys[i];
                    logs.Add(new LogInfo(LogState.Success, $"Key [{kv.Key}] and it's value [{kv.Value}] written", infoOp.Cmds[i]));
                }
                logs.Add(new LogInfo(LogState.Success, $"Wrote [{keys.Length}] values to [{fileName}]", cmd));
            }
            else
            {
                for (int i = 0; i < keys.Length; i++)
                {
                    IniKey kv = keys[i];
                    logs.Add(new LogInfo(LogState.Error, $"Could not write key [{kv.Key}] and it's value [{kv.Value}]", infoOp.Cmds[i]));
                }
                logs.Add(new LogInfo(LogState.Error, $"Could not write [{keys.Length}] values to [{fileName}]", cmd));
            }

            return(logs);
        }