File indexing completed on 2023-09-24 04:04:53

 /*  This file is part of the KDE libraries
  *  Copyright (C) 1999 Waldo Bastian <bastian@kde.org>
  *                2000-2007 David Faure <faure@kde.org>
  *
  *  This library is free software; you can redistribute it and/or
  *  modify it under the terms of the GNU Library General Public
  *  License version 2 as published by the Free Software Foundation;
  *
  *  This library 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
  *  Library General Public License for more details.
  *
  *  You should have received a copy of the GNU Library General Public License
  *  along with this library; see the file COPYING.LIB.  If not, write to
  *  the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  *  Boston, MA 02110-1301, USA.
  **/
 
 #include "kmimetype.h"
 #include "kmimetype_p.h"
 #include "kmimetyperepository_p.h"
 #include "qmimedatabase.h"
 
 #include <kdebug.h>
 #include <kde_file.h> // KDE::stat
 #include <kdeversion.h> // KDE_MAKE_VERSION
 #include <klocalizedstring.h>
 #include <qstandardpaths.h>
 #include <kurl.h>
 #include <kio/global.h>
 
 #include <QFile>
 
 #ifdef Q_OS_WIN
 #include <windows.h>
 #endif
 
 #ifndef S_ISSOCK
 #define S_ISSOCK(x) false
 #endif
 
 extern int servicesDebugArea();
 
 template class QExplicitlySharedDataPointer<KMimeType>;
 
 KMimeType::Ptr KMimeType::defaultMimeTypePtr()
 {
     return KMimeTypeRepository::self()->defaultMimeTypePtr();
 }
 
 bool KMimeType::isDefault() const
 {
     return name() == defaultMimeType();
 }
 
 KMimeType::Ptr KMimeType::mimeType(const QString &name, FindByNameOption options)
 {
     Q_UNUSED(options);
     QMimeDatabase db;
     QMimeType mime = db.mimeTypeForName(name);
     if (mime.isValid()) {
         return KMimeType::Ptr(new KMimeType(mime));
     } else {
         return KMimeType::Ptr();
     }
 }
 
 KMimeType::List KMimeType::allMimeTypes()
 {
     // This could be done faster...
     KMimeType::List lst;
     QMimeDatabase db;
     Q_FOREACH (const QMimeType &mimeType, db.allMimeTypes()) {
         Q_ASSERT(!mimeType.name().startsWith(QLatin1String("x-scheme-handler")));
         lst.append(KMimeType::Ptr(new KMimeType(mimeType)));
     }
     return lst;
 }
 
 // TODO used outside of kmimetype?
 bool KMimeType::isBufferBinaryData(const QByteArray &data)
 {
     // Check the first 32 bytes (see shared-mime spec)
     const char *p = data.data();
     const int end = qMin(32, data.size());
     for (int i = 0; i < end; ++i) {
         if ((unsigned char)(p[i]) < 32 && p[i] != 9 && p[i] != 10 && p[i] != 13) { // ASCII control character
             return true;
         }
     }
     return false;
 }
 
 // the windows-specific code, and the locked directory, are missing from QMimeDatabase.
 static KMimeType::Ptr findFromMode(const QString &path /*only used if is_local_file*/,
                                    mode_t mode /*0 if unknown*/,
                                    bool is_local_file)
 {
     if (is_local_file && (mode == 0 || mode == (mode_t) - 1)) {
         KDE_struct_stat buff;
         if (KDE::stat(path, &buff) != -1) {
             mode = buff.st_mode;
         }
     }
 
     if (S_ISDIR(mode)) {
         // KDE4 TODO: use an overlay instead
 #if 0
         // Special hack for local files. We want to see whether we
         // are allowed to enter the directory
         if (is_local_file) {
             if (KDE::access(path, R_OK) == -1) {
                 return KMimeType::mimeType("inode/directory-locked");
             }
         }
 #endif
         return KMimeType::mimeType(QLatin1String("inode/directory"));
     }
     if (S_ISCHR(mode)) {
         return KMimeType::mimeType(QLatin1String("inode/chardevice"));
     }
     if (S_ISBLK(mode)) {
         return KMimeType::mimeType(QLatin1String("inode/blockdevice"));
     }
     if (S_ISFIFO(mode)) {
         return KMimeType::mimeType(QLatin1String("inode/fifo"));
     }
     if (S_ISSOCK(mode)) {
         return KMimeType::mimeType(QLatin1String("inode/socket"));
     }
 #ifdef Q_OS_WIN
     // FIXME: distinguish between mounted & unmounted
     int size = path.size();
     if (size == 2 || size == 3) {
         //GetDriveTypeW is not defined in wince
 #ifndef _WIN32_WCE
         unsigned int type = GetDriveTypeW((LPCWSTR) path.utf16());
         switch (type) {
         case DRIVE_REMOVABLE:
             return KMimeType::mimeType(QLatin1String("media/floppy_mounted"));
         case DRIVE_FIXED:
             return KMimeType::mimeType(QLatin1String("media/hdd_mounted"));
         case DRIVE_REMOTE:
             return KMimeType::mimeType(QLatin1String("media/smb_mounted"));
         case DRIVE_CDROM:
             return KMimeType::mimeType(QLatin1String("media/cdrom_mounted"));
         case DRIVE_RAMDISK:
             return KMimeType::mimeType(QLatin1String("media/hdd_mounted"));
         default:
             break;
         };
 #else
         return KMimeType::mimeType(QLatin1String("media/hdd_mounted"));
 #endif
     }
 #endif
     // remote executable file? stop here (otherwise findFromContent can do that better for local files)
     if (!is_local_file && S_ISREG(mode) && (mode & (S_IXUSR | S_IXGRP | S_IXOTH))) {
         return KMimeType::mimeType(QLatin1String("application/x-executable"));
     }
 
     return KMimeType::Ptr();
 }
 
 /*
  Note: in KDE we want the file views to sniff in a delayed manner.
  So there's also a fast mode which is:
   if no glob matches, or if more than one glob matches, use default mimetype and mark as "can be refined".
 */
 
 KMimeType::Ptr KMimeType::findByUrl(const QUrl &url, mode_t mode,
                                     bool is_local_file, bool fast_mode,
                                     int *accuracy)
 {
     Q_UNUSED(mode); // special devices only matter locally; caller can use S_ISDIR by itself.
     Q_UNUSED(is_local_file); // QUrl can tell us...
     QMimeDatabase db;
     if (accuracy) {
         *accuracy = 80;    // not supported anymore; was it really used for anything?
     }
     if (fast_mode) {
         return KMimeType::Ptr(new KMimeType(db.mimeTypeForFile(url.path(), QMimeDatabase::MatchExtension)));
     }
     return KMimeType::Ptr(new KMimeType(db.mimeTypeForUrl(url)));
 }
 
 KMimeType::Ptr KMimeType::findByPath(const QString &path, mode_t mode,
                                      bool fast_mode, int *accuracy)
 {
     Q_UNUSED(mode); // special devices only matter locally; caller can use S_ISDIR by itself.
     QMimeDatabase db;
     if (accuracy) {
         *accuracy = 80;    // not supported anymore; was it really used for anything?
     }
     if (fast_mode) {
         return KMimeType::Ptr(new KMimeType(db.mimeTypeForFile(path, QMimeDatabase::MatchExtension)));
     }
     return KMimeType::Ptr(new KMimeType(db.mimeTypeForFile(path)));
 }
 
 KMimeType::Ptr KMimeType::findByNameAndContent(const QString &name, const QByteArray &data,
         mode_t mode, int *accuracy)
 {
     Q_UNUSED(mode); // If we have data, then this is a regular file anyway...
     if (accuracy) {
         *accuracy = 80;    // not supported anymore; was it really used for anything?
     }
     QMimeDatabase db;
     return KMimeType::Ptr(new KMimeType(db.mimeTypeForFileNameAndData(name, data)));
 }
 
 KMimeType::Ptr KMimeType::findByNameAndContent(const QString &name, QIODevice *device,
         mode_t mode, int *accuracy)
 {
     Q_UNUSED(mode); // If we have data, then this is a regular file anyway...
     if (accuracy) {
         *accuracy = 80;    // not supported anymore; was it really used for anything?
     }
     QMimeDatabase db;
     return KMimeType::Ptr(new KMimeType(db.mimeTypeForFileNameAndData(name, device)));
 }
 
 QString KMimeType::extractKnownExtension(const QString &fileName)
 {
     QMimeDatabase db;
     return db.suffixForFileName(fileName);
 }
 
 KMimeType::Ptr KMimeType::findByContent(const QByteArray &data, int *accuracy)
 {
     QMimeDatabase db;
     if (accuracy) {
         *accuracy = 80;    // not supported anymore; was it really used for anything?
     }
     return KMimeType::Ptr(new KMimeType(db.mimeTypeForData(data)));
 }
 
 KMimeType::Ptr KMimeType::findByContent(QIODevice *device, int *accuracy)
 {
     QMimeDatabase db;
     if (accuracy) {
         *accuracy = 80;    // not supported anymore; was it really used for anything?
     }
     return KMimeType::Ptr(new KMimeType(db.mimeTypeForData(device)));
 }
 
 KMimeType::Ptr KMimeType::findByFileContent(const QString &fileName, int *accuracy)
 {
     QFile device(fileName);
 #if 1
     // Look at mode first
     KMimeType::Ptr mimeFromMode = findFromMode(fileName, 0, true);
     if (mimeFromMode) {
         if (accuracy) {
             *accuracy = 100;
         }
         return mimeFromMode;
     }
 #endif
     QMimeDatabase db;
     KMimeType::Ptr mime(new KMimeType(db.mimeTypeForData(&device)));
     if (accuracy) {
         *accuracy = mime->isDefault() ? 0 : 80;    // not supported anymore; was it really used for anything?
     }
     return mime;
 }
 
 bool KMimeType::isBinaryData(const QString &fileName)
 {
     QFile file(fileName);
     if (!file.open(QIODevice::ReadOnly)) {
         return false;    // err, whatever
     }
     const QByteArray data = file.read(32);
     return isBufferBinaryData(data);
 }
 
 KMimeType::KMimeType(const QMimeType &mime)
     : d_ptr(new KMimeTypePrivate(mime))
 {
 }
 
 KMimeType::~KMimeType()
 {
     delete d_ptr;
 }
 
 QString KMimeType::favIconForUrl(const QUrl &url)
 {
     return KIO::favIconForUrl(url);
 }
 
 QString KMimeType::iconNameForUrl(const QUrl &url, mode_t mode)
 {
     Q_UNUSED(mode)
     return KIO::iconNameForUrl(url);
 }
 
 QString KMimeType::comment() const
 {
     return d_ptr->m_qmime.comment();
 }
 
 #ifndef KDELIBS4SUPPORT_NO_DEPRECATED
 QString KMimeType::parentMimeType() const
 {
     const QStringList parents = d_ptr->m_qmime.parentMimeTypes();
     if (!parents.isEmpty()) {
         return parents.first();
     }
     return QString();
 }
 #endif
 
 bool KMimeType::is(const QString &mimeTypeName) const
 {
     return d_ptr->m_qmime.inherits(mimeTypeName);
 }
 
 QStringList KMimeType::parentMimeTypes() const
 {
     return d_ptr->m_qmime.parentMimeTypes();
 }
 
 QStringList KMimeType::allParentMimeTypes() const
 {
     return d_ptr->m_qmime.allAncestors();
 }
 
 QString KMimeType::defaultMimeType()
 {
     return QLatin1String("application/octet-stream");
 }
 
 QString KMimeType::iconName() const
 {
     return d_ptr->m_qmime.iconName();
 }
 
 QStringList KMimeType::patterns() const
 {
     return d_ptr->m_qmime.globPatterns();
 }
 
 // TODO MOVE TO keditfiletype/mimetypedata.cpp
 QString KMimeType::userSpecifiedIconName() const
 {
     //d->ensureXmlDataLoaded();
     //return d->m_iconName;
     return QString();
 }
 
 int KMimeType::sharedMimeInfoVersion()
 {
     return KMimeTypeRepository::self()->sharedMimeInfoVersion();
 }
 
 QString KMimeType::mainExtension() const
 {
 #if 1 // HACK START - can be removed once shared-mime-info >= 0.70 is used/required.
     // The idea was: first usable pattern from m_lstPatterns.
     // But update-mime-database makes a mess of the order of the patterns,
     // because it uses a hash internally.
     static const struct {
         const char *mime;
         const char *extension;
     } s_hardcodedMimes[] = {
         { "text/plain", ".txt" }
     };
     if (patterns().count() > 1) {
         const QByteArray me = name().toLatin1();
         for (uint i = 0; i < sizeof(s_hardcodedMimes) / sizeof(*s_hardcodedMimes); ++i) {
             if (me == s_hardcodedMimes[i].mime) {
                 return QString::fromLatin1(s_hardcodedMimes[i].extension);
             }
         }
     }
 #endif // HACK END
 
     Q_FOREACH (const QString &pattern, patterns()) {
         // Skip if if looks like: README or *. or *.*
         // or *.JP*G or *.JP?
         if (pattern.startsWith(QLatin1String("*.")) &&
                 pattern.length() > 2 &&
                 pattern.indexOf(QLatin1Char('*'), 2) < 0 && pattern.indexOf(QLatin1Char('?'), 2) < 0) {
             return pattern.mid(1);
         }
     }
     // TODO we should also look into the parent mimetype's patterns, no?
     return QString();
 }
 
 bool KMimeType::matchFileName(const QString &filename, const QString &pattern)
 {
     return KMimeTypeRepository::matchFileName(filename, pattern);
 }
 
 /*
 int KMimeTypePrivate::serviceOffersOffset() const
 {
     return KMimeTypeFactory::self()->serviceOffersOffset(name());
 }
 */
 
 QString KMimeType::name() const
 {
     return d_ptr->m_qmime.name();
 }