move lib/MiddleWare/RPM.{h,cpp} to src/Daemon, the sole user

Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>

3 files changed, 281 insertions, 0 deletions

diff --git a/src/Daemon/Makefile.am b/src/Daemon/Makefile.am
index af2f406e..eb9a0594 100644
--- a/src/Daemon/Makefile.am
+++ b/src/Daemon/Makefile.am
@@ -3,6 +3,7 @@ sbin_PROGRAMS = abrt
 abrt_SOURCES = \
     ABRTPlugin.cpp ABRTPlugin.h \
     PluginManager.cpp PluginManager.h \
+    RPM.cpp RPM.h \
     MiddleWare.cpp MiddleWare.h \
     CrashWatcher.cpp CrashWatcher.h \
     CommLayerServer.h CommLayerServer.cpp \
diff --git a/src/Daemon/RPM.cpp b/src/Daemon/RPM.cpp
new file mode 100644
index 00000000..6659457f
--- /dev/null
+++ b/src/Daemon/RPM.cpp
@@ -0,0 +1,180 @@
+#include "RPM.h"
+#include "CommLayerInner.h"
+
+CRPM::CRPM()
+{
+    char *argv[] = {(char*)""};
+    m_poptContext = rpmcliInit(0, argv, NULL);
+}
+
+CRPM::~CRPM()
+{
+    rpmcliFini(m_poptContext);
+}
+
+void CRPM::LoadOpenGPGPublicKey(const std::string& pFileName)
+{
+    uint8_t* pkt = NULL;
+    size_t pklen;
+    pgpKeyID_t keyID;
+    if (pgpReadPkts(pFileName.c_str(), &pkt, &pklen) != PGPARMOR_PUBKEY)
+    {
+        free(pkt);
+        comm_layer_inner_warning("CRPM::LoadOpenGPGPublicKey(): Can not load public key " + pFileName);
+        return;
+    }
+    if (pgpPubkeyFingerprint(pkt, pklen, keyID) == 0)
+    {
+        char* fedoraFingerprint = pgpHexStr(keyID, sizeof(keyID));
+        if (fedoraFingerprint != NULL)
+        {
+            m_setFingerprints.insert(fedoraFingerprint);
+        }
+    }
+    free(pkt);
+}
+
+bool CRPM::CheckFingerprint(const std::string& pPackage)
+{
+    bool ret = false;
+    rpmts ts = rpmtsCreate();
+    rpmdbMatchIterator iter = rpmtsInitIterator(ts, RPMTAG_NAME, pPackage.c_str(), 0);
+    Header header;
+
+    if ((header = rpmdbNextIterator(iter)) != NULL)
+    {
+        rpmTag rpmTags[] = { RPMTAG_DSAHEADER, RPMTAG_RSAHEADER, RPMTAG_SHA1HEADER };
+        int ii;
+        for (ii = 0; ii < 3; ii++)
+        {
+            if (headerIsEntry(header, rpmTags[ii]))
+            {
+                rpmtd td = rpmtdNew();
+                headerGet(header, rpmTags[ii] , td, HEADERGET_DEFAULT);
+                char* pgpsig = rpmtdFormat(td, RPMTD_FORMAT_PGPSIG , NULL);
+                if (pgpsig)
+                {
+                    std::string PGPSignatureText = pgpsig;
+                    free(pgpsig);
+
+                    if (PGPSignatureText.find(" Key ID ") != std::string::npos)
+                    {
+                        std::string headerFingerprint = PGPSignatureText.substr(PGPSignatureText.find(" Key ID ") + sizeof (" Key ID ") - 1);
+
+                        rpmtdFree(td);
+                        if (headerFingerprint != "")
+                        {
+                            if (m_setFingerprints.find(headerFingerprint) != m_setFingerprints.end())
+                            {
+                                ret = true;
+                                break;
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
+    rpmdbFreeIterator(iter);
+    rpmtsFree(ts);
+    return ret;
+}
+
+bool CRPM::CheckHash(const std::string& pPackage, const std::string& pPath)
+{
+    bool ret = false;
+    rpmts ts = rpmtsCreate();
+    rpmdbMatchIterator iter = rpmtsInitIterator(ts, RPMTAG_NAME, pPackage.c_str(), 0);
+    Header header;
+    if ((header = rpmdbNextIterator(iter)) != NULL)
+    {
+        rpmfi fi = rpmfiNew(ts, header, RPMTAG_BASENAMES, RPMFI_NOHEADER);
+        pgpHashAlgo hashAlgo;
+        std::string headerHash;
+        char computedHash[1024] = "";
+
+        while(rpmfiNext(fi) != -1)
+        {
+            if (pPath == rpmfiFN(fi))
+            {
+                headerHash = rpmfiFDigestHex(fi, &hashAlgo);
+            }
+        }
+        rpmfiFree(fi);
+
+        rpmDoDigest(hashAlgo, pPath.c_str(), 1, (unsigned char*) computedHash, NULL);
+
+        if (headerHash != "" && headerHash == computedHash)
+        {
+            ret = true;
+        }
+    }
+    rpmdbFreeIterator(iter);
+    rpmtsFree(ts);
+    return ret;
+}
+
+std::string CRPM::GetDescription(const std::string& pPackage)
+{
+    std::string pDescription = "";
+    rpmts ts = rpmtsCreate();
+    rpmdbMatchIterator iter = rpmtsInitIterator(ts, RPMTAG_NAME, pPackage.c_str(), 0);
+    Header header;
+    if ((header = rpmdbNextIterator(iter)) != NULL)
+    {
+        rpmtd td = rpmtdNew();
+        headerGet(header, RPMTAG_SUMMARY, td, HEADERGET_DEFAULT);
+        const char* summary = rpmtdGetString(td);
+        headerGet(header, RPMTAG_DESCRIPTION, td, HEADERGET_DEFAULT);
+        const char* description = rpmtdGetString(td);
+        pDescription = summary + std::string("\n\n") + description;
+        rpmtdFree(td);
+
+    }
+    rpmdbFreeIterator(iter);
+    rpmtsFree(ts);
+    return pDescription;
+}
+std::string CRPM::GetComponent(const std::string& pFileName)
+{
+    std::string ret = "";
+    rpmts ts = rpmtsCreate();
+    rpmdbMatchIterator iter = rpmtsInitIterator(ts, RPMTAG_BASENAMES, pFileName.c_str(), 0);
+    Header header;
+    if ((header = rpmdbNextIterator(iter)) != NULL)
+    {
+        rpmtd td = rpmtdNew();
+        headerGet(header,RPMTAG_SOURCERPM, td, HEADERGET_DEFAULT);
+        const char * srpm = rpmtdGetString(td);
+        if (srpm != NULL)
+        {
+            std::string srcrpm(srpm);
+            ret = srcrpm.erase(srcrpm.rfind('-',srcrpm.rfind('-')-1));
+        }
+        rpmtdFree(td);
+    }
+
+    rpmdbFreeIterator(iter);
+    rpmtsFree(ts);
+    return ret;
+}
+std::string CRPM::GetPackage(const std::string& pFileName)
+{
+    std::string ret = "";
+    rpmts ts = rpmtsCreate();
+    rpmdbMatchIterator iter = rpmtsInitIterator(ts, RPMTAG_BASENAMES, pFileName.c_str(), 0);
+    Header header;
+    if ((header = rpmdbNextIterator(iter)) != NULL)
+    {
+        char* nerv = headerGetNEVR(header, NULL);
+        if (nerv != NULL)
+        {
+            ret = nerv;
+            free(nerv);
+        }
+    }
+
+    rpmdbFreeIterator(iter);
+    rpmtsFree(ts);
+    return ret;
+}
diff --git a/src/Daemon/RPM.h b/src/Daemon/RPM.h
new file mode 100644
index 00000000..4fe1eab4
--- /dev/null
+++ b/src/Daemon/RPM.h
@@ -0,0 +1,100 @@
+/*
+    RPMInfo.h - header file for rpm database
+              - it implements query for local rpm database
+
+    Copyright (C) 2009  Zdenek Prikryl (zprikryl@redhat.com)
+    Copyright (C) 2009  RedHat inc.
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License along
+    with this program; if not, write to the Free Software Foundation, Inc.,
+    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+    */
+
+#ifndef RPMINFO_H_
+#define RPMINFO_H_
+
+#include "MiddleWareTypes.h"
+
+#include <rpm/rpmcli.h>
+#include <rpm/rpmts.h>
+#include <rpm/rpmdb.h>
+
+/**
+ * A class. It is used for additional checks of package, which contains
+ * crashed application.
+ */
+class CRPM
+{
+    private:
+
+        typedef set_strings_t set_fingerprints_t;
+
+        /**
+         * A context for librpm library.
+         */
+        poptContext m_poptContext;
+        /**
+         * A set, which contains finger prints.
+         */
+        set_fingerprints_t m_setFingerprints;
+
+    public:
+        /**
+         * A constructior.
+         */
+        CRPM();
+        /**
+         * A destructor.
+         */
+        ~CRPM();
+        /**
+         * A method, which loads one GPG public key.
+         * @param pFileName A path to the public key.
+         */
+        void LoadOpenGPGPublicKey(const std::string& pFileName);
+        /**
+         * A method, which checks if package's finger print is valid.
+         * @param pPackage A package name.
+         */
+        bool CheckFingerprint(const std::string& pPackage);
+        /**
+         * A method, which checks if an application is modified by thir party.
+         * @param pPackage A package name. The package contains the application.
+         * @param pPath A path to the application.
+         */
+        bool CheckHash(const std::string& pPackage, const std::string& pPath);
+        /**
+         * A method, which gets a package description.
+         * @param pPackage A package name.
+         * @return A package description.
+         */
+        std::string GetDescription(const std::string& pPackage);
+        /**
+         * A method, which gets a package name. This package contains particular
+         * file. If the file doesn't belong to any package, empty string is
+         * returned.
+         * @param pFileName A file name.
+         * @return A package name.
+         */
+        std::string GetPackage(const std::string& pFileName);
+        /**
+         * A method to find a main package for given file. This package contains particular
+         * file. If the file doesn't belong to any package, empty string is
+         * returned.
+         * @param pFileName A file name.
+         * @return A package name.
+         */
+        std::string GetComponent(const std::string& pFileName);
+};
+
+#endif /* RPMINFO_H_ */