transport/sqlite/src/aaSqliteProtocolHandler.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim:set ts=2 sw=2 sts=2 et cindent tw=79 ft=cpp: */
   3 /*
   4  * Copyright (C) 2008-2010 Sergey Yanovich <ynvich@gmail.com>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 2 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public
  17  * License along with this program; if not, write to the
  18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  */
  21
  22 #include "aaSqliteProtocolHandler.h"
  23
  24 #include "nsStringAPI.h"
  25 #include "nsCOMPtr.h"
  26 #include "nsAutoPtr.h"
  27 #include "nsIURI.h"
  28 #include "nsIStandardURL.h"
  29 #include "nsNetCID.h"
  30 #include "nsComponentManagerUtils.h"
  31
  32 #include "aaSqlitePump.h"
  33 #include "aaSqliteChannel.h"
  34
  35 NS_IMPL_ISUPPORTS1(aaSqliteProtocolHandler, nsIProtocolHandler)
  36
  37 /* readonly attribute ACString scheme; */
  38 NS_IMETHODIMP
  39 aaSqliteProtocolHandler::GetScheme(nsACString & aScheme)
  40 {
  41   aScheme.AssignLiteral(AA_SQLITE_PROTOCOL);
  42   return NS_OK;
  43 }
  44
  45 /* readonly attribute long defaultPort; */
  46 NS_IMETHODIMP
  47 aaSqliteProtocolHandler::GetDefaultPort(PRInt32 *aDefaultPort)
  48 {
  49   NS_ENSURE_ARG_POINTER(aDefaultPort);
  50   *aDefaultPort = -1;
  51   return NS_OK;
  52 }
  53
  54 /* readonly attribute unsigned long protocolFlags; */
  55 NS_IMETHODIMP
  56 aaSqliteProtocolHandler::GetProtocolFlags(PRUint32 *aProtocolFlags)
  57 {
  58   NS_ENSURE_ARG_POINTER(aProtocolFlags);
  59   *aProtocolFlags = URI_STD | URI_IS_LOCAL_FILE;
  60   return NS_OK;
  61 }
  62
  63 /* nsIURI newURI (in AUTF8String aSpec, in string aOriginCharset, in nsIURI aBaseURI); */
  64 NS_IMETHODIMP
  65 aaSqliteProtocolHandler::NewURI(const nsACString & aSpec,
  66   const char *aOriginCharset, nsIURI *aBaseURI, nsIURI **_retval NS_OUTPARAM)
  67 {
  68   NS_ENSURE_ARG(!aSpec.IsEmpty());
  69   NS_ENSURE_ARG_POINTER(_retval);
  70   nsresult rv = NS_OK;
  71
  72   nsCOMPtr<nsIURI> uri = do_CreateInstance(NS_STANDARDURL_CONTRACTID, &rv);
  73   NS_ENSURE_SUCCESS(rv, rv);
  74
  75   nsCOMPtr<nsIStandardURL> surl = do_QueryInterface(uri, &rv);
  76   NS_ENSURE_SUCCESS(rv, rv);
  77
  78   rv = surl->Init(nsIStandardURL::URLTYPE_STANDARD, -1, aSpec, aOriginCharset,
  79                   aBaseURI);
  80   NS_ENSURE_SUCCESS(rv, rv);
  81
  82   rv = surl->SetMutable(PR_FALSE);
  83   NS_ENSURE_SUCCESS(rv, rv);
  84   NS_IF_ADDREF(*_retval = uri);
  85   return NS_OK;
  86 }
  87
  88 /* nsIChannel newChannel (in nsIURI aURI); */
  89 NS_IMETHODIMP
  90 aaSqliteProtocolHandler::NewChannel(nsIURI *aURI, nsIChannel **_retval NS_OUTPARAM)
  91 {
  92   NS_ENSURE_ARG_POINTER(aURI);
  93   NS_ENSURE_ARG_POINTER(_retval);
  94
  95   nsresult rv = NS_OK;
  96   PRBool sqliteScheme = PR_FALSE;
  97   rv = aURI->SchemeIs(AA_SQLITE_PROTOCOL, &sqliteScheme);
  98   NS_ENSURE_SUCCESS(rv, rv);
  99   NS_ENSURE_TRUE(sqliteScheme, NS_ERROR_FAILURE);
 100
 101   nsCOMPtr<nsIChannel> channel = new aaSqliteChannel(aURI);
 102   NS_ENSURE_TRUE(channel, NS_ERROR_OUT_OF_MEMORY);
 103
 104   NS_IF_ADDREF(*_retval = channel);
 105
 106   return NS_OK;
 107 }
 108
 109 /* boolean allowPort (in long port, in string scheme); */
 110 NS_IMETHODIMP
 111 aaSqliteProtocolHandler::AllowPort(PRInt32 port, const char *scheme,
 112   PRBool *_retval NS_OUTPARAM)
 113 {
 114   return NS_ERROR_NOT_IMPLEMENTED;
 115 }
 116