1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165
|
# Copyright (C) 2008-2010, Sebastian Riedel.
package Mojo::Content::Single;
use strict;
use warnings;
use base 'Mojo::Content';
use Mojo::Asset::File;
use Mojo::Asset::Memory;
use Mojo::Content::MultiPart;
__PACKAGE__->attr(asset => sub { Mojo::Asset::Memory->new });
sub body_contains {
my ($self, $chunk) = @_;
# Found
return 1 if $self->asset->contains($chunk) >= 0;
# Not found
return 0;
}
sub body_size { shift->asset->size }
sub get_body_chunk {
my ($self, $offset) = @_;
# Body generator
return $self->generate_body_chunk($offset) if $self->body_cb;
# Normal content
return $self->asset->get_chunk($offset);
}
sub parse {
my $self = shift;
# Parse headers and filter body
$self->SUPER::parse(@_);
# Still parsing headers or using a custom body parser
return $self if $self->is_state('headers') || $self->body_cb;
# Headers
my $headers = $self->headers;
# WebSocket handshakes have a static Content-Length
my $length =
$headers->sec_websocket_key1 ? 8
: $headers->sec_websocket_location ? 16
: undef;
# Don't waste memory
if ($self->asset->isa('Mojo::Asset::Memory')) {
$length ||= $self->headers->content_length;
# Upgrade to file storage
$self->asset(Mojo::Asset::File->new)
if !$length || $length > ($ENV{MOJO_MAX_MEMORY_SIZE} || 24576);
}
# Content needs to be upgraded to multipart
if ($self->is_multipart) {
# Shortcut
return $self if $self->isa('Mojo::Content::MultiPart');
# Need to upgrade
return Mojo::Content::MultiPart->new($self)->parse;
}
# Chunked body or relaxed content
if ($self->is_chunked || $self->relaxed) {
$self->asset->add_chunk($self->buffer->empty);
}
# Normal body
else {
# Slurp
$length ||= $self->headers->content_length || 0;
my $asset = $self->asset;
my $need = $length - $asset->size;
$asset->add_chunk($self->buffer->remove($need)) if $need > 0;
# Done
$self->done if $length <= $self->raw_body_size;
}
# With leftovers, maybe pipelined
if ($self->is_done) {
$self->state('done_with_leftovers') if $self->has_leftovers;
}
return $self;
}
1;
__END__
=head1 NAME
Mojo::Content::Single - HTTP 1.1 Content Container
=head1 SYNOPSIS
use Mojo::Content::Single;
my $content = Mojo::Content::Single->new;
$content->parse("Content-Length: 12\r\n\r\nHello World!");
=head1 DESCRIPTION
L<Mojo::Content::Single> is a container for HTTP 1.1 content as described in
RFC 2616.
=head1 ATTRIBUTES
L<Mojo::Content::Single> inherits all attributes from L<Mojo::Content> and
implements the following new ones.
=head2 C<asset>
my $asset = $content->asset;
$content = $content->asset(Mojo::Asset::Memory->new);
The actual content.
=head1 METHODS
L<Mojo::Content::Single> inherits all methods from L<Mojo::Content> and
implements the following new ones.
=head2 C<body_contains>
my $found = $content->body_contains('1234567');
Check if content contains a specific string.
=head2 C<body_size>
my $size = $content->body_size;
Content size in bytes.
=head2 C<get_body_chunk>
my $chunk = $content->get_body_chunk(0);
Get a chunk of content starting from a specfic position.
=head2 C<parse>
$content = $content->parse("Content-Length: 12\r\n\r\nHello World!");
Parse content.
=head1 SEE ALSO
L<Mojolicious>, L<Mojolicious::Guides>, L<http://mojolicious.org>.
=cut
|