--- /dev/null
+#!/usr/bin/tclsh
+# $Id: tkl-web-harvester,v 1.1 2003/08/14 08:02:10 marc Exp $
+#
+set loghandle stdout
+set robotsRunning 0
+set workdir [pwd]
+set idletime 15000
+set acceptLanguage {}
+set debuglevel 1
+set libdir ""
+
+
+proc logmsg {msg} {
+ global loghandle
+
+ puts $loghandle $msg
+ flush $loghandle
+}
+
+#proc dbgmsg {level msg} {
+# global debuglevel
+# if {[expr $debuglevel >= $level]} {
+# logmsg $msg
+# }
+#}
+proc dbgmsg {msg} {
+ global debuglevel
+ if {[expr $debuglevel >= 0]} {
+ logmsg $msg
+ }
+}
+# dbgmsg is always called with just one string!
+
+
+proc fnameEncode {fname} {
+ regsub -all {&} $fname {%38} fname
+ regsub -all {<} $fname {%3C} fname
+ regsub -all {>} $fname {%3E} fname
+ regsub -all {\?} $fname {%3F} fname
+ regsub -all {\*} $fname {%2A} fname
+ return $fname
+}
+
+proc fnameDecode {fname} {
+ regsub -all {%38} $fname {&} fname
+ regsub -all {%3C} $fname {<} fname
+ regsub -all {%3E} $fname {>} fname
+ regsub -all {%3F} $fname {?} fname
+ regsub -all {%2A} $fname {*} fname
+ return $fname
+}
+
+proc RobotFileNext1 {area lead} {
+ # dbgmsg "RobotFileNext1 area=$area lead=$lead"
+ if {[catch {set ns [glob ${area}/*]}]} {
+ return {}
+ }
+ foreach n $ns {
+ if {[file isfile $n]} {
+ set off [string last / $n]
+ # skip /
+ incr off
+ set end [string length $n]
+ # skip _.tkl
+ incr end -6
+ return $lead/[string range $n $off $end]
+ }
+ }
+ foreach n $ns {
+ if {[file isdirectory $n]} {
+ set off [string last / $n]
+ # skip /
+ incr off
+ set sb [RobotFileNext1 $n $lead/[string range $n $off end]]
+ if {[string length $sb]} {
+ return $sb
+ }
+ }
+ }
+ return {}
+}
+
+proc RobotWriteRecord {outf fromurl distance} {
+ puts $outf {<?xml version="1.0" encoding="ISO-8859-1" standalone="yes"?>}
+ 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
+ gets $inf
+ set distance [string trim [gets $inf]]
+ # dbgmsg "got distance = $distance"
+ gets $inf
+ gets $inf
+ set fromurl [string trim [gets $inf]]
+}
+
+proc RobotFileNext {task area} {
+ global control
+ global idletime ns
+ global status
+
+ # dbgmsg "RobotFileNext seq=$control($task,seq)"
+ if {$control($task,seq) < 0} {
+ return {}
+ }
+ set target $control($task,target)
+ if {$control($task,seq) == 0} {
+ if {[catch {set ns($task) [glob $target/$area/*]}]} {
+ puts "----------- DONE-------- target=$target"
+ return done
+ }
+ }
+ # dbgmsg "ns=$ns($task)"
+ set off [string length $target/$area]
+ incr off
+ set n [lindex $ns($task) $control($task,seq)]
+ # dbgmsg "n=$n"
+ if {![string length $n]} {
+ set control($task,seq) -1
+ set statusfile [open $target/status w]
+ puts $statusfile "$status($task,unvisited) $status($task,bad) $status($task,visited)"
+ close $statusfile
+ return wait
+ }
+ incr control($task,seq)
+ if {[file isfile $n/robots.txt_.tkl]} {
+ # dbgmsg "ok returning http://[string range $n $off end]/robots.txt"
+ return [fnameDecode 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 [fnameDecode $sb]
+ }
+ }
+ dbgmsg "no more work at end of RobotFileNext n=$n"
+ dbgmsg "ns=$ns($task)"
+ return {}
+}
+
+
+proc RobotFileExist {task area host path} {
+ global debuglevel control
+
+ if {$debuglevel > 3} {
+ dbgmsg "RobotFileExist begin area=$area host=$host path=$path"
+ }
+ set target $control($task,target)
+ return [file exists [fnameEncode $target/$area/$host${path}_.tkl]]
+}
+
+proc RobotFileUnlink {task area host path} {
+ global status control
+
+ set target $control($task,target)
+ # dbgmsg "RobotFileUnlink begin"
+ # dbgmsg "area=$area host=$host path=$path"
+ set npath [fnameEncode $target/$area/$host${path}_.tkl]
+ # dbgmsg "npath=$npath"
+ set comp [split $npath /]
+ if {[catch {exec rm $npath}]} return
+
+ set l [llength $comp]
+ incr l -2
+ incr status($task,$area) -1
+ for {set i $l} {$i > 0} {incr i -1} {
+ set path [join [lrange $comp 0 $i] /]
+ if {![catch {glob $path/*}]} return
+ exec rmdir $path
+ }
+ # dbgmsg "RobotFileUnlink end"
+}
+
+proc RobotFileClose {out} {
+ if [string compare $out stdout] {
+ close $out
+ }
+}
+
+proc RobotFileOpen {task area host path {mode w}} {
+ set orgPwd [pwd]
+ global workdir status debuglevel control
+
+ # dbgmsg "RobotFileOpen task=$task path=$path"
+
+ set target $control($task,target)
+ set path [fnameEncode $path]
+
+ if {![info exists workdir]} {
+ return stdout
+ }
+ if {$debuglevel > 3} {
+ dbgmsg "RobotFileOpen orgPwd=$orgPwd area=$area host=$host path=$path mode=$mode"
+ }
+ if {[string compare $orgPwd $workdir]} {
+ dbgmsg "ooops. RobotFileOpen failed"
+ dbgmsg "workdir = $workdir"
+ dbgmsg "pwd = $orgPwd"
+ exit 1
+ }
+
+ set comp [split $target/$area/$host /]
+ set len [llength $comp]
+ incr len -1
+
+ # dbgmsg "1 comp=$comp"
+
+ for {set i 0} {$i <= $len} {incr i} {
+ set d [lindex $comp $i]
+ if {[string length $d] == 0} {
+ cd /
+ } elseif {[catch {cd $d}]} {
+ exec mkdir $d
+ cd ./$d
+ if {![string compare $area unvisited] && $i == $len && $mode == "w"} {
+ if {[string compare $path /robots.txt]} {
+ set out [open robots.txt_.tkl w]
+ dbgmsg "creating robots.txt in $d"
+ close $out
+ incr status($task,unvisited)
+ }
+ }
+ }
+ }
+
+ set comp [split $path /]
+ set len [llength $comp]
+ incr len -1
+
+ # dbgmsg "2 path=$path comp=$comp"
+
+ for {set i 0} {$i < $len} {incr i} {
+ set d [lindex $comp $i]
+ if {[string length $d] > 0} {
+ if {[catch {cd $d}]} {
+ exec mkdir $d
+ cd ./$d
+ }
+ }
+ }
+ set d [lindex $comp $len]
+ set out [open ${d}_.tkl $mode]
+ if {$mode == "w"} {
+ incr status($task,$area)
+ }
+ cd $orgPwd
+ return $out
+}
+
+proc RobotStartJob {root task} {
+ global control
+
+ set fname "$root$task"
+ set f [open $fname r]
+ set xml [read $f]
+ dbgmsg "Reading $fname"
+ close $f
+ # task type must be 2
+ if {![regexp {<tasktype>([^<]*)</tasktype>} $xml x tasktype]} {
+ return
+ }
+ set tasktype [string trim $tasktype]
+ if {![string match 2 $tasktype]} {
+ return
+ }
+ # status must not be finished or error
+ if {![regexp {<status>([^<]*)</status>} $xml x status]} {
+ return
+ }
+ if {$status == "finished"} {
+ dbgmsg "already finished"
+ return
+ }
+ if {$status == "error"} {
+ dbgmsg "already finished due to error"
+ return
+ }
+ # ignore if task has already been processed
+ dbgmsg "status = $status"
+ if {![CreateTask $task]} {
+ return
+ }
+ set control($task,taskfname) $fname
+ dbgmsg "Reading $fname stage 2"
+ htmlSwitch $xml \
+ url {
+ lappend starturls $body
+ } filter {
+ set type $parm(type)
+ set action $parm(action)
+ if {$type == "domain"} {
+ $action url http://$body/*
+ }
+ if {$type == "url"} {
+ $action url $body
+ }
+ if {$type == "mime"} {
+ $action mime $body
+ }
+ } target {
+ set ex [file rootname [file tail $task]]
+ #set control($task,target) "$root$body/$ex"
+ set control($task,target) "$control(tmpdir)/$ex"
+ set control($task,output) "$root$body"
+ } distance {
+ set control($task,distance) $body
+ } status {
+ set control($task,filestatus) $body
+ } tasktype {
+ set control($task,tasktype) $body
+ }
+
+ if {[info exists starturls]} {
+ foreach url $starturls {
+ puts "marking start urls $url"
+ url $url
+ }
+ }
+
+ if {$status == "pending"} {
+ regsub {<status>[^<]*</status>} $xml {<status>running</status>} xml2
+ set f [open $fname w]
+ puts -nonewline $f $xml2
+ close $f
+ }
+}
+
+proc RobotDoneJob {task} {
+ global daemon_dir control
+
+ if {![info exists daemon_dir]} {
+ return
+ }
+ set fname $control($task,taskfname)
+ set f [open $fname r]
+ set xml [read $f]
+ dbgmsg "Reading $fname"
+ regexp {<status>([^<]*)</status>} $xml x status
+ dbgmsg "------"
+ dbgmsg "status = $status"
+ close $f
+
+ regsub {<status>[^<]*</status>} $xml {<status>finished</status>} xml2
+ set f [open $fname w]
+ puts -nonewline $f $xml2
+ close $f
+}
+
+proc RobotScanDir {} {
+ global daemon_dir
+
+ if {![info exists daemon_dir]} {
+ return
+ }
+ foreach d $daemon_dir {
+ if {[catch {set files [glob $d/*.spl]}]} {
+ return
+ }
+ foreach fname $files {
+ if {[file isfile $fname] && [file readable $fname]} {
+ set jobfile [open $fname]
+ gets $jobfile portalroot
+ gets $jobfile portaltask
+ close $jobfile
+
+ RobotStartJob $portalroot $portaltask
+ }
+ }
+ }
+}
+
+proc RobotRR {task} {
+ global control robotsRunning tasks robotsMax status
+
+ dbgmsg "RobotRR -- running=$robotsRunning max=$robotsMax---------------"
+ incr robotsRunning -1
+
+ # only one task gets through...
+ if {[string compare [lindex $tasks 0] $task]} {
+ return
+ }
+ dbgmsg "RobotRR. task = $task"
+ while {$robotsRunning} {
+ vwait robotsRunning
+ }
+ dbgmsg "Scan"
+ if {[catch {RobotScanDir} msg]} {
+ logmsg "RobotScanDir failed"
+ logmsg $msg
+ }
+ foreach t $tasks {
+ set target $control($t,target)
+ set statusfile [open $target/status w]
+ puts $statusfile "$status($t,unvisited) $status($t,bad) $status($t,visited)"
+ close $statusfile
+ set control($t,seq) 0
+ RobotStart $t
+ }
+}
+
+proc RobotDaemonSig {} {
+ global daemon_cnt
+
+ incr daemon_cnt
+}
+
+proc RobotDaemonLoop {} {
+ global daemon_cnt tasks robotsRunning status
+
+ set daemon_cnt 0
+ while 1 {
+ logmsg $daemon_cnt
+ RobotScanDir
+
+ if {[info exists tasks]} {
+ logmsg "daemon loop tasks $tasks"
+ foreach t $tasks {
+ set control($t,seq) 0
+ RobotStart $t
+ }
+ while {$robotsRunning} {
+ vwait robotsRunning
+ }
+ }
+ after 30000 RobotDaemonSig
+ vwait daemon_cnt
+ }
+}
+
+proc RobotRestart {task url sock} {
+ global URL robotsRunning
+
+ close $sock
+ after cancel $URL($sock,cancel)
+
+ foreach v [array names URL $task,$url,*] {
+ unset URL($v)
+ }
+
+ incr robotsRunning -1
+ RobotStart $task
+}
+
+proc RobotStart {task} {
+ global URL
+ global robotsRunning robotsMax idletime status tasks
+
+ # dbgmsg "RobotStart $task running=$robotsRunning"
+ while {1} {
+ set url [RobotFileNext $task unvisited]
+ if {[string compare $url done] == 0} {
+ dbgmsg "In RobotStart task $task done"
+
+ catch {unset ntasks}
+ foreach t $tasks {
+ if {[string compare $t $task]} {
+ lappend ntasks $t
+ } else {
+ dbgmsg "task $t done"
+ }
+ }
+ if {![info exists ntasks]} {
+ unset tasks
+ dbgmsg "all done"
+ } else {
+ set tasks $ntasks
+ }
+ RobotDoneJob $task
+ return
+ }
+ if {![string length $url]} {
+ return
+ }
+ incr robotsRunning
+ if {[string compare $url wait] == 0} {
+ after $idletime [list RobotRR $task]
+ return
+ }
+ set r [RobotGetUrl $task $url {}]
+ if {!$r} {
+ if {$robotsRunning >= $robotsMax} return
+ } else {
+ incr robotsRunning -1
+ if {![RobotFileExist $task bad $URL($task,$url,hostport) $URL($task,$url,path)]} {
+ set outf [RobotFileOpen $task bad $URL($task,$url,hostport) $URL($task,$url,path)]
+ RobotFileClose $outf
+ }
+ RobotFileUnlink $task unvisited $URL($task,$url,hostport) $URL($task,$url,path)
+ }
+ }
+}
+
+proc headSave {task url out} {
+ global URL
+
+ if {[info exists URL($task,$url,head,last-modified)]} {
+ puts $out "<lastmodified>$URL($task,$url,head,last-modified)</lastmodified>"
+ }
+ puts $out {<si>}
+ if {[info exists URL($task,$url,head,date)]} {
+ puts $out " <date>$URL($task,$url,head,date)</date>"
+ }
+ if {[info exists URL($task,$url,head,content-length)]} {
+ puts $out " <by>$URL($task,$url,head,content-length)</by>"
+ }
+ if {[info exists URL($task,$url,head,server)]} {
+ puts $out " <format>$URL($task,$url,head,server)</format>"
+ }
+ puts $out {</si>}
+ puts $out {<publisher>}
+ puts $out " <identifier>$url</identifier>"
+ if {[info exists URL($task,$url,head,content-type)]} {
+ puts $out " <type>$URL($task,$url,head,content-type)</type>"
+ }
+ puts $out {</publisher>}
+}
+
+proc RobotHref {task url hrefx hostx pathx} {
+ global URL control debuglevel
+ upvar $hrefx href
+ upvar $hostx host
+ upvar $pathx path
+
+ if {$debuglevel > 1} {
+ dbgmsg "Ref input url = $url href=$href"
+ }
+
+ if {[string first { } $href] >= 0} {
+ return 0
+ }
+ if {[string length $href] > 256} {
+ return 0
+ }
+
+# Skip pages that have ? in them
+# if {[string first {?} $url] >= 0 && [string first {?} $href] >= 0} {
+# return 0
+# }
+ # get method (if any)
+ if {![regexp {^([^/:]+):(.*)} $href x method hpath]} {
+ set hpath $href
+ set method http
+ } else {
+ if {[string compare $method http]} {
+ return 0
+ }
+ }
+ # get host (if any)
+ if {[regexp {^//([^/]+)([^\#]*)} $hpath x host surl]} {
+ if {![string length $surl]} {
+ set surl /
+ }
+ if {[info exist control($task,domains)]} {
+ set ok 0
+ foreach domain $control($task,domains) {
+ if {[string match $domain $host]} {
+ set ok 1
+ break
+ }
+ }
+ if {!$ok} {
+ return 0
+ }
+ }
+ } else {
+ regexp {^([^\#]*)} $hpath x surl
+ set host $URL($task,$url,hostport)
+ }
+ if {![string length $surl]} {
+ return 0
+ }
+ if {[string first / $surl]} {
+ # relative path
+ set curpath $URL($task,$url,path)
+ if {[info exists URL($task,$url,bpath)]} {
+ set curpath $URL($task,$url,bpath)
+ }
+ regexp {^([^\#?]*)} $curpath x dpart
+ set l [string last / $dpart]
+ if {[expr $l >= 0]} {
+ set surl [string range $dpart 0 $l]$surl
+ } else {
+ set surl $dpart/$surl
+ }
+ }
+ set surllist [split $surl /]
+ catch {unset path}
+ set pathl 0
+ foreach c $surllist {
+ switch -- $c {
+ .. {
+ if {$pathl > 1} {
+ incr pathl -2
+ set path [lrange $path 0 $pathl]
+ incr pathl
+ }
+ }
+ . {
+
+ }
+ default {
+ incr pathl
+ lappend path $c
+ }
+ }
+ }
+ if {$debuglevel > 4} {
+ dbgmsg "pathl=$pathl output path=$path"
+ }
+ set path [join $path /]
+ if {![string length $path]} {
+ set path /
+ }
+ regsub -all {~} $path {%7E} path
+ set href "$method://$host$path"
+
+ if {$debuglevel > 1} {
+ dbgmsg "Ref result = $href"
+ }
+ return [checkrule $task url $href]
+}
+
+proc RobotError {task url code} {
+ global URL
+
+ dbgmsg "Bad URL $url (code $code)"
+ set fromurl {}
+ set distance -1
+ if {[RobotFileExist $task unvisited $URL($task,$url,hostport) $URL($task,$url,path)]} {
+ set inf [RobotFileOpen $task unvisited $URL($task,$url,hostport) $URL($task,$url,path) r]
+ RobotReadRecord $inf fromurl distance
+ RobotFileClose $inf
+ }
+ RobotFileUnlink $task unvisited $URL($task,$url,hostport) $URL($task,$url,path)
+ if {![RobotFileExist $task bad $URL($task,$url,hostport) $URL($task,$url,path)]} {
+ set outf [RobotFileOpen $task bad $URL($task,$url,hostport) $URL($task,$url,path)]
+ RobotWriteRecord $outf $fromurl $distance
+ RobotFileClose $outf
+ }
+}
+
+proc RobotRedirect {task url tourl code} {
+ global URL
+
+ dbgmsg "Redirecting from $url to $tourl"
+
+ set distance {}
+ set fromurl {}
+ if {[RobotFileExist $task unvisited $URL($task,$url,hostport) $URL($task,$url,path)]} {
+ set inf [RobotFileOpen $task unvisited $URL($task,$url,hostport) $URL($task,$url,path) r]
+ RobotReadRecord $inf fromurl distance
+ RobotFileClose $inf
+ }
+ if {![RobotFileExist $task bad $URL($task,$url,hostport) $URL($task,$url,path)]} {
+ set outf [RobotFileOpen $task bad $URL($task,$url,hostport) $URL($task,$url,path)]
+ RobotWriteRecord $outf $fromurl $distance
+ RobotFileClose $outf
+ }
+ if {[RobotHref $task $url tourl host path]} {
+ if {![RobotFileExist $task visited $host $path]} {
+ if {![RobotFileExist $task unvisited $host $path]} {
+ set outf [RobotFileOpen $task unvisited $host $path]
+ RobotWriteRecord $outf $fromurl $distance
+ RobotFileClose $outf
+ }
+ } else {
+ set olddistance {}
+ set inf [RobotFileOpen $task 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
+ }
+ dbgmsg "distance=$distance olddistance=$olddistance"
+ if {[expr $distance < $olddistance]} {
+ set outf [RobotFileOpen $task unvisited $host $path]
+ RobotWriteRecord $outf $tourl $distance
+ RobotFileClose $outf
+ }
+ }
+ }
+ if {[catch {RobotFileUnlink $task unvisited $URL($task,$url,hostport) $URL($task,$url,path)}]} {
+ dbgmsg "unlink failed"
+ exit 1
+ }
+}
+
+proc wellform {body} {
+ regsub -all {<!--[^-]*-->} $body { } abody
+ regsub -all -nocase {<script[^<]*</script>} $abody {} body
+ regsub -all {<[^\>]+>} $body {} abody
+ regsub -all { } $abody { } body
+ regsub -all {&} $body {&} abody
+ return $abody
+}
+
+proc link {task url out href body distance} {
+ global URL control
+ if {[expr $distance > $control($task,distance)]} return
+
+ if {![RobotHref $task $url href host path]} return
+
+ if ($control($task,cr)) {
+ puts $out "<cr>"
+ puts $out "<identifier>$href</identifier>"
+ set abody [wellform $body]
+ puts $out "<description>$abody</description>"
+ puts $out "</cr>"
+ }
+
+ if {![RobotFileExist $task visited $host $path]} {
+ set olddistance 1000
+ if {![RobotFileExist $task bad $host $path]} {
+ if {[RobotFileExist $task unvisited $host $path]} {
+ set inf [RobotFileOpen $task 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 $task unvisited $host $path]
+ RobotWriteRecord $outf $url $distance
+ RobotFileClose $outf
+ }
+ } elseif {[string compare $href $url]} {
+ set inf [RobotFileOpen $task visited $host $path r]
+ RobotReadRecord $inf xurl olddistance
+ close $inf
+ if {[string length $olddistance] == 0} {
+ set olddistance 1000
+ }
+ if {[expr $distance < $olddistance]} {
+ dbgmsg "OK remarking url=$url href=$href"
+ dbgmsg "olddistance = $olddistance"
+ dbgmsg "newdistance = $distance"
+ set outf [RobotFileOpen $task unvisited $host $path]
+ RobotWriteRecord $outf $url $distance
+ RobotFileClose $outf
+ }
+ }
+}
+
+proc RobotTextTkl {task url out} {
+ global URL control
+
+ # set title so we can emit it for the body
+ set title {}
+ # if true, nothing will be indexed
+ set noindex 0
+ # if true, nothing will be followed
+ set nofollow 0
+
+ puts $control($task,output)
+
+ set out stdout
+ set distance distance
+
+ htmlSwitch $URL($task,$url,buf) \
+ title {
+ # når title tag er hittet, er body set til indholdet af tagget
+ set title $body
+ } -nonest meta {
+ #puts -nonewline $out "<meta"
+ # al er list med attribut navne som fandtes ind i parm hash
+ set al [array names parm]
+ # løkke igennem attributer
+ foreach a $al {
+ set al [string tolower $a]
+ #puts -nonewline $out " $al"
+ #puts -nonewline $out {="}
+ #puts -nonewline $out $parm($a)
+ #puts -nonewline $out {"}
+ unset parm($al)
+ }
+ #puts $out "></meta>"
+ } body {
+ # don't print title of document content if noindex is used
+ if {!$noindex} {
+ #puts $out "<title>$title</title>"
+ # xml compilancy added
+ set bbody [wellform $body]
+ #puts $out "<documentcontent>"
+ #puts $out $bbody
+ #puts $out "</documentcontent>"
+ }
+ } -nonest base {
+ # <base href=.. >
+ if {![info exists parm(href)]} {
+ continue
+ }
+ set href [string trim $parm(href)]
+ } a {
+ # <a href="...."> .. </a>
+ # we're not using nonest - otherwise body isn't set
+ if {$nofollow} continue
+ if {![info exists parm(href)]} {
+ continue
+ }
+ #puts "link $task $url $out [string trim $parm(href)] $body $distance"
+ } -nonest area {
+ if {$nofollow} continue
+ if {![info exists parm(href)]} {
+ continue
+ }
+ #puts "link $task $url $out [string trim $parm(href)] $body $distance"
+ } -nonest frame {
+ if {![info exists parm(src)]} {
+ continue
+ }
+ #puts "link $task $url $out [string trim $parm(src)] $body $fdistance"
+ }
+}
+
+proc RobotTextHtml {task url out} {
+ global URL control
+
+ # set title so we can emit it for the body
+ set title {}
+ # if true, nothing will be indexed
+ set noindex 0
+ # if true, nothing will be followed
+ set nofollow 0
+
+ set distance 0
+ set fdistance 0
+ if {$control($task,distance) < 1000 && [info exists URL($task,$url,dist)]} {
+ set fdistance $URL($task,$url,dist)
+ set distance [expr $fdistance + 1]
+ }
+ htmlSwitch $URL($task,$url,buf) \
+ title {
+ set title $body
+ } -nonest meta {
+ # collect metadata and save NAME= CONTENT=..
+ set metaname {}
+ set metacontent {}
+ puts -nonewline $out "<meta"
+ set al [array names parm]
+ foreach a $al {
+ set al [string tolower $a]
+ puts -nonewline $out " $al"
+ puts -nonewline $out {="}
+ puts -nonewline $out $parm($a)
+ puts -nonewline $out {"}
+ switch -- $al {
+ "name" {
+ set metaname [string tolower $parm($a)]
+ }
+ "content" {
+ set metacontent $parm($a)
+ }
+ }
+ unset parm($al)
+ }
+ puts $out "></meta>"
+ # go through robots directives (af any)
+ if {![string compare $metaname robots]} {
+ set direcs [split [string tolower $metacontent] ,]
+ if {[lsearch $direcs noindex] >= 0} {
+ set noindex 1
+ }
+ if {[lsearch $direcs nofollow] >= 0} {
+ set nofollow 1
+ }
+ }
+ } body {
+ # don't print title of document content if noindex is used
+ if {!$noindex} {
+ puts $out "<title>$title</title>"
+ set bbody [wellform $body]
+ puts $out "<documentcontent>"
+ puts $out $bbody
+ puts $out "</documentcontent>"
+ }
+ } -nonest base {
+ # <base href=.. >
+ if {![info exists parm(href)]} {
+ continue
+ }
+ set href [string trim $parm(href)]
+ if {![RobotHref $task $url href host path]} continue
+ set URL($task,$url,bpath) $path
+ } a {
+ # <a href="...."> .. </a>
+ # we're not using nonest - otherwise body isn't set
+ if {$nofollow} continue
+ if {![info exists parm(href)]} {
+ continue
+ }
+ link $task $url $out [string trim $parm(href)] $body $distance
+ } -nonest area {
+ if {$nofollow} continue
+ if {![info exists parm(href)]} {
+ continue
+ }
+ link $task $url $out [string trim $parm(href)] $body $distance
+ } -nonest frame {
+ if {![info exists parm(src)]} {
+ continue
+ }
+ link $task $url $out [string trim $parm(src)] $body $fdistance
+ }
+}
+
+proc RobotsTxt {task url} {
+ global agent URL
+
+ RobotsTxt0 $task URL(URL($task,$url,hostport),robots) $URL($task,$url,buf)
+}
+
+proc RobotsTxt0 {task v buf} {
+ global URL agent
+ set section 0
+ foreach l [split $buf \n] {
+ if {[regexp {([-A-Za-z]+):[ ]*([^\# ]+)} $l match cmd arg]} {
+ set arg [string trim $arg]
+ dbgmsg "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} {
+ dbgmsg "rule [list 0 $arg]"
+ lappend $v [list 0 $arg]
+ }
+ }
+ allow {
+ if {$section} {
+ dbgmsg "rule [list 1 $arg]"
+ lappend $v [list 1 $arg]
+ }
+ }
+ }
+ }
+ }
+}
+
+proc RobotTextPlain {task url out} {
+ global URL
+
+ puts $out "<documentcontent>"
+ regsub -all {<} $URL($task,$url,buf) {\<} content
+ puts $out $content
+ puts $out "</documentcontent>"
+
+ if {![string compare $URL($task,$url,path) /robots.txt]} {
+ RobotsTxt $task $url
+ }
+}
+
+proc RobotWriteMetadata {task url out} {
+ global URL
+
+ set charset $URL($task,$url,charset)
+ puts $out "<?xml version=\"1.0\" encoding=\"$charset\" standalone=\"yes\"?>"
+ puts $out "<zmbot>"
+
+ set distance 1000
+ if {[RobotFileExist $task unvisited $URL($task,$url,hostport) $URL($task,$url,path)]} {
+ set inf [RobotFileOpen $task unvisited $URL($task,$url,hostport) $URL($task,$url,path) r]
+ RobotReadRecord $inf fromurl distance
+ RobotFileClose $inf
+ }
+ set URL($task,$url,dist) $distance
+ puts $out "<distance>"
+ puts $out " $distance"
+ puts $out "</distance>"
+ headSave $task $url $out
+ logmsg "Parsing $url distance=$distance"
+ switch $URL($task,$url,head,content-type) {
+ text/html {
+ if {[string length $distance]} {
+ RobotTextHtml $task $url $out
+ RobotTextTkl $task $url $out
+ }
+ }
+ text/plain {
+ RobotTextPlain $task $url $out
+ }
+ }
+ puts $out "</zmbot>"
+}
+
+proc Robot200 {task url} {
+ global URL
+
+ set out [RobotFileOpen $task raw $URL($task,$url,hostport) $URL($task,$url,path)]
+ puts -nonewline $out $URL($task,$url,buf)
+ RobotFileClose $out
+
+ set out [RobotFileOpen $task visited $URL($task,$url,hostport) $URL($task,$url,path)]
+ RobotWriteMetadata $task $url $out
+ RobotFileClose $out
+
+ RobotFileUnlink $task unvisited $URL($task,$url,hostport) $URL($task,$url,path)
+}
+
+proc RobotReadContent {task url sock binary} {
+ global URL
+
+ set buffer [read $sock 16384]
+ set readCount [string length $buffer]
+
+ if {$readCount <= 0} {
+ Robot200 $task $url
+ RobotRestart $task $url $sock
+ } elseif {!$binary && [string first \0 $buffer] >= 0} {
+ Robot200 $task $url
+ RobotRestart $task $url $sock
+ } else {
+ # dbgmsg "Got $readCount bytes"
+ set URL($task,$url,buf) $URL($task,$url,buf)$buffer
+ }
+}
+
+proc RobotReadHeader {task url sock} {
+ global URL debuglevel
+
+ if {$debuglevel > 1} {
+ dbgmsg "HTTP head $url"
+ }
+ if {[catch {set buffer [read $sock 2148]}]} {
+ RobotError $task $url 404
+ RobotRestart $task $url $sock
+ return
+ }
+ set readCount [string length $buffer]
+
+ if {$readCount <= 0} {
+ RobotError $task $url 404
+ RobotRestart $task $url $sock
+ } else {
+ # dbgmsg "Got $readCount bytes"
+ set URL($task,$url,buf) $URL($task,$url,buf)$buffer
+
+ set n [string first \r\n\r\n $URL($task,$url,buf)]
+ if {$n > 1} {
+ set code 0
+ set version {}
+ set headbuf [string range $URL($task,$url,buf) 0 $n]
+ incr n 4
+ set URL($task,$url,charset) ISO-8859-1
+ set URL($task,$url,buf) [string range $URL($task,$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($task,$url,head,[string tolower $name]) [string trim $value]
+ }
+ regexp {^Content-Type:.*charset=([A-Za-z0-9_-]*)} $line x URL($task,$url,charset)
+ }
+ dbgmsg "HTTP CODE $code"
+ set URL($task,$url,state) skip
+ switch $code {
+ 301 {
+ RobotRedirect $task $url $URL($task,$url,head,location) 301
+ RobotRestart $task $url $sock
+ }
+ 302 {
+ RobotRedirect $task $url $URL($task,$url,head,location) 302
+ RobotRestart $task $url $sock
+ }
+ 200 {
+ if {![info exists URL($task,$url,head,content-type)]} {
+ set URL($task,$url,head,content-type) {}
+ }
+ set binary 1
+ switch -glob -- $URL($task,$url,head,content-type) {
+ text/* {
+ set binary 0
+ }
+ }
+ if {![regexp {/robots.txt$} $url]} {
+ if {![checkrule $task mime $URL($task,$url,head,content-type)]} {
+ RobotError $task $url mimedeny
+ RobotRestart $task $url $sock
+ return
+ }
+ }
+ fileevent $sock readable [list RobotReadContent $task $url $sock $binary]
+ }
+ default {
+ RobotError $task $url $code
+ RobotRestart $task $url $sock
+ }
+ }
+ }
+ }
+}
+
+proc RobotSockCancel {task url sock} {
+
+ logmsg "RobotSockCancel sock=$sock url=$url"
+ RobotError $task $url 401
+ RobotRestart $task $url $sock
+}
+
+proc RobotConnect {task url sock} {
+ global URL agent acceptLanguage
+
+ fconfigure $sock -translation {lf crlf} -blocking 0
+ fileevent $sock readable [list RobotReadHeader $task $url $sock]
+ puts $sock "GET $URL($task,$url,path) HTTP/1.0"
+ puts $sock "Host: $URL($task,$url,host)"
+ puts $sock "User-Agent: $agent"
+ if {[string length $acceptLanguage]} {
+ puts $sock "Accept-Language: $acceptLanguage"
+ }
+ puts $sock ""
+ set URL($sock,cancel) [after 30000 [list RobotSockCancel $task $url $sock]]
+ if {[catch {flush $sock}]} {
+ RobotError $task $url 404
+ RobotRestart $task $url $sock
+ }
+}
+
+proc RobotNop {} {
+
+}
+
+proc RobotGetUrl {task url phost} {
+ global URL robotsRunning
+ flush stdout
+ dbgmsg "Retrieve running=$robotsRunning url=$url task=$task"
+ if {![regexp {([^:]+)://([^/]+)(.*)} $url x method hostport path]} {
+ return -1
+ }
+ if {![regexp {([^:]+):([0-9]+)} $hostport x host port]} {
+ set port 80
+ set host $hostport
+ }
+ set URL($task,$url,method) $method
+ set URL($task,$url,host) $host
+ set URL($task,$url,hostport) $hostport
+ set URL($task,$url,path) $path
+ set URL($task,$url,state) head
+ set URL($task,$url,buf) {}
+
+ if {[string compare $path /robots.txt]} {
+ set ok 1
+ if {![info exists URL($hostport,robots)]} {
+ dbgmsg "READING robots.txt for host $hostport"
+ if {[RobotFileExist $task visited $hostport /robots.txt]} {
+ set inf [RobotFileOpen $task visited $hostport /robots.txt r]
+ set buf [read $inf 32768]
+ close $inf
+ } else {
+ set buf "User-agent: *\nAllow: /\n"
+ }
+ RobotsTxt0 $task 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} {
+ dbgmsg "skipped due to robots.txt"
+ return -1
+ }
+ }
+ if [catch {set sock [socket -async $host $port]}] {
+ return -1
+ }
+ RobotConnect $task $url $sock
+
+ return 0
+}
+
+proc loadlib {} {
+ global libdir
+
+ if {![llength [info commands htmlSwitch]]} {
+ if {[info exists env(tclrobot_lib)]} {
+ set d $env(tclrobot_lib)
+ } else {
+ if { $libdir > "" } {
+ set d $libdir
+ } else {
+ set d .
+ }
+ }
+ set e [info sharedlibextension]
+ dbgmsg "About to load $d/tclrobot$e"
+ if {[catch {load $d/tclrobot$e}]} {
+ dbgmsg "Didn't get at $d, trying directly"
+ load tclrobot$e
+ }
+ dbgmsg "Loaded tclrobot$e all right"
+ }
+}
+
+set agent "zmbot/0.2"
+if {![catch {set os [exec uname -s -r]}]} {
+ set agent "$agent ($os)"
+}
+
+dbgmsg "agent: $agent"
+
+proc bgerror {m} {
+ global errorInfo
+ dbgmsg "BGERROR $m"
+ dbgmsg $errorInfo
+}
+
+# Rules: allow, deny, url
+
+proc checkrule {task type this} {
+ global control
+ global debuglevel
+
+ set default_ret 1
+
+ if {$debuglevel > 3} {
+ dbgmsg "CHECKRULE $type $this"
+ }
+ if {[info exist control($task,alrules)]} {
+ foreach l $control($task,alrules) {
+ if {$debuglevel > 3} {
+ dbgmsg "consider $l"
+ }
+ # consider type
+ if {[lindex $l 1] != $type} continue
+ # consider mask (! negates)
+ set masks [lindex $l 2]
+ set ok 0
+ set default_ret 0
+ foreach mask $masks {
+ if {$debuglevel > 4} {
+ dbgmsg "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} {
+ dbgmsg "ok = $ok"
+ }
+ if {!$ok} continue
+ # OK, we have a match
+ if {[lindex $l 0] == "allow"} {
+ if {$debuglevel > 3} {
+ dbgmsg "CHECKRULE MATCH OK"
+ }
+ return 1
+ } else {
+ if {$debuglevel > 3} {
+ dbgmsg "CHECKFULE MATCH FAIL"
+ }
+ return 0
+ }
+ }
+ }
+ if {$debuglevel > 3} {
+ dbgmsg "CHECKRULE MATCH DEFAULT $default_ret"
+ }
+ return $default_ret
+}
+
+
+proc url {href} {
+ global debuglevel task
+
+ if {[RobotHref $task http://www.indexdata.dk/ href host path]} {
+ if {![RobotFileExist $task visited $host $path]} {
+ set outf [RobotFileOpen $task unvisited $host $path]
+ RobotWriteRecord $outf href 0
+ RobotFileClose $outf
+ }
+ }
+}
+
+proc deny {type stuff} {
+ global control task
+
+ lappend control($task,alrules) [list deny $type $stuff]
+}
+
+proc allow {type stuff} {
+ global control task
+
+ lappend control($task,alrules) [list allow $type $stuff]
+}
+
+proc debug {level} {
+ global debuglevel
+
+ set debuglevel $level
+}
+
+proc CreateTask {t} {
+ global tasks task status control
+
+ set task $t
+
+ if {[info exists tasks]} {
+ if {[lsearch -exact $tasks $t] >= 0} {
+ return 0
+ }
+ }
+
+ lappend tasks $t
+ set status($t,unvisited) 0
+ set status($t,visited) 0
+ set status($t,bad) 0
+ set status($t,raw) 0
+ set status($t,active) 1
+ set control($t,seq) 0
+ set control($t,distance) 10
+ set control($t,target) tmp
+ set control($t,output) output
+ set control($t,cr) 0
+ return 1
+}
+
+# Little utility that ensures that at least one task is present (main).
+proc CreateMainTask {} {
+ global tasks
+ if {![info exist tasks]} {
+ CreateTask main
+ }
+}
+
+# Parse options
+
+set i 0
+set l [llength $argv]
+
+if {$l < 1} {
+ puts {tclrobot: usage:}
+ puts {tclrobot [-j jobs] [-p pid] [-T tmpdir] [-o logfile] [-i idle] [-c
+ count] [-d domain] [-D spooldir] [-r rules] [-L libdir] [url ..]}
+ logmsg " Example: -c 3 -d '*.dk' http://www.indexdata.dk/"
+
+ exit 1
+}
+
+
+while {$i < $l} {
+ set arg [lindex $argv $i]
+ switch -glob -- $arg {
+ -o* {
+ set fname [string range $arg 2 end]
+ if {![string length $fname]} {
+ set fname [lindex $argv [incr i]]
+ }
+ set loghandle [open $fname a]
+ #dbgmsg "agent: $agent"
+ #dbgmsg "-o $fname"
+ }
+ -p* {
+ set pidfname [string range $arg 2 end]
+ if {![string length $pidfname]} {
+ set pidfname [lindex $argv [incr i]]
+ }
+ #dbgmsg "-p $pidfname"
+ if {[file exists $pidfname]} {
+ set pf [open $pidfname]
+ gets $pf oldpid
+ close $pf
+ logmsg "File $pidfname already exist. pid=$oldpid"
+ if {[file isdirectory /proc/$oldpid]} {
+ logmsg "And it's apparently running. Exiting."
+ exit 1
+ }
+ }
+ set pf [open $pidfname w]
+ puts $pf [pid]
+ close $pf
+ }
+ -T* {
+ set tmpdir [string range $arg 2 end]
+ if {![string length $tmpdir]} {
+ set tmpdir [lindex $argv [incr i]]
+ }
+ set control(tmpdir) $tmpdir
+ }
+ -L* {
+ set libdir [string range $arg 2 end]
+ if {![string length $libdir]} {
+ set libdir [lindex $argv [incr i]]
+ }
+ }
+ -t* {
+ set t [string range $arg 2 end]
+ if {![string length $t]} {
+ set t [lindex $argv [incr i]]
+ }
+ CreateTask $t
+ }
+ -D* {
+ set dir [string range $arg 2 end]
+ if {![string length $dir]} {
+ set dir [lindex $argv [incr i]]
+ }
+ lappend daemon_dir $dir
+ }
+ -j* {
+ set robotsMax [string range $arg 2 end]
+ if {![string length $robotsMax]} {
+ set robotsMax [lindex $argv [incr i]]
+ }
+ }
+ -c* {
+ CreateMainTask
+ set control($task,distance) [string range $arg 2 end]
+ if {![string length $control($task,distance)]} {
+ set control($task,distance) [lindex $argv [incr i]]
+ }
+ }
+ -d* {
+ CreateMainTask
+ set dom [string range $arg 2 end]
+ if {![string length $dom]} {
+ set dom [lindex $argv [incr i]]
+ }
+ lappend control($task,domains) $dom
+ }
+ -i* {
+ set idletime [string range $arg 2 end]
+ if {![string length $idletime]} {
+ set idletime [lindex $argv [incr i]]
+ }
+ }
+ -l* {
+ CreateMainTask
+ set acceptLanguage [string range $arg 2 end]
+ if {![string length $acceptLanguage]} {
+ set acceptLanguage [lindex $argv [incr i]]
+ }
+ }
+ -r* {
+ CreateMainTask
+ set rfile [string range $arg 2 end]
+ if {![string length $rfile]} {
+ set rfile [lindex $argv [incr i]]
+ }
+ catch {unset maxdistance}
+ source $rfile
+ if {[info exists maxdistance]} {
+ set control($task,distance) $maxdistance
+ }
+ }
+ default {
+ CreateMainTask
+ set href $arg
+ #dbgmsg "in default: arg= $arg !!!"
+ loadlib
+ if {[RobotHref $task http://www.indexdata.dk/ href host path]} {
+ if {![RobotFileExist $task visited $host $path]} {
+ set outf [RobotFileOpen $task unvisited $host $path]
+ RobotWriteRecord $outf href 0
+ RobotFileClose $outf
+ }
+ }
+ }
+ }
+ incr i
+}
+
+
+dbgmsg "Parsed args, now loading"
+loadlib
+
+if {![info exist robotsMax]} {
+ set robotsMax 5
+}
+
+if {[info exist daemon_dir]} {
+ logmsg "Daemon mode"
+ RobotDaemonLoop
+} else {
+ foreach t $tasks {
+ logmsg "task $t"
+ logmsg "max distance=$control($t,distance)"
+ if {[info exists control($t,domains)]} {
+ logmsg "domains=$control($t,domains)"
+ }
+ }
+ logmsg "max jobs=$robotsMax"
+
+ foreach t $tasks {
+ RobotStart $t
+ }
+
+ while {$robotsRunning} {
+ vwait robotsRunning
+ }
+
+ if {[info exists tasks]} {
+ foreach t $tasks {
+ set statusfile [open $t/status w]
+ puts $statusfile "$status($t,unvisited) $status($t,bad) $status($t,visited)"
+ close $statusfile
+ }
+ }
+}
+
projects / tclrobot.git / commitdiff