Back to index

lightning-sunbird  0.9+nobinonly
MapiMimeTypes.cpp
Go to the documentation of this file.
00001 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
00002 /* ***** BEGIN LICENSE BLOCK *****
00003  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
00004  *
00005  * The contents of this file are subject to the Mozilla Public License Version
00006  * 1.1 (the "License"); you may not use this file except in compliance with
00007  * the License. You may obtain a copy of the License at
00008  * http://www.mozilla.org/MPL/
00009  *
00010  * Software distributed under the License is distributed on an "AS IS" basis,
00011  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
00012  * for the specific language governing rights and limitations under the
00013  * License.
00014  *
00015  * The Original Code is mozilla.org code.
00016  *
00017  * The Initial Developer of the Original Code is
00018  * Netscape Communications Corporation.
00019  * Portions created by the Initial Developer are Copyright (C) 1998
00020  * the Initial Developer. All Rights Reserved.
00021  *
00022  * Contributor(s):
00023  *
00024  * Alternatively, the contents of this file may be used under the terms of
00025  * either of the GNU General Public License Version 2 or later (the "GPL"),
00026  * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
00027  * in which case the provisions of the GPL or the LGPL are applicable instead
00028  * of those above. If you wish to allow use of your version of this file only
00029  * under the terms of either the GPL or the LGPL, and not to allow others to
00030  * use your version of this file under the terms of the MPL, indicate your
00031  * decision by deleting the provisions above and replace them with the notice
00032  * and other provisions required by the GPL or the LGPL. If you do not delete
00033  * the provisions above, a recipient may use your version of this file under
00034  * the terms of any one of the MPL, the GPL or the LGPL.
00035  *
00036  * ***** END LICENSE BLOCK ***** */
00037 
00038 #include "nscore.h"
00039 #include "nsCRT.h"
00040 #include "nsString.h"
00041 #include "MapiMimeTypes.h"
00042 
00043 PRUint8 CMimeTypes::m_mimeBuffer[kMaxMimeTypeSize];
00044 
00045 
00046 BOOL CMimeTypes::GetKey( HKEY root, LPCTSTR pName, PHKEY pKey)
00047 {
00048        LONG result = RegOpenKeyEx( root, pName, 0, KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS, pKey);
00049        return (result == ERROR_SUCCESS);
00050 }
00051 
00052 BOOL CMimeTypes::GetValueBytes( HKEY rootKey, LPCTSTR pValName, LPBYTE *ppBytes)
00053 {
00054        LONG   err;
00055        DWORD  bufSz;
00056        
00057        *ppBytes = NULL;
00058        // Get the installed directory
00059        err = RegQueryValueEx( rootKey, pValName, NULL, NULL, NULL, &bufSz); 
00060        if (err == ERROR_SUCCESS) {
00061               *ppBytes = new BYTE[bufSz];
00062               err = RegQueryValueEx( rootKey, pValName, NULL, NULL, *ppBytes, &bufSz);
00063               if (err == ERROR_SUCCESS) {
00064                      return( TRUE);
00065               }
00066               delete *ppBytes;
00067               *ppBytes = NULL;
00068        }
00069        return( FALSE);
00070 }
00071 
00072 void CMimeTypes::ReleaseValueBytes( LPBYTE pBytes)
00073 {
00074        if (pBytes)
00075               delete pBytes;
00076 }
00077 
00078 BOOL CMimeTypes::GetMimeTypeFromReg( const nsCString& ext, LPBYTE *ppBytes)
00079 {
00080        HKEY   extensionKey;
00081        BOOL   result = FALSE;
00082 
00083 
00084        *ppBytes = NULL;
00085        if (GetKey( HKEY_CLASSES_ROOT, ext.get(), &extensionKey)) {
00086               result = GetValueBytes( extensionKey, "Content Type", ppBytes);
00087               RegCloseKey( extensionKey);
00088        }
00089        
00090        return( result);
00091 }
00092 
00093 
00094 
00095 PRUint8 * CMimeTypes::GetMimeType( nsCString& theExt)
00096 {
00097        nsCString     ext = theExt;
00098        if (ext.Length()) {
00099               if (ext.First() != '.') {
00100                      ext = ".";
00101                      ext += theExt;
00102               }
00103        }
00104        
00105        
00106        BOOL   result = FALSE;  
00107        int           len;
00108 
00109        if (!ext.Length())
00110               return( NULL);
00111        LPBYTE pByte;
00112        if (GetMimeTypeFromReg( ext, &pByte)) {
00113               len = strlen( (const char *) pByte);
00114               if (len && (len < kMaxMimeTypeSize)) {
00115                      memcpy( m_mimeBuffer, pByte, len);
00116                      m_mimeBuffer[len] = 0;
00117                      result = TRUE;
00118               }
00119               ReleaseValueBytes( pByte);
00120        }
00121 
00122        if (result)
00123               return( m_mimeBuffer);
00124 
00125        return( NULL);
00126 }
00127 
00128