X-Git-Url: http://git.rot13.org/?p=perl-cwmp.git;a=blobdiff_plain;f=lib%2FCWMP%2FSession.pm;h=4ce8a07315c1061d9571b93fa9b9ad3e96d165b3;hp=ff6d95d4b525e2332096fa6c7fac88f3de9edaf4;hb=c132d8c37368aad6f9ff64ec873f3648407044df;hpb=6ae55ee7caaa8c33a10a983dcc3b50928eb52914 diff --git a/lib/CWMP/Session.pm b/lib/CWMP/Session.pm index ff6d95d..4ce8a07 100644 --- a/lib/CWMP/Session.pm +++ b/lib/CWMP/Session.pm @@ -7,22 +7,23 @@ use warnings; use base qw/Class::Accessor/; __PACKAGE__->mk_accessors( qw/ debug -store +create_dump +session sock state store / ); -use HTTP::Daemon; use Data::Dump qw/dump/; -use Carp qw/confess cluck croak/; -use File::Slurp; +use Carp qw/carp confess cluck croak/; -use CWMP::Request; +use CWMP::Parser; use CWMP::Methods; use CWMP::Store; +#use Devel::LeakTrace::Fast; + =head1 NAME CWMP::Session - implement logic of CWMP protocol @@ -33,8 +34,9 @@ CWMP::Session - implement logic of CWMP protocol my $server = CWMP::Session->new({ sock => $io_socket_object, - store => 'state.db', + store => { ... }, debug => 1, + create_dump => 1, }); =cut @@ -43,12 +45,10 @@ sub new { my $class = shift; my $self = $class->SUPER::new( @_ ); - confess "need sock" unless $self->sock; + confess "need store" unless $self->store; $self->debug( 0 ) unless $self->debug; - warn "created ", __PACKAGE__, "(", dump( @_ ), ") for ", $self->sock->peerhost, "\n" if $self->debug; - my $store_obj = CWMP::Store->new({ debug => $self->debug, %{ $self->store }, @@ -59,9 +59,42 @@ sub new { # FIXME looks ugly. Should we have separate accessor for this? $self->store( $store_obj ); + $self->create_dump( 1 ) if $self->debug > 2; + return $self; } +my $vendor_data = { + 'InternetGatewayDevice.ManagementServer.PeriodicInformEnable' => 1, + 'InternetGatewayDevice.ManagementServer.PeriodicInformInterval' => 17, + 'InternetGatewayDevice.DeviceInfo.ProvisioningCode' => 'test provision', +}; + +our $set_tried; + +sub vendor_hook { + my ( $self, $uid, $stored, $queue ) = @_; + warn "# vendor_hook $uid ",dump($stored) if $self->debug > 2; + + my @refresh; + + foreach my $n ( keys %$vendor_data ) { + if ( defined $stored->{$n} && $vendor_data->{$n} ne $stored->{$n} ) { + next if $set_tried->{$uid}->{$n}++; + push @refresh, $n; + $queue->enqueue( 'SetParameterValues', { $n => $vendor_data->{$n} } ); + } + } + + if ( @refresh ) { + $queue->enqueue( 'GetParameterValues', [ @refresh ] ); + warn "vendor_hook $uid SetParameterValues ", dump( @refresh ); + return $self->dispatch( 'GetParameterValues', [ @refresh ] ); + } + + return; +} + =head2 process_request One request from client/response from server cycle. Call multiple times to @@ -74,61 +107,34 @@ of requests in single session. =cut -my $dump_nr = 0; sub process_request { - my $self = shift; - - my $sock = $self->sock || die "no sock?"; - -# die "not IO::Socket::INET but ", ref( $sock ) unless ( ref($sock) eq 'Net::Server::Proto::TCP' ); - - if ( ! $sock->connected ) { - warn "SOCKET NOT CONNECTED\n"; - return 0; - } - - bless $sock, 'HTTP::Daemon::ClientConn'; - - # why do I have to do this? - # solution from http://use.perl.org/~Matts/journal/12896 - ${*$sock}{'httpd_daemon'} = HTTP::Daemon->new; - - my $r = $sock->get_request || confess "can't get_request"; - - my $xml = $r->content; + my ( $self, $ip, $xml ) = @_; my $size = length( $xml ); - warn "<<<< ", $sock->peerhost, " [" . localtime() . "] ", $r->method, " ", $r->uri, " $size bytes\n"; - - $dump_nr++; - my $file = sprintf("dump/%04d-%s.request", $dump_nr, $sock->peerhost); - - if ( $self->debug > 2 ) { - write_file( $file, $r->as_string ); - warn "### request dumped to file: $file\n"; - } - my $state; if ( $size > 0 ) { - die "no SOAPAction header in ",dump($xml) unless defined ( $r->header('SOAPAction') ); - warn "## request payload: ",length($xml)," bytes\n$xml\n" if $self->debug; - $state = CWMP::Request->parse( $xml ); + $state = CWMP::Parser->parse( $xml ); - if ( defined( $state->{_dispatch} ) && $self->debug > 2 ) { - my $type = sprintf("dump/%04d-%s-%s", $dump_nr, $sock->peerhost, $state->{_dispatch}); - symlink $file, $type || warn "can't symlink $file -> $type: $!"; - } + warn "## acquired state = ", dump( $state ), "\n" if $self->debug; - warn "## acquired state = ", dump( $state ), "\n"; + if ( ! defined( $state->{DeviceId} ) ) { + if ( $self->state ) { + warn "## state without DeviceId, using old one...\n"; + $state->{DeviceId} = $self->state->{DeviceId}; + } else { + warn "WARNING: state without DeviceId, and I don't have old one!\n"; + warn "## state = ",dump( $state ); + } + } $self->state( $state ); - $self->store->update_state( ID => $state->{ID}, $state ); + $self->store->update_state( $state ); } else { @@ -139,47 +145,95 @@ sub process_request { #warn "last request state = ", dump( $state ), "\n" if $self->debug > 1; } + my $uid = $self->store->state_to_uid( $state ); - $sock->send(join("\r\n", - 'HTTP/1.1 200 OK', - 'Content-Type: text/xml; charset="utf-8"', - 'Server: AcmeCWMP/42', - 'SOAPServer: AcmeCWMP/42' - )."\r\n"); - - $sock->send( "Set-Cookie: ID=" . $state->{ID} . "; path=/\r\n" ) if ( $state->{ID} ); - - my $uid = $self->store->ID_to_uid( $state->{ID}, $state ); + my $to_uid = join(" ", grep { defined($_) } "to $uid", + # board + $state->{Parameter}->{'InternetGatewayDevice.DeviceInfo.HardwareVersion'}, + # version + $state->{Parameter}->{'InternetGatewayDevice.DeviceInfo.SoftwareVersion'}, + # summary +# $state->{Parameter}->{'InternetGatewayDevice.DeviceSummary'}, + ) . "\n"; my $queue = CWMP::Queue->new({ id => $uid, debug => $self->debug, }); - my $job; $xml = ''; if ( my $dispatch = $state->{_dispatch} ) { $xml = $self->dispatch( $dispatch ); - } elsif ( $job = $queue->dequeue ) { + } elsif ( my $job = $queue->dequeue ) { $xml = $self->dispatch( $job->dispatch ); - } elsif ( $size == 0 ) { - warn ">>> no more queued commands, closing connection to $uid\n"; - return 0; + $job->finish; } else { - warn ">>> empty response to $uid\n"; - $state->{NoMoreRequests} = 1; - $xml = $self->dispatch( 'xml', sub {} ); + my $stored = $self->store->get_state( $uid ); + if ( ! defined $stored->{ParameterInfo} ) { + $xml = $self->dispatch( 'GetParameterNames', [ 'InternetGatewayDevice.', 1 ] ); + } else { + my @params = + grep { m/\.$/ && ! m/\.\d+\.$/ } + keys %{ $stored->{ParameterInfo} } + ; + if ( @params ) { + warn "# GetParameterNames ", dump( @params ); + my $first = shift @params; + delete $stored->{ParameterInfo}->{$first}; + $xml = $self->dispatch( 'GetParameterNames', [ $first, 1 ] ); + foreach ( @params ) { + $queue->enqueue( 'GetParameterNames', [ $_, 1 ] ); + delete $stored->{ParameterInfo}->{ $_ }; + } + $self->store->set_state( $uid, $stored ); + } else { + + my @params = sort + grep { ! exists $stored->{Parameter}->{$_} } + grep { ! m/\.$/ && ! m/NumberOfEntries/ } + keys %{ $stored->{ParameterInfo} } + ; + if ( @params ) { + warn "# GetParameterValues ", dump( @params ); + my $first = shift @params; + $xml = $self->dispatch( 'GetParameterValues', [ $first ] ); + while ( @params ) { + my @chunk = splice @params, 0, 16; # FIXME 16 seems to be max + $queue->enqueue( 'GetParameterValues', [ @chunk ] ); + } + + } elsif ( $xml = $self->vendor_hook( $uid, $stored, $queue ) ) { + + warn "vendor_hook triggered\n"; + + } else { + + warn ">>> empty response $to_uid"; + $state->{NoMoreRequests} = 1; + $xml = ''; + + } + } + } } - $sock->send( "Content-Length: " . length( $xml ) . "\r\n\r\n" ); - $sock->send( $xml ) or die "can't send response"; + my $status = length($xml) ? 200 : 204; - warn ">>>> " . $sock->peerhost . " [" . localtime() . "] sent ", length( $xml )," bytes to $uid\n"; + my $out = join("\r\n", + "HTTP/1.1 $status OK", + 'Content-Type: text/xml; charset="utf-8"', + 'Server: Perl-CWMP/42', + 'SOAPServer: Perl-CWMP/42' + ) . "\r\n"; + + $out .= "Set-Cookie: ID=" . $state->{ID} . "; path=/\r\n" if $state->{ID}; + + $out .= "Content-Length: " . length( $xml ) . "\r\n\r\n"; + $out .= $xml if length($xml); - $job->finish if $job; warn "### request over for $uid\n" if $self->debug; - return 1; # next request + return $out; # next request }; =head2 dispatch @@ -193,8 +247,6 @@ If debugging level of 3 or more, it will create dumps of responses named C<< dum sub dispatch { my $self = shift; -warn "##!!! dispatch(",dump( @_ ),")\n"; - my $dispatch = shift || die "no dispatch?"; my $args = shift; @@ -204,11 +256,6 @@ warn "##!!! dispatch(",dump( @_ ),")\n"; warn ">>> dispatching to $dispatch with args ",dump( $args ),"\n"; my $xml = $response->$dispatch( $self->state, $args ); warn "## response payload: ",length($xml)," bytes\n$xml\n" if $self->debug; - if ( $self->debug > 2 ) { - my $file = sprintf("dump/%04d-%s.response", $dump_nr++, $self->sock->peerhost); - write_file( $file, $xml ); - warn "### response dump: $file\n"; - } return $xml; } else { confess "can't dispatch to $dispatch";