#!/usr/bin/tclsh
-# $Id: robot.tcl,v 1.5 1999/12/27 11:49:31 adam Exp $
+# $Id: robot.tcl,v 1.25 2001/11/07 11:50:07 adam Exp $
#
-proc RobotFileNext {area} {
+proc RobotFileNext1 {area lead} {
+ # puts "RobotFileNext1 area=$area lead=$lead"
if {[catch {set ns [glob ${area}/*]}]} {
return {}
}
- set off [string first / $area]
- incr off
-
foreach n $ns {
if {[file isfile $n]} {
- if {[string first :.html $n] > 0} {
- return http://[string range $area/ $off end]
- }
- return http://[string range $n $off end]
+ set off [string last / $n]
+ incr off 2
+ return $lead/[string range $n $off end]
}
- if {[file isdirectory $n]} {
- set sb [RobotFileNext $n]
+ }
+ foreach n $ns {
+ if {[file isdirectory $n]} {
+ set off [string last / $n]
+ incr off 2
+ set sb [RobotFileNext1 $n $lead/[string range $n $off end]]
if {[string length $sb]} {
return $sb
}
return {}
}
+proc RobotWriteRecord {outf fromurl distance} {
+ puts $outf "<zmbot>"
+ puts $outf "<distance>"
+ puts $outf $distance
+ puts $outf "</distance>"
+ puts $outf "<fromurl>"
+ puts $outf $fromurl
+ puts $outf "</fromurl>"
+ puts $outf "</zmbot>"
+}
+
+proc RobotReadRecord {inf fromurlx distancex} {
+ upvar $fromurlx fromurl
+ upvar $distancex distance
+ gets $inf
+ gets $inf
+ set distance [string trim [gets $inf]]
+ # puts "got distance = $distance"
+ gets $inf
+ gets $inf
+ set fromurl [string trim [gets $inf]]
+}
+
+proc RobotFileNext {area} {
+ global robotSeq global idletime ns
+
+ # puts "RobotFileNext robotSeq=$robotSeq"
+ if {$robotSeq < 0} {
+ return {}
+ }
+ if {$robotSeq == 0} {
+ if {[catch {set ns [glob ${area}/*]}]} {
+ return {}
+ }
+ }
+ set off [string length $area]
+ incr off
+ set n [lindex $ns $robotSeq]
+ if {![string length $n]} {
+ set robotSeq -1
+ flush stdout
+ puts "Round robin"
+ return wait
+ }
+ incr robotSeq
+ if {[file isfile $n/frobots.txt]} {
+ puts "ok returning http://[string range $n $off end]/robots.txt"
+ return http://[string range $n $off end]/robots.txt
+ } elseif {[file isdirectory $n]} {
+ set sb [RobotFileNext1 $n http://[string range $n $off end]]
+ if {[string length $sb]} {
+ return $sb
+ }
+ }
+ puts "no more work at end of RobotFileNext n=$n"
+ puts "ns=$ns"
+ return {}
+}
+
+
proc RobotFileExist {area host path} {
- set comp [split $area/$host$path /]
- set l [llength $comp]
+ # puts "RobotFileExist begin area=$area host=$host path=$path"
+ set lpath [split $path /]
+ set l [llength $lpath]
incr l -1
- if {![string length [lindex $comp $l]]} {
- set comp [split $area/$host$path:.html /]
- } elseif {[file exists [join $comp /]]} {
- return 1
- } else {
- set comp [split $area/$host$path/:.html /]
- }
- return [file exists [join $comp /]]
+ set t [lindex $lpath $l]
+ incr l -1
+ set npath $area/$host[join [lrange $lpath 0 $l] /d]/f$t
+ # puts "RobotFileExist end npath=$npath"
+ return [file exists $npath]
}
proc RobotFileUnlink {area host path} {
- set comp [split $area/$host$path /]
+ # puts "RobotFileUnlink begin"
+ # puts "area=$area host=$host path=$path"
+ set lpath [split $path /]
+ set l [llength $lpath]
+ incr l -1
+ set t [lindex $lpath $l]
+ incr l -1
+ set npath $area/$host[join [lrange $lpath 0 $l] /d]/f$t
+ # puts "npath=$npath"
+ set comp [split $npath /]
set l [llength $comp]
incr l -1
- if {![string length [lindex $comp $l]]} {
- set comp [split $area/$host$path:.html /]
- }
if {[catch {exec rm [join $comp /]}]} return
incr l -1
for {set i $l} {$i > 0} {incr i -1} {
if {![catch {glob $path/*}]} return
exec rmdir ./$path
}
+ # puts "RobotFileUnlink end"
}
-proc RobotFileOpen {area host path} {
+proc RobotFileClose {out} {
+ if [string compare $out stdout] {
+ close $out
+ }
+}
+
+proc RobotFileOpen {area host path {mode w}} {
set orgPwd [pwd]
global workdir
- #puts "RobotFileOpen orgPwd=$orgPwd area=$area host=$host path=$path"
+ if {![info exists workdir]} {
+ return stdout
+ }
+ #puts "RobotFileOpen orgPwd=$orgPwd area=$area host=$host path=$path mode=$mode"
if {[string compare $orgPwd $workdir]} {
+ puts "ooops. RobotFileOpen failed"
puts "workdir = $workdir"
puts "pwd = $orgPwd"
exit 1
set len [llength $comp]
incr len -1
for {set i 0} {$i < $len} {incr i} {
- set d [lindex $comp $i]
+ if {$i > 1} {
+ set d "d[lindex $comp $i]"
+ } else {
+ set d [lindex $comp $i]
+ }
if {[catch {cd ./$d}]} {
exec mkdir $d
cd ./$d
+ if {![string compare $area unvisited] && $i == 1 && $mode == "w"} {
+ set out [open frobots.txt w]
+ puts "creating robots.txt in $d"
+ close $out
+ }
}
}
set d [lindex $comp $len]
if {[string length $d]} {
- set out [open $d w]
+ if {[file isdirectory $d]} {
+ set out [open $d/f $mode]
+ } else {
+ set out [open f$d $mode]
+ }
} else {
- set out [open :.html w]
+ set out [open f $mode]
}
cd $orgPwd
- #puts "RobotFileStop"
return $out
}
-proc RobotRestart {} {
+proc RobotRR {} {
+ global robotSeq robotsRunning
+
+ incr robotsRunning -1
+ while {$robotsRunning} {
+ vwait robotsRunning
+ }
+ set robotSeq 0
+ RobotStart
+}
+
+proc RobotRestart {url sock} {
+ global URL robotsRunning
+
+ close $sock
+ after cancel $URL($sock,cancel)
+
+ foreach v [array names URL $url,*] {
+ unset URL($v)
+ }
+
+ incr robotsRunning -1
+ RobotStart
+}
+
+proc RobotStart {} {
global URL
-
- while {1} {
+ global robotsRunning robotsMax idletime
+
+ # puts "RobotStart"
+ while {1} {
set url [RobotFileNext unvisited]
if {![string length $url]} {
- puts "No more unvisited"
- break
+ return
+ }
+ incr robotsRunning
+ if {[string compare $url wait] == 0} {
+ after $idletime RobotRR
+ return
}
set r [RobotGetUrl $url {}]
if {!$r} {
- puts "RobotGetUrl returned 0 on url=$url"
- return
+ if {$robotsRunning >= $robotsMax} return
} else {
- RobotFileUnlink unvisited $URL($url,host) $URL($url,path)
- }
+ incr robotsRunning -1
+ if {![RobotFileExist bad $URL($url,hostport) $URL($url,path)]} {
+ set outf [RobotFileOpen bad $URL($url,hostport) $URL($url,path)]
+ RobotFileClose $outf
+ }
+ RobotFileUnlink unvisited $URL($url,hostport) $URL($url,path)
+ }
}
- exit 0
}
-proc headSave {url out title} {
+proc headSave {url out} {
global URL
- puts $out {<meta>}
- puts $out "<title>$title</title>"
if {[info exists URL($url,head,last-modified)]} {
puts $out "<lastmodified>$URL($url,head,last-modified)</lastmodified>"
}
}
proc RobotHref {url hrefx hostx pathx} {
- global URL domains
+ global URL domains debuglevel
upvar $hrefx href
upvar $hostx host
upvar $pathx path
- # puts "Ref url = $url href=$href"
+ if {$debuglevel > 1} {
+ puts "Ref input url = $url href=$href"
+ }
+
+ if {[string first { } $href] >= 0} {
+ return 0
+ }
+ if {[string length $href] > 256} {
+ return 0
+ }
+ if {[string first {?} $href] >= 0} {
+ return 0
+ }
+ if {[string first {?} $url] >= 0 && [string first {?} $href] >= 0} {
+ return 0
+ }
# get method (if any)
if {![regexp {^([^/:]+):(.*)} $href x method hpath]} {
set hpath $href
}
}
# get host (if any)
- if {![regexp {^//([^/]+)(.*)} $hpath x host epath]} {
- set epath $hpath
- set host $URL($url,host)
- } else {
- if {![string length $epath]} {
- set epath /
+ if {[regexp {^//([^/]+)([^\#]*)} $hpath x host surl]} {
+ if {![string length $surl]} {
+ set surl /
}
- set ok 0
- foreach domain $domains {
- if {[string match $domain $host]} {
- set ok 1
- break
+ if {[info exist domains]} {
+ set ok 0
+ foreach domain $domains {
+ if {[string match $domain $host]} {
+ set ok 1
+ break
+ }
}
- }
- if {!$ok} {
- return 0
- }
+ if {!$ok} {
+ return 0
+ }
+ }
+ } else {
+ regexp {^([^\#]*)} $hpath x surl
+ set host $URL($url,hostport)
}
- if {[regexp {^(\#|\?)} $epath]} {
- # within page
+ if {![string length $surl]} {
return 0
- } elseif {![regexp {^([/][^\#?]*)} $epath x path]} {
+ }
+ if {[string first / $surl]} {
# relative path
- set ext [file extension $URL($url,path)]
- if {[string compare $ext {}]} {
- set dpart [file dirname $URL($url,path)]
+ regexp {^([^\#?]*)} $URL($url,path) x dpart
+ set l [string last / $dpart]
+ if {[expr $l >= 0]} {
+ set surl [string range $dpart 0 $l]$surl
} else {
- set dpart $URL($url,path)
+ set surl $dpart/$surl
}
- regexp {^([^\#?]+)} $epath x path
- set path [string trimright $dpart /]/$path
- }
- set c [split $path /]
- set i [llength $c]
- incr i -1
- set path [lindex $c $i]
- incr i -1
- while {$i >= 0} {
- switch -- [lindex $c $i] {
+ }
+ set surllist [split $surl /]
+ catch {unset path}
+ set pathl 0
+ foreach c $surllist {
+ switch -- $c {
.. {
- incr i -2
- }
- . {
- incr i -1
+ if {$pathl > 0} {
+ incr pathl -2
+ set path [lrange $path 0 $pathl]
+ incr pathl
+ }
}
- default {
- set path [lindex $c $i]/$path
- incr i -1
+ . {
+
+ }
+ default {
+ incr pathl
+ lappend path $c
}
}
- }
+ }
+ if {$pathl} {
+ set path [join $path /]
+ } else {
+ set path ""
+ }
+ regsub -all {~} $path {%7E} path
set href "$method://$host$path"
- # puts "Ref href = $href"
- return 1
-}
-
-proc Robot401 {url} {
- global URL
- puts "Bad link $url"
- RobotFileUnlink unvisited $URL($url,host) $URL($url,path)
- if {![RobotFileExist forbidden $URL($url,host) $URL($url,path)]} {
- set outf [RobotFileOpen forbidden $URL($url,host) $URL($url,path)]
- close $outf
+ if {$debuglevel > 1} {
+ puts "Ref result = $href"
}
+ return [checkrule url $href]
}
-proc Robot404 {url} {
+proc RobotError {url code} {
global URL
- puts "Bad link $url"
- RobotFileUnlink unvisited $URL($url,host) $URL($url,path)
- if {![RobotFileExist bad $URL($url,host) $URL($url,path)]} {
- set outf [RobotFileOpen bad $URL($url,host) $URL($url,path)]
- close $outf
+ puts "Bad URL $url (code $code)"
+ set fromurl {}
+ set distance -1
+ if {[RobotFileExist unvisited $URL($url,hostport) $URL($url,path)]} {
+ set inf [RobotFileOpen unvisited $URL($url,hostport) $URL($url,path) r]
+ RobotReadRecord $inf fromurl distance
+ RobotFileClose $inf
+ }
+ RobotFileUnlink unvisited $URL($url,hostport) $URL($url,path)
+ if {![RobotFileExist bad $URL($url,hostport) $URL($url,path)]} {
+ set outf [RobotFileOpen bad $URL($url,hostport) $URL($url,path)]
+ RobotWriteRecord $outf $fromurl $distance
+ RobotFileClose $outf
}
}
-proc Robot301 {url tourl} {
+proc RobotRedirect {url tourl code} {
global URL
puts "Redirecting from $url to $tourl"
- RobotFileUnlink unvisited $URL($url,host) $URL($url,path)
+
+ set distance {}
+ set fromurl {}
+ if {[RobotFileExist unvisited $URL($url,hostport) $URL($url,path)]} {
+ set inf [RobotFileOpen unvisited $URL($url,hostport) $URL($url,path) r]
+ RobotReadRecord $inf fromurl distance
+ RobotFileClose $inf
+ }
+ if {![RobotFileExist bad $URL($url,hostport) $URL($url,path)]} {
+ set outf [RobotFileOpen bad $URL($url,hostport) $URL($url,path)]
+ RobotWriteRecord $outf $fromurl $distance
+ RobotFileClose $outf
+ }
if {[RobotHref $url tourl host path]} {
- if {![RobotFileExist unvisited $host $path]} {
- set outf [RobotFileOpen unvisited $host $path]
- close $outf
+ if {![RobotFileExist visited $host $path]} {
+ if {![RobotFileExist unvisited $host $path]} {
+ set outf [RobotFileOpen unvisited $host $path]
+ RobotWriteRecord $outf $fromurl $distance
+ RobotFileClose $outf
+ }
+ } else {
+ set olddistance {}
+ set inf [RobotFileOpen visited $host $path r]
+ RobotReadRecord $inf oldurl olddistance
+ RobotFileClose $inf
+ if {[string length $olddistance] == 0} {
+ set olddistance 1000
+ }
+ if {[string length $distance] == 0} {
+ set distance 1000
+ }
+ puts "distance=$distance olddistance=$olddistance"
+ if {[expr $distance < $olddistance]} {
+ set outf [RobotFileOpen unvisited $host $path]
+ RobotWriteRecord $outf $tourl $distance
+ RobotFileClose $outf
+ }
}
}
+ if {[catch {RobotFileUnlink unvisited $URL($url,hostport) $URL($url,path)}]} {
+ puts "unlink failed"
+ exit 1
+ }
}
-proc Robot200 {url} {
- global URL domains
-
- # puts "Parsing $url"
- set out [RobotFileOpen visited $URL($url,host) $URL($url,path)]
- set ti 0
- if {[info exists URL($url,buf)]} {
- set htmlContent $URL($url,buf)
-
- htmlSwitch $htmlContent \
+proc RobotTextHtml {url out} {
+ global URL maxdistance
+
+ set distance 0
+ if {$maxdistance < 1000 && [info exists URL($url,dist)]} {
+ set distance [expr $URL($url,dist) + 1]
+ }
+ htmlSwitch $URL($url,buf) \
title {
- if {!$ti} {
- headSave $url $out $body
- set ti 1
+ puts $out "<title>$body</title>"
+ } -nonest meta {
+ puts -nonewline $out "<meta"
+ foreach a [array names parm] {
+ puts -nonewline $out " $a"
+ puts -nonewline $out {="}
+ puts -nonewline $out $parm($a)
+ puts -nonewline $out {"}
}
+ puts $out {></meta>}
} body {
- regsub -all -nocase {<script.*</script>} $body {} abody
- regsub -all {<[^\>]+>} $abody {} nbody
+ regsub -all {<!--[^-]*->} $body { } abody
+ regsub -all -nocase {<script[^<]*</script>} $abody {} bbody
+ regsub -all {<[^\>]+>} $bbody {} nbody
puts $out "<documentcontent>"
puts $out $nbody
puts $out "</documentcontent>"
- } a {
+ } -nonest a {
if {![info exists parm(href)]} {
puts "no href"
continue
}
- if {!$ti} {
- headSave $url $out "untitled"
- set ti 1
- }
- if {1} {
- set href $parm(href)
+ if {[expr $distance <= $maxdistance]} {
+ set href [string trim $parm(href)]
if {![RobotHref $url href host path]} continue
puts $out "<cr>"
puts $out "</cr>"
if {![RobotFileExist visited $host $path]} {
- if {[catch {set outf [RobotFileOpen unvisited $host $path]} msg]} {
- puts "--- Error $msg"
- exit 1
+ set olddistance 1000
+ if {![RobotFileExist bad $host $path]} {
+ if {[RobotFileExist unvisited $host $path]} {
+ set inf [RobotFileOpen unvisited $host $path r]
+ RobotReadRecord $inf oldurl olddistance
+ RobotFileClose $inf
+ }
+ } else {
+ set olddistance 0
+ }
+ if {[string length $olddistance] == 0} {
+ set olddistance 1000
+ }
+ if {[expr $distance < $olddistance]} {
+ set outf [RobotFileOpen unvisited $host $path]
+ RobotWriteRecord $outf $url $distance
+ RobotFileClose $outf
+ }
+ } elseif {[string compare $href $url]} {
+ set inf [RobotFileOpen visited $host $path r]
+ RobotReadRecord $inf xurl olddistance
+ close $inf
+ if {[string length $olddistance] == 0} {
+ set olddistance 1000
+ }
+ if {[expr $distance < $olddistance]} {
+ puts "OK remarking url=$url href=$href"
+ puts "olddistance = $olddistance"
+ puts "newdistance = $distance"
+ set outf [RobotFileOpen unvisited $host $path]
+ RobotWriteRecord $outf $url $distance
+ RobotFileClose $outf
}
- close $outf
}
}
+ } -nonest area {
+ if {![info exists parm(href)]} {
+ puts "no href"
+ continue
+ }
+ if {[expr $distance <= $maxdistance]} {
+ set href [string trim $parm(href)]
+ if {![RobotHref $url href host path]} continue
+
+ puts $out "<cr>"
+ puts $out "<identifier>$href</identifier>"
+ puts $out "<description></description>"
+ puts $out "</cr>"
+
+ if {![RobotFileExist visited $host $path]} {
+ set olddistance 1000
+ if {![RobotFileExist bad $host $path]} {
+ if {[RobotFileExist unvisited $host $path]} {
+ set inf [RobotFileOpen unvisited $host $path r]
+ RobotReadRecord $inf oldurl olddistance
+ RobotFileClose $inf
+ }
+ } else {
+ set olddistance 0
+ }
+ if {[string length $olddistance] == 0} {
+ set olddistance 1000
+ }
+ if {[expr $distance < $olddistance]} {
+ set outf [RobotFileOpen unvisited $host $path]
+ RobotWriteRecord $outf $url $distance
+ RobotFileClose $outf
+ }
+ } elseif {[string compare $href $url]} {
+ set inf [RobotFileOpen visited $host $path r]
+ RobotReadRecord $inf xurl olddistance
+ close $inf
+ if {[string length $olddistance] == 0} {
+ set olddistance 1000
+ }
+ if {[expr $distance < $olddistance]} {
+ puts "OK remarking url=$url href=$href"
+ puts "olddistance = $olddistance"
+ puts "newdistance = $distance"
+ set outf [RobotFileOpen unvisited $host $path]
+ RobotWriteRecord $outf $url $distance
+ RobotFileClose $outf
+ }
+ }
+ }
+ }
+}
+
+proc RobotsTxt {url} {
+ global agent URL
+
+ RobotsTxt0 URL(URL($url,hostport),robots) $URL($url,buf)
+}
+
+proc RobotsTxt0 {v buf} {
+ global URL agent
+ set section 0
+ foreach l [split $buf \n] {
+ if {[regexp {([-A-Za-z]+):[ ]*([^\# ]+)} $l match cmd arg]} {
+ puts "cmd=$cmd arg=$arg"
+ switch -- [string tolower $cmd] {
+ user-agent {
+ if {$section} break
+ set pat [string tolower $arg]*
+ set section [string match $pat $agent]
+ }
+ disallow {
+ if {$section} {
+ puts "rule [list 0 $arg]"
+ lappend $v [list 0 $arg]
+ }
+ }
+ allow {
+ if {$section} {
+ puts "rule [list 1 $arg]"
+ lappend $v [list 1 $arg]
+ }
+ }
+ }
+ }
+ }
+}
+
+proc RobotTextPlain {url out} {
+ global URL
+
+ puts $out "<documentcontent>"
+ regsub -all {<} $URL($url,buf) {\<} content
+ puts $out $content
+ puts $out "</documentcontent>"
+
+ if {![string compare $URL($url,path) /robots.txt]} {
+ RobotsTxt $url
+ }
+}
+
+proc RobotWriteMetadata {url out} {
+ global URL domains
+
+ puts $out "<zmbot>"
+
+ set distance 1000
+ if {[RobotFileExist unvisited $URL($url,hostport) $URL($url,path)]} {
+ set inf [RobotFileOpen unvisited $URL($url,hostport) $URL($url,path) r]
+ RobotReadRecord $inf fromurl distance
+ RobotFileClose $inf
+ }
+ set URL($url,dist) $distance
+ puts $out "<distance>"
+ puts $out " $distance"
+ puts $out "</distance>"
+ headSave $url $out
+ puts "Parsing $url distance=$distance"
+ switch $URL($url,head,content-type) {
+ text/html {
+ if {[string length $distance]} {
+ RobotTextHtml $url $out
+ }
+ }
+ text/plain {
+ RobotTextPlain $url $out
+ }
+ application/pdf {
+ set pdff [open test.pdf w]
+ puts -nonewline $pdff $URL($url,buf)
+ close $pdff
}
}
- if {!$ti} {
- headSave $url $out "untitled"
- set ti 1
+ puts $out "</zmbot>"
+}
+
+proc Robot200 {url} {
+ global URL domains
+
+ set out [RobotFileOpen raw $URL($url,hostport) $URL($url,path)]
+ puts -nonewline $out $URL($url,buf)
+ RobotFileClose $out
+
+ if {![checkrule mime $URL($url,head,content-type)]} {
+ RobotError $url mimedeny
+ return
}
- puts $out "</meta>"
- close $out
- # puts "Parsing done"
- RobotFileUnlink unvisited $URL($url,host) $URL($url,path)
+ set out [RobotFileOpen visited $URL($url,hostport) $URL($url,path)]
+ RobotWriteMetadata $url $out
+ RobotFileClose $out
+
+ RobotFileUnlink unvisited $URL($url,hostport) $URL($url,path)
}
-proc RobotReadBody {url sock} {
+proc RobotReadContent {url sock binary} {
global URL
set buffer [read $sock 16384]
set readCount [string length $buffer]
-
+
if {$readCount <= 0} {
- close $sock
Robot200 $url
- RobotRestart
+ RobotRestart $url $sock
+ } elseif {!$binary && [string first \0 $buffer] >= 0} {
+ Robot200 $url
+ RobotRestart $url $sock
} else {
# puts "Got $readCount bytes"
set URL($url,buf) $URL($url,buf)$buffer
}
}
-proc RobotReadHead {url sock} {
- global URL
+proc RobotReadHeader {url sock} {
+ global URL debuglevel
- set buffer [read $sock 8192]
+ if {$debuglevel > 1} {
+ puts "HTTP head $url"
+ }
+ if {[catch {set buffer [read $sock 2148]}]} {
+ RobotError $url 404
+ RobotRestart $url $sock
+ }
set readCount [string length $buffer]
if {$readCount <= 0} {
- Robot404 $url
- close $sock
- RobotRestart
+ RobotError $url 404
+ RobotRestart $url $sock
} else {
# puts "Got $readCount bytes"
set URL($url,buf) $URL($url,buf)$buffer
- set n [string first \n\n $URL($url,buf)]
+ set n [string first \r\n\r\n $URL($url,buf)]
if {$n > 1} {
set code 0
set version {}
set headbuf [string range $URL($url,buf) 0 $n]
- incr n
- incr n
+ incr n 4
set URL($url,buf) [string range $URL($url,buf) $n end]
regexp {^HTTP/([0-9.]+)[ ]+([0-9]+)} $headbuf x version code
set lines [split $headbuf \n]
foreach line $lines {
- if {[regexp {^([^:]+):[ ]+(.*)} $line x name value]} {
- set URL($url,head,[string tolower $name]) $value
+ if {[regexp {^([^:]+):[ ]+([^;]*)} $line x name value]} {
+ set URL($url,head,[string tolower $name]) [string trim $value]
}
}
- puts "code = $code"
+ puts "HTTP CODE $code"
set URL($url,state) skip
switch $code {
301 {
- Robot301 $url $URL($url,head,location)
- close $sock
- RobotRestart
+ RobotRedirect $url $URL($url,head,location) 301
+ RobotRestart $url $sock
}
302 {
- Robot301 $url $URL($url,head,location)
- close $sock
- RobotRestart
- }
- 404 {
- Robot404 $url
- close $sock
- RobotRestart
- }
- 401 {
- Robot401 $url
- close $sock
- RobotRestart
+ RobotRedirect $url $URL($url,head,location) 302
+ RobotRestart $url $sock
}
200 {
- if {[info exists URL($url,head,content-type)]} {
- if {![string compare $URL($url,head,content-type) text/html]} {
- set URL($url,state) html
- }
+ if {![info exists URL($url,head,content-type)]} {
+ set URL($url,head,content-type) {}
}
- if {[string compare $URL($url,state) html]} {
- close $sock
- Robot200 $url
- RobotRestart
- } else {
- fileevent $sock readable [list RobotReadBody $url $sock]
+ set binary 0
+ switch $URL($url,head,content-type) {
+ application/pdf {
+ set binary 1
+ }
}
+ fileevent $sock readable [list RobotReadContent $url $sock $binary]
}
default {
- Robot404 $url
- close $sock
- RobotRestart
+ RobotError $url $code
+ RobotRestart $url $sock
}
}
}
}
}
+proc RobotSockCancel {url sock} {
+
+ puts "RobotSockCancel sock=$sock url=$url"
+ RobotError $url 401
+ RobotRestart $url $sock
+}
+
proc RobotConnect {url sock} {
- global URL
+ global URL agent acceptLanguage
- fconfigure $sock -translation {auto crlf} -blocking 0
- puts "Reading $url"
- fileevent $sock readable [list RobotReadHead $url $sock]
+ fconfigure $sock -translation {lf crlf} -blocking 0
+ fileevent $sock readable [list RobotReadHeader $url $sock]
puts $sock "GET $URL($url,path) HTTP/1.0"
puts $sock "Host: $URL($url,host)"
+ puts $sock "User-Agent: $agent"
+ if {[string length $acceptLanguage]} {
+ puts $sock "Accept-Language: $acceptLanguage"
+ }
puts $sock ""
flush $sock
+ set URL($sock,cancel) [after 30000 [list RobotSockCancel $url $sock]]
}
proc RobotNop {} {
}
proc RobotGetUrl {url phost} {
- global URL
- puts "---------"
- puts $url
- if {![regexp {([^:]+)://([^/]+)([^ ]*)} $url x method hostport path]} {
+ global URL robotsRunning
+ flush stdout
+ puts "Retrieve $robotsRunning url=$url"
+ if {![regexp {([^:]+)://([^/]+)(.*)} $url x method hostport path]} {
return -1
}
if {![regexp {([^:]+):([0-9]+)} $hostport x host port]} {
}
set URL($url,method) $method
set URL($url,host) $host
- set URL($url,port) $port
+ set URL($url,hostport) $hostport
set URL($url,path) $path
set URL($url,state) head
set URL($url,buf) {}
+
+ if {[string compare $path /robots.txt]} {
+ set ok 1
+ if {![info exists URL($hostport,robots)]} {
+ puts "READING robots.txt for host $hostport"
+ if {[RobotFileExist visited $hostport /robots.txt]} {
+ set inf [RobotFileOpen visited $hostport /robots.txt r]
+ set buf [read $inf 32768]
+ close $inf
+ } else {
+ set buf "User-agent: *\nAllow: /\n"
+ }
+ RobotsTxt0 URL($hostport,robots) $buf
+ }
+ if {[info exists URL($hostport,robots)]} {
+ foreach l $URL($hostport,robots) {
+ if {[string first [lindex $l 1] $path] == 0} {
+ set ok [lindex $l 0]
+ break
+ }
+ }
+ }
+ if {!$ok} {
+ puts "skipped due to robots.txt"
+ return -1
+ }
+ }
if [catch {set sock [socket -async $host $port]}] {
return -1
}
}
}
-if {[llength $argv] < 2} {
- puts "Tclrobot: usage <domain> <start>"
- puts " Example: '*.dk' www.indexdata.dk"
- exit 1
+set agent "zmbot/0.1"
+if {![catch {set os [exec uname -s -r]}]} {
+ set agent "$agent ($os)"
+}
+
+puts "agent: $agent"
+
+proc bgerror {m} {
+ global errorInfo
+ puts "BGERROR $m"
+ puts $errorInfo
}
+set robotsRunning 0
+set robotSeq 0
set workdir [pwd]
+set idletime 60000
+set acceptLanguage {}
+
+set i 0
+set l [llength $argv]
+
+if {$l < 2} {
+ puts {tclrobot: usage:}
+ puts {tclrobot [-j jobs] [-i idle] [-c count] [-d domain] [-r rules] [url ..]}
+ puts " Example: -c 3 -d '*.dk' http://www.indexdata.dk/"
+ exit 1
+}
+
+# Rules: allow, deny, url
+set debuglevel 0
+
+proc checkrule {type this} {
+ global alrules
+ global debuglevel
+
+ if {$debuglevel > 3} {
+ puts "CHECKRULE $type $this"
+ }
+ if {[info exist alrules]} {
+ foreach l $alrules {
+ if {$debuglevel > 3} {
+ puts "consider $l"
+ }
+ # consider type
+ if {[lindex $l 1] != $type} continue
+ # consider mask (! negates)
+ set masks [lindex $l 2]
+ set ok 0
+ foreach mask $masks {
+ if {$debuglevel > 4} {
+ puts "consider single mask $mask"
+ }
+ if {[string index $mask 0] == "!"} {
+ set mask [string range $mask 1 end]
+ if {[string match $mask $this]} continue
+ } else {
+ if {![string match $mask $this]} continue
+ }
+ set ok 1
+ }
+ if {$debuglevel > 4} {
+ puts "ok = $ok"
+ }
+ if {!$ok} continue
+ # OK, we have a match
+ if {[lindex $l 0] == "allow"} {
+ if {$debuglevel > 3} {
+ puts "CHECKRULE MATCH OK"
+ }
+ return 1
+ } else {
+ if {$debuglevel > 3} {
+ puts "CHECKFULE MATCH FAIL"
+ }
+ return 0
+ }
+ }
+ }
+ if {$debuglevel > 3} {
+ puts "CHECKRULE MATCH OK"
+ }
+ return 1
+}
+
+
+proc url {href} {
+ global debuglevel
-set domains [lindex $argv 0]
-set site [lindex $argv 1]
-if {[string length $site]} {
- set x [RobotFileOpen unvisited $site /]
- close $x
+ if {[RobotHref http://www.indexdata.dk/ href host path]} {
+ if {![RobotFileExist visited $host $path]} {
+ set outf [RobotFileOpen unvisited $host $path]
+ RobotWriteRecord $outf href 0
+ RobotFileClose $outf
+ }
+ }
}
+proc deny {type stuff} {
+ global alrules
+
+ lappend alrules [list deny $type $stuff]
+}
-RobotRestart
-vwait forever
+proc allow {type stuff} {
+ global alrules
+
+ lappend alrules [list allow $type $stuff]
+}
+
+proc debug {level} {
+ global debuglevel
+
+ set debuglevel $level
+}
+
+# Parse options
+
+while {$i < $l} {
+ set arg [lindex $argv $i]
+ switch -glob -- $arg {
+ -j* {
+ set robotsMax [string range $arg 2 end]
+ if {![string length $robotsMax]} {
+ set robotsMax [lindex $argv [incr i]]
+ }
+ }
+ -c* {
+ set maxdistance [string range $arg 2 end]
+ if {![string length $maxdistance]} {
+ set maxdistance [lindex $argv [incr i]]
+ }
+ }
+ -d* {
+ set dom [string range $arg 2 end]
+ if {![string length $dom]} {
+ set dom [lindex $argv [incr i]]
+ }
+ lappend domains $dom
+ }
+ -i* {
+ set idletime [string range $arg 2 end]
+ if {![string length $idletime]} {
+ set idletime [lindex $argv [incr i]]
+ }
+ }
+ -l* {
+ set acceptLanguage [string range $arg 2 end]
+ if {![string length $acceptLanguage]} {
+ set acceptLanguage [lindex $argv [incr i]]
+ }
+ }
+ -r* {
+ set rfile [string range $arg 2 end]
+ if {![string length $rfile]} {
+ set rfile [lindex $argv [incr i]]
+ }
+ source $rfile
+ }
+ default {
+ set href $arg
+ if {[RobotHref http://www.indexdata.dk/ href host path]} {
+ if {![RobotFileExist visited $host $path]} {
+ set outf [RobotFileOpen unvisited $host $path]
+ RobotWriteRecord $outf href 0
+ RobotFileClose $outf
+ }
+ }
+ }
+ }
+ incr i
+}
+
+if {![info exist domains]} {
+ set domains {*}
+}
+if {![info exist maxdistance]} {
+ set maxdistance 50
+}
+if {![info exist robotsMax]} {
+ set robotsMax 5
+}
+
+puts "domains=$domains"
+puts "max distance=$maxdistance"
+puts "max jobs=$robotsMax"
+
+RobotStart
+
+while {$robotsRunning} {
+ vwait robotsRunning
+}