/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ /* ***** BEGIN LICENSE BLOCK ***** * Version: MPL 1.1/GPL 2.0/LGPL 2.1 * * The contents of this file are subject to the Mozilla Public License Version * 1.1 (the "License"); you may not use this file except in compliance with * the License. You may obtain a copy of the License at * http://www.mozilla.org/MPL/ * * Software distributed under the License is distributed on an "AS IS" basis, * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License * for the specific language governing rights and limitations under the * License. * * The Original Code is Mozilla XForms support. * * The Initial Developer of the Original Code is * IBM Corporation. * Portions created by the Initial Developer are Copyright (C) 2004 * the Initial Developer. All Rights Reserved. * * Contributor(s): * Darin Fisher * * Alternatively, the contents of this file may be used under the terms of * either the GNU General Public License Version 2 or later (the "GPL"), or * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), * in which case the provisions of the GPL or the LGPL are applicable instead * of those above. If you wish to allow use of your version of this file only * under the terms of either the GPL or the LGPL, and not to allow others to * use your version of this file under the terms of the MPL, indicate your * decision by deleting the provisions above and replace them with the notice * and other provisions required by the GPL or the LGPL. If you do not delete * the provisions above, a recipient may use your version of this file under * the terms of any one of the MPL, the GPL or the LGPL. * * ***** END LICENSE BLOCK ***** */ #ifdef DEBUG_darinf #include #define LOG(args) printf args #else #define LOG(args) #endif #include #include "nsXFormsSubmissionElement.h" #include "nsIInstanceElementPrivate.h" #include "nsIXTFGenericElementWrapper.h" #include "nsIDOMDocument.h" #include "nsIDOMElement.h" #include "nsIDOMText.h" #include "nsIDOMCDATASection.h" #include "nsIDOMEvent.h" #include "nsIDOMDocumentEvent.h" #include "nsIDOMEventTarget.h" #include "nsIDOMEventListener.h" #include "nsIDOM3Node.h" #include "nsIDOMXMLDocument.h" #include "nsIDOMXPathResult.h" #include "nsIDOMSerializer.h" #include "nsIDOMDOMImplementation.h" #include "nsIDOMProcessingInstruction.h" #include "nsIDOMParser.h" #include "nsComponentManagerUtils.h" #include "nsIDocShell.h" #include "nsIStringStream.h" #include "nsIInputStream.h" #include "nsIStorageStream.h" #include "nsIMultiplexInputStream.h" #include "nsIMIMEInputStream.h" #include "nsINameSpaceManager.h" #include "nsIDocument.h" #include "nsIFileURL.h" #include "nsIMIMEService.h" #include "nsIUploadChannel.h" #include "nsIHttpChannel.h" #include "nsIScriptSecurityManager.h" #include "nsIPipe.h" #include "nsLinebreakConverter.h" #include "nsEscape.h" #include "nsString.h" #include "nsMemory.h" #include "nsCOMPtr.h" #include "nsNetUtil.h" #include "nsXFormsUtils.h" // namespace literals #define NAMESPACE_XML_SCHEMA \ NS_LITERAL_STRING("http://www.w3.org/2001/XMLSchema") #define NAMESPACE_XML_SCHEMA_INSTANCE \ NS_LITERAL_STRING("http://www.w3.org/2001/XMLSchema-instance") // submission methods #define METHOD_GET 0x01 #define METHOD_POST 0x02 #define METHOD_PUT 0x04 // submission encodings #define ENCODING_XML 0x10 // application/xml #define ENCODING_URL 0x20 // application/x-www-form-urlencoded #define ENCODING_MULTIPART_RELATED 0x40 // multipart/related #define ENCODING_MULTIPART_FORM_DATA 0x80 // multipart/form-data struct SubmissionFormat { const char *method; PRUint32 format; }; static const SubmissionFormat sSubmissionFormats[] = { { "post", ENCODING_XML | METHOD_POST }, { "get", ENCODING_URL | METHOD_GET }, { "put", ENCODING_XML | METHOD_PUT }, { "multipart-post", ENCODING_MULTIPART_RELATED | METHOD_POST }, { "form-data-post", ENCODING_MULTIPART_FORM_DATA | METHOD_POST }, { "urlencoded-post", ENCODING_URL | METHOD_POST } }; static PRUint32 GetSubmissionFormat(nsIDOMElement *aElement) { nsAutoString method; aElement->GetAttribute(NS_LITERAL_STRING("method"), method); NS_ConvertUTF16toUTF8 utf8method(method); for (PRUint32 i=0; i mime = do_GetService("@mozilla.org/mime;1"); if (mime) mime->GetTypeFromFile(file, result); if (result.IsEmpty()) result.Assign("application/octet-stream"); } static PRBool HasToken(const nsString &aTokenList, const nsString &aToken) { PRInt32 i = aTokenList.Find(aToken); if (i == kNotFound) return PR_FALSE; // else, check that leading and trailing characters are either // not present or whitespace (#x20, #x9, #xD or #xA). if (i > 0) { PRUnichar c = aTokenList[i - 1]; if (!(c == 0x20 || c == 0x9 || c == 0xD || c == 0xA)) return PR_FALSE; } if (i + aToken.Length() < aTokenList.Length()) { PRUnichar c = aTokenList[i + aToken.Length()]; if (!(c == 0x20 || c == 0x9 || c == 0xD || c == 0xA)) return PR_FALSE; } return PR_TRUE; } // structure used to store information needed to generate attachments // for multipart/related submission. struct SubmissionAttachment { nsString uri; nsCString cid; }; // an array of SubmissionAttachment objects class SubmissionAttachmentArray : nsVoidArray { public: SubmissionAttachmentArray() {} ~SubmissionAttachmentArray() { for (PRUint32 i=0; iuri = uri; a->cid = cid; AppendElement(a); return NS_OK; } PRUint32 Count() const { return (PRUint32) nsVoidArray::Count(); } SubmissionAttachment *Item(PRUint32 index) { return (SubmissionAttachment *) ElementAt(index); } }; // nsISupports NS_IMPL_ISUPPORTS_INHERITED1(nsXFormsSubmissionElement, nsXFormsStubElement, nsIRequestObserver) // nsIXTFElement NS_IMETHODIMP nsXFormsSubmissionElement::OnDestroyed() { mElement = nsnull; return NS_OK; } NS_IMETHODIMP nsXFormsSubmissionElement::HandleDefault(nsIDOMEvent *aEvent, PRBool *aHandled) { nsAutoString type; aEvent->GetType(type); if (type.EqualsLiteral("xforms-submit")) { nsresult rv = Submit(); if (NS_FAILED(rv)) SubmitEnd(PR_FALSE); *aHandled = PR_TRUE; } else { *aHandled = PR_FALSE; } return NS_OK; } // nsIXTFGenericElement NS_IMETHODIMP nsXFormsSubmissionElement::OnCreated(nsIXTFGenericElementWrapper *aWrapper) { aWrapper->SetNotificationMask(nsIXTFElement::NOTIFY_HANDLE_DEFAULT); nsCOMPtr node; aWrapper->GetElementNode(getter_AddRefs(node)); // It's ok to keep a weak pointer to mElement. mElement will have an // owning reference to this object, so as long as we null out mElement in // OnDestroyed, it will always be valid. mElement = node; NS_ASSERTION(mElement, "Wrapper is not an nsIDOMElement, we'll crash soon"); return NS_OK; } NS_IMETHODIMP nsXFormsSubmissionElement::OnStartRequest(nsIRequest *request, nsISupports *ctx) { return NS_OK; } NS_IMETHODIMP nsXFormsSubmissionElement::OnStopRequest(nsIRequest *request, nsISupports *ctx, nsresult status) { LOG(("xforms submission complete [status=%x]\n", status)); nsCOMPtr channel = do_QueryInterface(request); NS_ASSERTION(channel, "request should be a channel"); PRBool succeeded = NS_SUCCEEDED(status); if (succeeded) { PRUint32 avail = 0; mPipeIn->Available(&avail); if (avail > 0) { nsAutoString replace; mElement->GetAttribute(NS_LITERAL_STRING("replace"), replace); nsresult rv; if (replace.EqualsLiteral("instance")) rv = LoadReplaceInstance(channel); else if (replace.IsEmpty() || replace.EqualsLiteral("all")) rv = LoadReplaceAll(channel); else rv = NS_OK; succeeded = NS_SUCCEEDED(rv); } } mPipeIn = 0; SubmitEnd(succeeded); return NS_OK; } // private methods already_AddRefed nsXFormsSubmissionElement::GetModel() { nsCOMPtr parentNode; mElement->GetParentNode(getter_AddRefs(parentNode)); nsIModelElementPrivate *model = nsnull; if (parentNode) CallQueryInterface(parentNode, &model); return model; } nsresult nsXFormsSubmissionElement::LoadReplaceInstance(nsIChannel *channel) { // replace instance document nsCString contentType, contentCharset; channel->GetContentType(contentType); channel->GetContentCharset(contentCharset); // use DOM parser to construct nsIDOMDocument nsCOMPtr parser = do_CreateInstance("@mozilla.org/xmlextras/domparser;1"); NS_ENSURE_STATE(parser); PRUint32 contentLength; mPipeIn->Available(&contentLength); nsCOMPtr newDoc; parser->ParseFromStream(mPipeIn, contentCharset.get(), contentLength, contentType.get(), getter_AddRefs(newDoc)); NS_ENSURE_STATE(newDoc); // check for parsererror tag? XXX is this needed? or, is there a better way? nsCOMPtr docElem; newDoc->GetDocumentElement(getter_AddRefs(docElem)); if (docElem) { nsAutoString tagName, namespaceURI; docElem->GetTagName(tagName); docElem->GetNamespaceURI(namespaceURI); // XXX this is somewhat of a hack. we should instead be listening for an // 'error' event from the DOM, but gecko doesn't implement that event yet. if (tagName.EqualsLiteral("parsererror") && namespaceURI.EqualsLiteral("http://www.mozilla.org/newlayout/xml/parsererror.xml")) { NS_WARNING("resulting instance document could not be parsed"); return NS_ERROR_UNEXPECTED; } } nsCOMPtr model = GetModel(); if (model) { nsCOMPtr instance; model->FindInstanceElement(EmptyString(), getter_AddRefs(instance)); if (instance) { instance->SetDocument(newDoc); } } return NS_OK; } nsresult nsXFormsSubmissionElement::LoadReplaceAll(nsIChannel *channel) { // use nsIDocShell::loadStream, which may not be perfect ;-) // XXX do we need to transfer nsIChannel::securityInfo ??? nsCOMPtr domDoc; mElement->GetOwnerDocument(getter_AddRefs(domDoc)); nsCOMPtr doc = do_QueryInterface(domDoc); NS_ENSURE_STATE(doc); // the container is the docshell, and we use it as our provider of // notification callbacks. nsCOMPtr container = doc->GetContainer(); nsCOMPtr docshell = do_QueryInterface(container); nsCOMPtr uri; nsCString contentType, contentCharset; channel->GetURI(getter_AddRefs(uri)); channel->GetContentType(contentType); channel->GetContentCharset(contentCharset); return docshell->LoadStream(mPipeIn, uri, contentType, contentCharset, nsnull); } nsresult nsXFormsSubmissionElement::Submit() { LOG(("+++ nsXFormsSubmissionElement::Submit\n")); nsresult rv; // 1. ensure that we are not currently processing a xforms-submit (see E37) NS_ENSURE_STATE(!mSubmissionActive); mSubmissionActive = PR_TRUE; // 2. get selected node from the instance data (use xpath, gives us node // iterator) nsCOMPtr data; rv = GetSelectedInstanceData(getter_AddRefs(data)); NS_ENSURE_TRUE(NS_SUCCEEDED(rv) && data, rv); // 3. revalidate selected instance data (only for namespaces considered for // serialization) // XXX call nsISchemaValidator::validate on each node // 4. serialize instance data PRUint32 format = GetSubmissionFormat(mElement); NS_ENSURE_STATE(format != 0); nsCOMPtr stream; nsCAutoString uri, contentType; rv = SerializeData(data, format, uri, getter_AddRefs(stream), contentType); NS_ENSURE_SUCCESS(rv, rv); // 5. dispatch network request rv = SendData(format, uri, stream, contentType); NS_ENSURE_SUCCESS(rv, rv); return rv; } nsresult nsXFormsSubmissionElement::SubmitEnd(PRBool succeeded) { LOG(("xforms submission complete [%s]\n", succeeded ? "success" : "failure")); // XForms 1.0 errata (E37) limits a element to performing // only one submission at a time, contrary to the spec which places the // limitation on the element. mSubmissionActive = PR_FALSE; nsCOMPtr domDoc; mElement->GetOwnerDocument(getter_AddRefs(domDoc)); nsCOMPtr doc = do_QueryInterface(domDoc); nsCOMPtr event; doc->CreateEvent(NS_LITERAL_STRING("Events"), getter_AddRefs(event)); NS_ENSURE_TRUE(event, NS_ERROR_OUT_OF_MEMORY); event->InitEvent(succeeded ? NS_LITERAL_STRING("xforms-submit-done") : NS_LITERAL_STRING("xforms-submit-error"), PR_TRUE, PR_FALSE); // XForms 1.0 errata (E10) states that the target of the xforms-submit-error // event is the element instead of the element. nsCOMPtr target = do_QueryInterface(mElement); PRBool cancelled; return target->DispatchEvent(event, &cancelled); } nsresult nsXFormsSubmissionElement::GetSelectedInstanceData(nsIDOMNode **result) { nsCOMPtr model; nsCOMPtr bind; nsCOMPtr xpRes = nsXFormsUtils::EvaluateNodeBinding(mElement, 0, NS_LITERAL_STRING("ref"), NS_LITERAL_STRING("/"), nsIDOMXPathResult::FIRST_ORDERED_NODE_TYPE, getter_AddRefs(model), getter_AddRefs(bind)); if (!xpRes) return NS_ERROR_UNEXPECTED; return xpRes->GetSingleNodeValue(result); } PRBool nsXFormsSubmissionElement::GetBooleanAttr(const nsAString &name, PRBool defaultVal) { nsAutoString value; mElement->GetAttribute(name, value); // use defaultVal when value does not match a legal literal if (!value.IsEmpty()) { if (value.EqualsLiteral("true") || value.EqualsLiteral("1")) return PR_TRUE; if (value.EqualsLiteral("false") || value.EqualsLiteral("0")) return PR_FALSE; } return defaultVal; } void nsXFormsSubmissionElement::GetDefaultInstanceData(nsIDOMNode **result) { *result = nsnull; // default element is the first child node of // our parent, which should be a element. nsCOMPtr parent; mElement->GetParentNode(getter_AddRefs(parent)); if (!parent) { NS_WARNING("no parent node!"); return; } nsCOMPtr model = do_QueryInterface(parent); if (!model) { NS_WARNING("parent node is not a model"); return; } nsCOMPtr instanceDoc; model->GetInstanceDocument(EmptyString(), getter_AddRefs(instanceDoc)); nsCOMPtr instanceDocElem; instanceDoc->GetDocumentElement(getter_AddRefs(instanceDocElem)); NS_ADDREF(*result = instanceDocElem); } nsresult nsXFormsSubmissionElement::SerializeData(nsIDOMNode *data, PRUint32 format, nsCString &uri, nsIInputStream **stream, nsCString &contentType) { // initialize uri to the given action nsAutoString action; mElement->GetAttribute(NS_LITERAL_STRING("action"), action); CopyUTF16toUTF8(action, uri); // 'get' method: // The URI is constructed as follows: // o The submit URI from the action attribute is examined. If it does not // already contain a ? (question mark) character, one is appended. If it // does already contain a question mark character, then a separator // character from the attribute separator is appended. // o The serialized form data is appended to the URI. if (format & ENCODING_XML) return SerializeDataXML(data, format, stream, contentType, nsnull); if (format & ENCODING_URL) return SerializeDataURLEncoded(data, format, uri, stream, contentType); if (format & ENCODING_MULTIPART_RELATED) return SerializeDataMultipartRelated(data, format, stream, contentType); if (format & ENCODING_MULTIPART_FORM_DATA) return SerializeDataMultipartFormData(data, format, stream, contentType); NS_WARNING("unsupported submission encoding"); return NS_ERROR_UNEXPECTED; } nsresult nsXFormsSubmissionElement::SerializeDataXML(nsIDOMNode *data, PRUint32 format, nsIInputStream **stream, nsCString &contentType, SubmissionAttachmentArray *attachments) { nsAutoString mediaType; mElement->GetAttribute(NS_LITERAL_STRING("mediaType"), mediaType); if (mediaType.IsEmpty()) contentType.AssignLiteral("application/xml"); else CopyUTF16toUTF8(mediaType, contentType); nsAutoString encoding; mElement->GetAttribute(NS_LITERAL_STRING("encoding"), encoding); if (encoding.IsEmpty()) encoding.AssignLiteral("UTF-8"); nsCOMPtr storage; NS_NewStorageStream(4096, PR_UINT32_MAX, getter_AddRefs(storage)); NS_ENSURE_TRUE(storage, NS_ERROR_OUT_OF_MEMORY); nsCOMPtr sink; storage->GetOutputStream(0, getter_AddRefs(sink)); NS_ENSURE_TRUE(sink, NS_ERROR_OUT_OF_MEMORY); nsCOMPtr serializer = do_GetService("@mozilla.org/xmlextras/xmlserializer;1"); NS_ENSURE_TRUE(serializer, NS_ERROR_UNEXPECTED); nsCOMPtr doc; data->GetOwnerDocument(getter_AddRefs(doc)); // owner doc is null when the data node is the document (e.g., ref="/") if (!doc) doc = do_QueryInterface(data); NS_ENSURE_TRUE(doc, NS_ERROR_UNEXPECTED); // clone and possibly modify the document for submission nsCOMPtr newDoc; CreateSubmissionDoc(doc, encoding, attachments, getter_AddRefs(newDoc)); NS_ENSURE_TRUE(newDoc, NS_ERROR_UNEXPECTED); nsresult rv = serializer->SerializeToStream(newDoc, sink, NS_LossyConvertUTF16toASCII(encoding)); NS_ENSURE_SUCCESS(rv, rv); // close the output stream, so that the input stream will not return // NS_BASE_STREAM_WOULD_BLOCK when it reaches end-of-stream. sink->Close(); return storage->NewInputStream(0, stream); } nsresult nsXFormsSubmissionElement::CreateSubmissionDoc(nsIDOMDocument *source, const nsString &encoding, SubmissionAttachmentArray *attachments, nsIDOMDocument **result) { PRBool indent = GetBooleanAttr(NS_LITERAL_STRING("indent"), PR_FALSE); PRBool omit_xml_declaration = GetBooleanAttr(NS_LITERAL_STRING("omit-xml-declaration"), PR_FALSE); nsAutoString cdataElements; mElement->GetAttribute(NS_LITERAL_STRING("cdata-section-elements"), cdataElements); // XXX cdataElements contains space delimited QNames. these may have // namespace prefixes relative to our document. we need to translate // them to the corresponding namespace prefix in the source document. // // XXX we'll just assume that the QNames correspond to node names since // it's unclear how to resolve the namespace prefixes any other way. nsCOMPtr impl; source->GetImplementation(getter_AddRefs(impl)); NS_ENSURE_TRUE(impl, NS_ERROR_UNEXPECTED); nsCOMPtr doc; impl->CreateDocument(EmptyString(), EmptyString(), nsnull, getter_AddRefs(doc)); NS_ENSURE_TRUE(doc, NS_ERROR_UNEXPECTED); if (!omit_xml_declaration) { nsAutoString buf = NS_LITERAL_STRING("version=\"1.0\" encoding=\"") + encoding + NS_LITERAL_STRING("\""); nsCOMPtr pi; doc->CreateProcessingInstruction(NS_LITERAL_STRING("xml"), buf, getter_AddRefs(pi)); nsCOMPtr newChild; doc->AppendChild(pi, getter_AddRefs(newChild)); } // recursively walk the source document, copying nodes as appropriate CopyChildren(source, doc, doc, attachments, cdataElements, indent, 0); NS_ADDREF(*result = doc); return NS_OK; } nsresult nsXFormsSubmissionElement::CopyChildren(nsIDOMNode *source, nsIDOMNode *dest, nsIDOMDocument *destDoc, SubmissionAttachmentArray *attachments, const nsString &cdataElements, PRBool indent, PRUint32 depth) { nsCOMPtr sourceChild, node, destChild; source->GetFirstChild(getter_AddRefs(sourceChild)); while (sourceChild) { // if not indenting, then strip all unnecessary whitespace // XXX importing the entire node is not quite right here... we also have // to iterate over the attributes since the attributes could somehow // (remains to be determined) reference external entities. destDoc->ImportNode(sourceChild, PR_FALSE, getter_AddRefs(destChild)); NS_ENSURE_TRUE(destChild, NS_ERROR_UNEXPECTED); PRUint16 type; destChild->GetNodeType(&type); if (type == nsIDOMNode::PROCESSING_INSTRUCTION_NODE) { nsCOMPtr pi = do_QueryInterface(destChild); NS_ENSURE_TRUE(pi, NS_ERROR_UNEXPECTED); // ignore "" since we would have already inserted this. nsAutoString target; pi->GetTarget(target); if (!target.EqualsLiteral("xml")) dest->AppendChild(destChild, getter_AddRefs(node)); } else if (type == nsIDOMNode::TEXT_NODE) { // honor cdata-section-elements (see xslt spec section 16.1) if (cdataElements.IsEmpty()) { dest->AppendChild(destChild, getter_AddRefs(node)); } else { sourceChild->GetParentNode(getter_AddRefs(node)); NS_ENSURE_STATE(node); nsAutoString name; node->GetNodeName(name); // check to see if name is mentioned on cdataElements if (HasToken(cdataElements, name)) { nsCOMPtr textNode = do_QueryInterface(destChild); NS_ENSURE_STATE(textNode); nsAutoString textData; textNode->GetData(textData); nsCOMPtr cdataNode; destDoc->CreateCDATASection(textData, getter_AddRefs(cdataNode)); dest->AppendChild(cdataNode, getter_AddRefs(node)); } else { dest->AppendChild(destChild, getter_AddRefs(node)); } } } else { // if |destChild| is an element node of type 'xsd:anyURI', and if we have // an attachments array, then we need to perform multipart/related // processing (i.e., generate a ContentID for the child of this element, // and append a new attachment to the attachments array). PRUint32 encType; if (attachments && NS_SUCCEEDED(GetElementEncodingType(destChild, &encType)) && encType == ELEMENT_ENCTYPE_URI) { // ok, looks like we have a URI to process sourceChild->GetFirstChild(getter_AddRefs(node)); NS_ENSURE_TRUE(node, NS_ERROR_UNEXPECTED); node->GetNodeType(&type); NS_ENSURE_TRUE(type == nsIDOMNode::TEXT_NODE, NS_ERROR_UNEXPECTED); nsString value; node->GetNodeValue(value); nsCString cid; MakeMultipartContentID(cid); nsAutoString cidURI; cidURI.AssignLiteral("cid:"); AppendASCIItoUTF16(cid, cidURI); nsCOMPtr text; destDoc->CreateTextNode(cidURI, getter_AddRefs(text)); NS_ENSURE_TRUE(text, NS_ERROR_UNEXPECTED); destChild->AppendChild(text, getter_AddRefs(node)); dest->AppendChild(destChild, getter_AddRefs(node)); attachments->Append(value, cid); } else { dest->AppendChild(destChild, getter_AddRefs(node)); // recurse nsresult rv = CopyChildren(sourceChild, destChild, destDoc, attachments, cdataElements, indent, depth + 1); NS_ENSURE_SUCCESS(rv, rv); } } sourceChild->GetNextSibling(getter_AddRefs(node)); sourceChild.swap(node); } return NS_OK; } nsresult nsXFormsSubmissionElement::SerializeDataURLEncoded(nsIDOMNode *data, PRUint32 format, nsCString &uri, nsIInputStream **stream, nsCString &contentType) { nsCAutoString separator; { nsAutoString temp; mElement->GetAttribute(NS_LITERAL_STRING("separator"), temp); if (temp.IsEmpty()) { separator.AssignLiteral(";"); } else { // XXX validate input? take only the first character? CopyUTF16toUTF8(temp, separator); } } if (format & METHOD_GET) { if (uri.FindChar('?') == kNotFound) uri.Append('?'); else uri.Append(separator); AppendURLEncodedData(data, separator, uri); *stream = nsnull; contentType.Truncate(); } else if (format & METHOD_POST) { nsCAutoString buf; AppendURLEncodedData(data, separator, buf); // make new stream NS_NewCStringInputStream(stream, buf); NS_ENSURE_TRUE(*stream, NS_ERROR_UNEXPECTED); contentType.AssignLiteral("application/x-www-form-urlencoded"); } else { NS_WARNING("unexpected submission format"); return NS_ERROR_UNEXPECTED; } return NS_OK; } void nsXFormsSubmissionElement::AppendURLEncodedData(nsIDOMNode *data, const nsCString &separator, nsCString &buf) { // 1. Each element node is visited in document order. Each element that has // one text node child is selected for inclusion. // 2. Element nodes selected for inclusion are encoded as EltName=value{sep}, // where = is a literal character, {sep} is the separator character from the // separator attribute on submission, EltName represents the element local // name, and value represents the contents of the text node. // NOTE: // The encoding of EltName and value are as follows: space characters are // replaced by +, and then non-ASCII and reserved characters (as defined // by [RFC 2396] as amended by subsequent documents in the IETF track) are // escaped by replacing the character with one or more octets of the UTF-8 // representation of the character, with each octet in turn replaced by // %HH, where HH represents the uppercase hexadecimal notation for the // octet value and % is a literal character. Line breaks are represented // as "CR LF" pairs (i.e., %0D%0A). #ifdef DEBUG_darinf nsAutoString nodeName; data->GetNodeName(nodeName); LOG(("+++ AppendURLEncodedData: inspecting <%s>\n", NS_ConvertUTF16toUTF8(nodeName).get())); #endif nsCOMPtr child; data->GetFirstChild(getter_AddRefs(child)); if (!child) return; PRUint16 childType; child->GetNodeType(&childType); nsCOMPtr sibling; child->GetNextSibling(getter_AddRefs(sibling)); if (!sibling && childType == nsIDOMNode::TEXT_NODE) { nsAutoString localName; data->GetLocalName(localName); nsAutoString value; child->GetNodeValue(value); LOG((" appending data for <%s>\n", NS_ConvertUTF16toUTF8(localName).get())); nsCString encLocalName, encValue; URLEncode(localName, encLocalName); URLEncode(value, encValue); buf.Append(encLocalName + NS_LITERAL_CSTRING("=") + encValue + separator); } else { // call AppendURLEncodedData on each child node do { AppendURLEncodedData(child, separator, buf); child->GetNextSibling(getter_AddRefs(sibling)); child.swap(sibling); } while (child); } } nsresult nsXFormsSubmissionElement::SerializeDataMultipartRelated(nsIDOMNode *data, PRUint32 format, nsIInputStream **stream, nsCString &contentType) { NS_ASSERTION(format & METHOD_POST, "unexpected submission method"); nsCAutoString boundary; MakeMultipartBoundary(boundary); nsCOMPtr multiStream = do_CreateInstance("@mozilla.org/io/multiplex-input-stream;1"); NS_ENSURE_TRUE(multiStream, NS_ERROR_UNEXPECTED); nsCAutoString type, start; MakeMultipartContentID(start); // XXX we need to extend SerializeDataXML so that it has a mode in which it // generates ContentIDs for elements of type xsd:anyURI, and returns a // list of ContentID <-> URI mappings. nsCOMPtr xml; SubmissionAttachmentArray attachments; SerializeDataXML(data, ENCODING_XML | METHOD_POST, getter_AddRefs(xml), type, &attachments); // XXX we should output a 'charset=' with the 'Content-Type' header nsCString postDataChunk; postDataChunk += NS_LITERAL_CSTRING("--") + boundary + NS_LITERAL_CSTRING("\r\nContent-Type: ") + type + NS_LITERAL_CSTRING("\r\nContent-ID: <") + start + NS_LITERAL_CSTRING(">\r\n\r\n"); nsresult rv = AppendPostDataChunk(postDataChunk, multiStream); NS_ENSURE_SUCCESS(rv, rv); multiStream->AppendStream(xml); for (PRUint32 i=0; i file; nsCOMPtr fileStream; rv = CreateFileStream(a->uri, getter_AddRefs(file), getter_AddRefs(fileStream)); NS_ENSURE_SUCCESS(rv, rv); nsCAutoString type; GetMimeTypeFromFile(file, type); postDataChunk += NS_LITERAL_CSTRING("\r\n--") + boundary + NS_LITERAL_CSTRING("\r\nContent-Type: ") + type + NS_LITERAL_CSTRING("\r\nContent-Transfer-Encoding: binary") + NS_LITERAL_CSTRING("\r\nContent-ID: <") + a->cid + NS_LITERAL_CSTRING(">\r\n\r\n"); rv = AppendPostDataChunk(postDataChunk, multiStream); NS_ENSURE_SUCCESS(rv, rv); multiStream->AppendStream(fileStream); } // final boundary postDataChunk += NS_LITERAL_CSTRING("\r\n--") + boundary + NS_LITERAL_CSTRING("--\r\n"); rv = AppendPostDataChunk(postDataChunk, multiStream); NS_ENSURE_SUCCESS(rv, rv); contentType = NS_LITERAL_CSTRING("multipart/related; boundary=") + boundary + NS_LITERAL_CSTRING("; type=\"") + type + NS_LITERAL_CSTRING("\"; start=\"<") + start + NS_LITERAL_CSTRING(">\""); NS_ADDREF(*stream = multiStream); return NS_OK; } nsresult nsXFormsSubmissionElement::SerializeDataMultipartFormData(nsIDOMNode *data, PRUint32 format, nsIInputStream **stream, nsCString &contentType) { NS_ASSERTION(format & METHOD_POST, "unexpected submission method"); // This format follows the rules for multipart/form-data MIME data streams in // [RFC 2388], with specific requirements of this serialization listed below: // o Each element node is visited in document order. // o Each element that has exactly one text node child is selected for // inclusion. // o Element nodes selected for inclusion are as encoded as // Content-Disposition: form-data MIME parts as defined in [RFC 2387], with // the name parameter being the element local name. // o Element nodes of any datatype populated by upload are serialized as the // specified content and additionally have a Content-Disposition filename // parameter, if available. // o The Content-Type must be text/plain except for xsd:base64Binary, // xsd:hexBinary, and derived types, in which case the header represents the // media type of the attachment if known, otherwise // application/octet-stream. If a character set is applicable, the // Content-Type may have a charset parameter. nsCAutoString boundary; MakeMultipartBoundary(boundary); nsCOMPtr multiStream = do_CreateInstance("@mozilla.org/io/multiplex-input-stream;1"); NS_ENSURE_TRUE(multiStream, NS_ERROR_UNEXPECTED); nsCString postDataChunk; nsresult rv = AppendMultipartFormData(data, boundary, postDataChunk, multiStream); NS_ENSURE_SUCCESS(rv, rv); postDataChunk += NS_LITERAL_CSTRING("--") + boundary + NS_LITERAL_CSTRING("--\r\n\r\n"); rv = AppendPostDataChunk(postDataChunk, multiStream); NS_ENSURE_SUCCESS(rv, rv); contentType = NS_LITERAL_CSTRING("multipart/form-data; boundary=") + boundary; NS_ADDREF(*stream = multiStream); return NS_OK; } nsresult nsXFormsSubmissionElement::AppendMultipartFormData(nsIDOMNode *data, const nsCString &boundary, nsCString &postDataChunk, nsIMultiplexInputStream *multiStream) { #ifdef DEBUG_darinf nsAutoString nodeName; data->GetNodeName(nodeName); LOG(("+++ AppendMultipartFormData: inspecting <%s>\n", NS_ConvertUTF16toUTF8(nodeName).get())); #endif nsresult rv; nsCOMPtr child; data->GetFirstChild(getter_AddRefs(child)); if (!child) return NS_OK; PRUint16 childType; child->GetNodeType(&childType); nsCOMPtr sibling; child->GetNextSibling(getter_AddRefs(sibling)); if (!sibling && childType == nsIDOMNode::TEXT_NODE) { nsAutoString localName; data->GetLocalName(localName); nsAutoString value; child->GetNodeValue(value); LOG((" appending data for <%s>\n", NS_ConvertUTF16toUTF8(localName).get())); PRUint32 encType; rv = GetElementEncodingType(data, &encType); NS_ENSURE_SUCCESS(rv, rv); NS_ConvertUTF16toUTF8 encName(localName); encName.Adopt(nsLinebreakConverter::ConvertLineBreaks(encName.get(), nsLinebreakConverter::eLinebreakAny, nsLinebreakConverter::eLinebreakNet)); postDataChunk += NS_LITERAL_CSTRING("--") + boundary + NS_LITERAL_CSTRING("\r\nContent-Disposition: form-data; name=\"") + encName + NS_LITERAL_CSTRING("\""); nsCAutoString contentType; nsCOMPtr fileStream; if (encType == ELEMENT_ENCTYPE_URI) { // 'value' contains an absolute URI reference nsCOMPtr file; rv = CreateFileStream(value, getter_AddRefs(file), getter_AddRefs(fileStream)); NS_ENSURE_SUCCESS(rv, rv); nsAutoString leafName; file->GetLeafName(leafName); postDataChunk += NS_LITERAL_CSTRING("; filename=\"") + NS_ConvertUTF16toUTF8(leafName) + NS_LITERAL_CSTRING("\""); // use mime service to get content-type GetMimeTypeFromFile(file, contentType); } else if (encType == ELEMENT_ENCTYPE_STRING) { contentType.AssignLiteral("text/plain; charset=UTF-8"); } else { contentType.AssignLiteral("application/octet-stream"); } postDataChunk += NS_LITERAL_CSTRING("\r\nContent-Type: ") + contentType + NS_LITERAL_CSTRING("\r\n"); if (encType == ELEMENT_ENCTYPE_URI) { AppendPostDataChunk(postDataChunk, multiStream); multiStream->AppendStream(fileStream); postDataChunk += NS_LITERAL_CSTRING("\r\n"); } else { // for base64binary and hexBinary types, we assume that the data is // already encoded. this assumption is based on section 8.1.6 of the // xforms spec. // XXX UTF-8 ok? NS_ConvertUTF16toUTF8 encValue(value); encValue.Adopt(nsLinebreakConverter::ConvertLineBreaks(encValue.get(), nsLinebreakConverter::eLinebreakAny, nsLinebreakConverter::eLinebreakNet)); postDataChunk += encValue + NS_LITERAL_CSTRING("\r\n"); } } else { // call AppendMultipartFormData on each child node do { rv = AppendMultipartFormData(child, boundary, postDataChunk, multiStream); if (NS_FAILED(rv)) return rv; child->GetNextSibling(getter_AddRefs(sibling)); child.swap(sibling); } while (child); } return NS_OK; } nsresult nsXFormsSubmissionElement::AppendPostDataChunk(nsCString &postDataChunk, nsIMultiplexInputStream *multiStream) { nsCOMPtr stream; NS_NewCStringInputStream(getter_AddRefs(stream), postDataChunk); NS_ENSURE_TRUE(stream, NS_ERROR_OUT_OF_MEMORY); multiStream->AppendStream(stream); postDataChunk.Truncate(); return NS_OK; } nsresult nsXFormsSubmissionElement::GetElementEncodingType(nsIDOMNode *node, PRUint32 *encType) { *encType = ELEMENT_ENCTYPE_STRING; // default nsCOMPtr element = do_QueryInterface(node); NS_ENSURE_TRUE(element, NS_ERROR_UNEXPECTED); nsAutoString type; element->GetAttributeNS(NS_LITERAL_STRING(NS_NAMESPACE_XML_SCHEMA_INSTANCE), NS_LITERAL_STRING("type"), type); if (!type.IsEmpty()) { // check for 'xsd:base64binary', 'xsd:hexBinary', or 'xsd:anyURI' // XXX need to handle derived types (fixing bug 263384 will help) // get 'xsd' namespace prefix nsCOMPtr dom3Node = do_QueryInterface(node); NS_ENSURE_TRUE(dom3Node, NS_ERROR_UNEXPECTED); nsAutoString prefix; dom3Node->LookupPrefix(NS_LITERAL_STRING(NS_NAMESPACE_XML_SCHEMA), prefix); if (prefix.IsEmpty()) { NS_WARNING("namespace prefix not found! -- assuming 'xsd'"); prefix.AssignLiteral("xsd"); // XXX HACK HACK HACK } if (type.Length() > prefix.Length() && prefix.Equals(StringHead(type, prefix.Length())) && type.CharAt(prefix.Length()) == PRUnichar(':')) { const nsSubstring &tail = Substring(type, prefix.Length() + 1); if (tail.Equals(NS_LITERAL_STRING("anyURI"))) *encType = ELEMENT_ENCTYPE_URI; else if (tail.Equals(NS_LITERAL_STRING("base64binary"))) *encType = ELEMENT_ENCTYPE_BASE64; else if (tail.Equals(NS_LITERAL_STRING("hexBinary"))) *encType = ELEMENT_ENCTYPE_HEX; } } return NS_OK; } nsresult nsXFormsSubmissionElement::CreateFileStream(const nsString &absURI, nsIFile **resultFile, nsIInputStream **resultStream) { LOG(("nsXFormsSubmissionElement::CreateFileStream [%s]\n", NS_ConvertUTF16toUTF8(absURI).get())); nsCOMPtr uri; NS_NewURI(getter_AddRefs(uri), absURI); NS_ENSURE_TRUE(uri, NS_ERROR_UNEXPECTED); // restrict to file:// -- XXX is this correct? PRBool schemeIsFile = PR_FALSE; uri->SchemeIs("file", &schemeIsFile); NS_ENSURE_TRUE(schemeIsFile, NS_ERROR_UNEXPECTED); // NOTE: QI to nsIFileURL just means that the URL corresponds to a // local file resource, which is not restricted to file:// nsCOMPtr fileURL = do_QueryInterface(uri); NS_ENSURE_TRUE(fileURL, NS_ERROR_UNEXPECTED); fileURL->GetFile(resultFile); NS_ENSURE_TRUE(*resultFile, NS_ERROR_UNEXPECTED); return NS_NewLocalFileInputStream(resultStream, *resultFile); } nsresult nsXFormsSubmissionElement::SendData(PRUint32 format, const nsCString &uriSpec, nsIInputStream *stream, const nsCString &contentType) { LOG(("+++ sending to uri=%s [stream=%p]\n", uriSpec.get(), (void*) stream)); nsCOMPtr domDoc; mElement->GetOwnerDocument(getter_AddRefs(domDoc)); nsCOMPtr doc = do_QueryInterface(domDoc); NS_ENSURE_STATE(doc); nsCOMPtr ios = do_GetIOService(); NS_ENSURE_STATE(ios); // Any parameters appended to uriSpec are already ASCII-encoded per the rules // of section 11.6. Use our standard document charset based canonicalization // for any other non-ASCII bytes. (This might be important for compatibility // with legacy CGI processors.) nsCOMPtr uri; ios->NewURI(uriSpec, doc->GetDocumentCharacterSet().get(), doc->GetDocumentURI(), getter_AddRefs(uri)); NS_ENSURE_STATE(uri); nsresult rv; nsAutoString replace; mElement->GetAttribute(NS_LITERAL_STRING("replace"), replace); if (replace.IsEmpty() || replace.EqualsLiteral("all")) { // check to see if we're allowed to load this URI nsCOMPtr secMan = do_GetService(NS_SCRIPTSECURITYMANAGER_CONTRACTID); NS_ENSURE_STATE(secMan); rv = secMan->CheckConnect(nsnull, uri, "XForms", "submission"); if (NS_FAILED(rv)) { // We need to return success here so that JS will get a proper // exception thrown later. Native calls should always result in // CheckConnect() succeeding, but in case JS calls C++ which calls // this code the exception might be lost. return NS_OK; } } // wrap the entire upload stream in a buffered input stream, so that // it can be read in large chunks. // XXX necko should probably do this (or something like this) for us. nsCOMPtr bufferedStream; if (stream) { NS_NewBufferedInputStream(getter_AddRefs(bufferedStream), stream, 4096); NS_ENSURE_TRUE(bufferedStream, NS_ERROR_UNEXPECTED); } nsCOMPtr channel; ios->NewChannelFromURI(uri, getter_AddRefs(channel)); NS_ENSURE_STATE(channel); if (bufferedStream) { nsCOMPtr uploadChannel = do_QueryInterface(channel); NS_ENSURE_STATE(uploadChannel); // this in effect sets the request method of the channel to 'PUT' rv = uploadChannel->SetUploadStream(bufferedStream, contentType, -1); NS_ENSURE_SUCCESS(rv, rv); } if (format & METHOD_POST) { nsCOMPtr httpChannel = do_QueryInterface(channel); NS_ENSURE_STATE(httpChannel); rv = httpChannel->SetRequestMethod(NS_LITERAL_CSTRING("POST")); NS_ENSURE_SUCCESS(rv, rv); } // set loadGroup and notificationCallbacks nsCOMPtr loadGroup = doc->GetDocumentLoadGroup(); channel->SetLoadGroup(loadGroup); // the container is the docshell, and we use it as our provider of // notification callbacks. nsCOMPtr container = doc->GetContainer(); nsCOMPtr callbacks = do_QueryInterface(container); // XXX this means that we allow redirects to any URI, and that is a // security hole that will need to be plugged. of course, we only // need to worry about this problem for replace='instance' channel->SetNotificationCallbacks(callbacks); // create a pipe in which to store the response (yeah, this kind of // sucks since we'll use a lot of memory if the response is large). // // pipe uses non-blocking i/o since we are just using it for temporary // storage. // // pipe's maximum size is unlimited (gasp!) nsCOMPtr pipeOut; rv = NS_NewPipe(getter_AddRefs(mPipeIn), getter_AddRefs(pipeOut), 0, PR_UINT32_MAX, PR_TRUE, PR_TRUE); NS_ENSURE_SUCCESS(rv, rv); // use a simple stream listener to tee our data into the pipe, and // notify us when the channel starts and stops. nsCOMPtr listener; rv = NS_NewSimpleStreamListener(getter_AddRefs(listener), pipeOut, this); NS_ENSURE_SUCCESS(rv, rv); rv = channel->AsyncOpen(listener, nsnull); NS_ENSURE_SUCCESS(rv, rv); return rv; } // factory constructor nsresult NS_NewXFormsSubmissionElement(nsIXTFElement **aResult) { *aResult = new nsXFormsSubmissionElement(); if (!*aResult) return NS_ERROR_OUT_OF_MEMORY; NS_ADDREF(*aResult); return NS_OK; }