File indexing completed on 2024-04-28 03:53:48

 /*
     This file is part of the KDE libraries
 
     SPDX-FileCopyrightText: 2002-2003 Oswald Buddenhagen <ossi@kde.org>
     SPDX-FileCopyrightText: 2003 Waldo Bastian <bastian@kde.org>
 
     SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "kmacroexpander_p.h"
 
 #include <QRegularExpression>
 #include <QStack>
 #include <QStringList>
 
 namespace KMacroExpander
 {
 enum Quoting {
     noquote,
     singlequote,
     doublequote,
     dollarquote,
     paren,
     subst,
     group,
     math,
 };
 typedef struct {
     Quoting current;
     bool dquote;
 } State;
 typedef struct {
     QString str;
     int pos;
 } Save;
 
 }
 
 using namespace KMacroExpander;
 
 // #pragma message("TODO: Import these methods into Qt")
 
 inline static bool isSpecial(QChar cUnicode)
 {
     static const uchar iqm[] = {0xff, 0xff, 0xff, 0xff, 0xdf, 0x07, 0x00, 0xd8, 0x00, 0x00, 0x00, 0x38, 0x01, 0x00, 0x00, 0x78}; // 0-32 \'"$`<>|;&(){}*?#!~[]
 
     uint c = cUnicode.unicode();
     return (c < sizeof(iqm) * 8) && (iqm[c / 8] & (1 << (c & 7)));
 }
 
 static QString quoteArg(const QString &arg)
 {
     if (!arg.length()) {
         return QStringLiteral("''");
     }
     for (int i = 0; i < arg.length(); i++) {
         if (isSpecial(arg.unicode()[i])) {
             QChar q(QLatin1Char('\''));
             return q + QString(arg).replace(q, QLatin1String("'\\''")) + q;
         }
     }
     return arg;
 }
 
 static QString joinArgs(const QStringList &args)
 {
     QString ret;
     for (QStringList::ConstIterator it = args.begin(); it != args.end(); ++it) {
         if (!ret.isEmpty()) {
             ret.append(QLatin1Char(' '));
         }
         ret.append(quoteArg(*it));
     }
     return ret;
 }
 
 bool KMacroExpanderBase::expandMacrosShellQuote(QString &str, int &pos)
 {
     int len;
     int pos2;
     ushort ec = d->escapechar.unicode();
     State state = {noquote, false};
     QStack<State> sstack;
     QStack<Save> ostack;
     QStringList rst;
     QString rsts;
 
     while (pos < str.length()) {
         ushort cc = str.unicode()[pos].unicode();
         if (ec != 0) {
             if (cc != ec) {
                 goto nohit;
             }
             if (!(len = expandEscapedMacro(str, pos, rst))) {
                 goto nohit;
             }
         } else {
             if (!(len = expandPlainMacro(str, pos, rst))) {
                 goto nohit;
             }
         }
         if (len < 0) {
             pos -= len;
             continue;
         }
         if (state.dquote) {
             rsts = rst.join(QLatin1Char(' '));
             const static QRegularExpression regex(QStringLiteral("([$`\"\\\\])"));
             rsts.replace(regex, QStringLiteral("\\\\1"));
         } else if (state.current == dollarquote) {
             rsts = rst.join(QLatin1Char(' '));
             const static QRegularExpression regex(QStringLiteral("(['\\\\])"));
             rsts.replace(regex, QStringLiteral("\\\\1"));
         } else if (state.current == singlequote) {
             rsts = rst.join(QLatin1Char(' '));
             rsts.replace(QLatin1Char('\''), QLatin1String("'\\''"));
         } else {
             if (rst.isEmpty()) {
                 str.remove(pos, len);
                 continue;
             } else {
                 rsts = joinArgs(rst);
             }
         }
         rst.clear();
         str.replace(pos, len, rsts);
         pos += rsts.length();
         continue;
     nohit:
         if (state.current == singlequote) {
             if (cc == '\'') {
                 state = sstack.pop();
             }
         } else if (cc == '\\') {
             // always swallow the char -> prevent anomalies due to expansion
             pos += 2;
             continue;
         } else if (state.current == dollarquote) {
             if (cc == '\'') {
                 state = sstack.pop();
             }
         } else if (cc == '$') {
             cc = str.unicode()[++pos].unicode();
             if (cc == '(') {
                 sstack.push(state);
                 if (str.unicode()[pos + 1].unicode() == '(') {
                     Save sav = {str, pos + 2};
                     ostack.push(sav);
                     state.current = math;
                     pos += 2;
                     continue;
                 } else {
                     state.current = paren;
                     state.dquote = false;
                 }
             } else if (cc == '{') {
                 sstack.push(state);
                 state.current = subst;
             } else if (!state.dquote) {
                 if (cc == '\'') {
                     sstack.push(state);
                     state.current = dollarquote;
                 } else if (cc == '"') {
                     sstack.push(state);
                     state.current = doublequote;
                     state.dquote = true;
                 }
             }
             // always swallow the char -> prevent anomalies due to expansion
         } else if (cc == '`') {
             str.replace(pos, 1, QStringLiteral("$( ")); // add space -> avoid creating $((
             pos2 = pos += 3;
             for (;;) {
                 if (pos2 >= str.length()) {
                     pos = pos2;
                     return false;
                 }
                 cc = str.unicode()[pos2].unicode();
                 if (cc == '`') {
                     break;
                 }
                 if (cc == '\\') {
                     cc = str.unicode()[++pos2].unicode();
                     if (cc == '$' || cc == '`' || cc == '\\' || (cc == '"' && state.dquote)) {
                         str.remove(pos2 - 1, 1);
                         continue;
                     }
                 }
                 pos2++;
             }
             str[pos2] = QLatin1Char(')');
             sstack.push(state);
             state.current = paren;
             state.dquote = false;
             continue;
         } else if (state.current == doublequote) {
             if (cc == '"') {
                 state = sstack.pop();
             }
         } else if (cc == '\'') {
             if (!state.dquote) {
                 sstack.push(state);
                 state.current = singlequote;
             }
         } else if (cc == '"') {
             if (!state.dquote) {
                 sstack.push(state);
                 state.current = doublequote;
                 state.dquote = true;
             }
         } else if (state.current == subst) {
             if (cc == '}') {
                 state = sstack.pop();
             }
         } else if (cc == ')') {
             if (state.current == math) {
                 if (str.unicode()[pos + 1].unicode() == ')') {
                     state = sstack.pop();
                     pos += 2;
                 } else {
                     // false hit: the $(( was a $( ( in fact
                     // ash does not care, but bash does
                     pos = ostack.top().pos;
                     str = ostack.top().str;
                     ostack.pop();
                     state.current = paren;
                     state.dquote = false;
                     sstack.push(state);
                 }
                 continue;
             } else if (state.current == paren) {
                 state = sstack.pop();
             } else {
                 break;
             }
         } else if (cc == '}') {
             if (state.current == KMacroExpander::group) {
                 state = sstack.pop();
             } else {
                 break;
             }
         } else if (cc == '(') {
             sstack.push(state);
             state.current = paren;
         } else if (cc == '{') {
             sstack.push(state);
             state.current = KMacroExpander::group;
         }
         pos++;
     }
     return sstack.empty();
 }