/* * Copyright (C) 2004 Apple Computer, Inc. All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #import "KWQLogging.h" #import "KWQExceptions.h" #import "KWQKJobClasses.h" #import "KWQLoader.h" #import "KWQResourceLoader.h" #import "KWQString.h" #import "KWQFoundationExtras.h" #import "formdata.h" using khtml::FormData; namespace KIO { // The allocations and releases in TransferJobPrivate are // definitely Cocoa-exception-free (either simple Foundation // classes or our own KWQResourceLoader which avoides doing work // in dealloc class TransferJobPrivate { public: TransferJobPrivate(const KURL& kurl) : status(0) , metaData(KWQRetainNSRelease([[NSMutableDictionary alloc] initWithCapacity:17])) , URL(kurl) , loader(nil) , method("GET") , response(nil) , assembledResponseHeaders(true) , retrievedCharset(true) { } TransferJobPrivate(const KURL& kurl, const FormData &_postData) : status(0) , metaData(KWQRetainNSRelease([[NSMutableDictionary alloc] initWithCapacity:17])) , URL(kurl) , loader(nil) , method("POST") , postData(_postData) , response(nil) , assembledResponseHeaders(true) , retrievedCharset(true) { } ~TransferJobPrivate() { KWQRelease(response); KWQRelease(metaData); KWQRelease(loader); } int status; NSMutableDictionary *metaData; KURL URL; KWQResourceLoader *loader; QString method; FormData postData; NSURLResponse *response; bool assembledResponseHeaders; bool retrievedCharset; QString responseHeaders; }; TransferJob::TransferJob(const KURL &url, bool reload, bool deliverAllData) : d(new TransferJobPrivate(url)), m_deliverAllData(deliverAllData), m_data(this, SIGNAL(data(KIO::Job*, const char*, int))), m_redirection(this, SIGNAL(redirection(KIO::Job*, const KURL&))), m_result(this, deliverAllData ? SIGNAL(result(KIO::Job*, NSData *)) : SIGNAL(result(KIO::Job*))), m_receivedResponse(this, SIGNAL(receivedResponse(KIO::Job*, NSURLResponse *))) { } TransferJob::TransferJob(const KURL &url, const FormData &postData, bool deliverAllData) : d(new TransferJobPrivate(url, postData)), m_deliverAllData(deliverAllData), m_data(this, SIGNAL(data(KIO::Job*, const char*, int))), m_redirection(this, SIGNAL(redirection(KIO::Job*, const KURL&))), m_result(this, deliverAllData ? SIGNAL(result(KIO::Job*, NSData *)) : SIGNAL(result(KIO::Job*))), m_receivedResponse(this, SIGNAL(receivedResponse(KIO::Job*, NSURLResponse *))) { } TransferJob::~TransferJob() { // This will cancel the handle, and who knows what that could do KWQ_BLOCK_EXCEPTIONS; [d->loader jobWillBeDeallocated]; KWQ_UNBLOCK_EXCEPTIONS; delete d; } bool TransferJob::isErrorPage() const { return d->status != 0; } int TransferJob::error() const { return d->status; } void TransferJob::setError(int e) { d->status = e; } QString TransferJob::errorText() const { LOG(NotYetImplemented, "not yet implemented"); return QString::null; } void TransferJob::assembleResponseHeaders() const { if (!d->assembledResponseHeaders) { if ([d->response isKindOfClass:[NSHTTPURLResponse class]]) { NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *)d->response; NSDictionary *headers = [httpResponse allHeaderFields]; d->responseHeaders = QString::fromNSString(KWQHeaderStringFromDictionary(headers, [httpResponse statusCode])); } d->assembledResponseHeaders = true; } } void TransferJob::retrieveCharset() const { if (!d->retrievedCharset) { NSString *charset = [d->response textEncodingName]; if (charset) { [d->metaData setObject:charset forKey:@"charset"]; } d->retrievedCharset = true; } } QString TransferJob::queryMetaData(const QString &key) const { if (key == "HTTP-Headers") { assembleResponseHeaders(); return d->responseHeaders; } if (key == "charset") { // this will put it in the regular metadata dictionary retrieveCharset(); } NSString *value = [d->metaData objectForKey:key.getNSString()]; return value ? QString::fromNSString(value) : QString::null; } void TransferJob::addMetaData(const QString &key, const QString &value) { [d->metaData setObject:value.getNSString() forKey:key.getNSString()]; } void TransferJob::addMetaData(const QMap<QString, QString> &keysAndValues) { QMapConstIterator<QString, QString> it = keysAndValues.begin(); QMapConstIterator<QString, QString> end = keysAndValues.end(); while (it != end) { [d->metaData setObject:it.data().getNSString() forKey:it.key().getNSString()]; ++it; } } void TransferJob::kill() { delete this; } void TransferJob::setLoader(KWQResourceLoader *loader) { KWQRetain(loader); KWQRelease(d->loader); d->loader = loader; } KURL TransferJob::url() const { return d->URL; } FormData TransferJob::postData() const { return d->postData; } QString TransferJob::method() const { return d->method; } void TransferJob::emitData(const char *data, int size) { m_data.call(this, data, size); } void TransferJob::emitRedirection(const KURL &url) { m_redirection.call(this, url); } void TransferJob::emitResult(NSData *allData) { m_deliverAllData ? m_result.call(this, allData) : m_result.call(this); } void TransferJob::emitReceivedResponse(NSURLResponse *response) { d->assembledResponseHeaders = false; d->retrievedCharset = false; d->response = response; KWQRetain(d->response); m_receivedResponse.call(this, response); } } // namespace KIO