1
0
mirror of https://github.com/j-a-n/raspberrymatic-addon-rmupdate.git synced 2023-10-10 13:37:40 +02:00
raspberrymatic-addon-rmupdate/addon/www/rest.cgi
2020-01-06 21:12:54 +01:00

251 lines
7.9 KiB
Tcl

#!/bin/tclsh
# RaspMatic update addon
#
# Copyright (C) 2019 Jan Schneider <oss@janschneider.net>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#lappend auto_path /www
#set env(TCLLIBPATH) [list /www /usr/local/addons/rmupdate/lib]
#source once.tcl
#source session.tcl
load tclrega.so
source /usr/local/addons/rmupdate/lib/rmupdate.tcl
package require http
proc get_http_header {request header_name} {
upvar $request req
set header_name [string toupper $header_name]
array set meta $req(meta)
foreach header [array names meta] {
if {$header_name == [string toupper $header] } then {
return $meta($header)
}
}
return ""
}
proc get_session {sid} {
if {[regexp {@([0-9a-zA-Z]{10})@} $sid match sidnr]} {
return [lindex [rega_script "Write(system.GetSessionVarStr('$sidnr'));"] 1]
}
return ""
}
proc check_session {sid} {
if {[get_session $sid] != ""} {
# check and renew session
set url "http://127.0.0.1/pages/index.htm?sid=$sid"
set request [::http::geturl $url]
set code [::http::code $request]
::http::cleanup $request
if {[lindex $code 1] == 200} {
return 1
}
}
return 0
}
proc login {username password} {
set request [::http::geturl "http://127.0.0.1/login.htm" -query [::http::formatQuery tbUsername $username tbPassword $password]]
set code [::http::code $request]
set location [get_http_header $request "location"]
::http::cleanup $request
if {[string first "error" $location] != -1} {
error "Invalid username oder password" "Unauthorized" 401
}
if {![regexp {sid=@([0-9a-zA-Z]{10})@} $location match sid]} {
error "Too many sessions" "Service Unavailable" 503
}
return $sid
}
proc process {} {
global env
if { [info exists env(QUERY_STRING)] } {
set query $env(QUERY_STRING)
set sid ""
set pairs [split $query "&"]
foreach pair $pairs {
if {[regexp "^(\[^=\]+)=(.*)$" $pair match varname value]} {
if {$varname == "sid"} {
set sid $value
} elseif {$varname == "path"} {
set path [split $value "/"]
}
}
}
set plen [expr [llength $path] - 1]
if {[lindex $path 1] == "login"} {
set data [read stdin $env(CONTENT_LENGTH)]
regexp {\"username\"\s*:\s*\"([^\"]*)\"} $data match username
regexp {\"password\"\s*:\s*\"([^\"]*)\"} $data match password
set sid [login $username $password]
return "\"${sid}\""
}
if {![check_session $sid]} {
error "Invalid session" "Unauthorized" 401
}
if {[lindex $path 1] == "install_addon_archive"} {
set archive_file "/tmp/uploaded_addon.tar.gz"
catch {fconfigure stdin -translation binary}
catch {fconfigure stdin -encoding binary}
set out [open $archive_file w]
catch {fconfigure $out -translation binary}
catch {fconfigure $out -encoding binary}
puts -nonewline $out [read stdin]
close $out
set res [rmupdate::install_addon "" "file://${archive_file}"]
return "\"${res}\""
}
set data ""
if { [info exists env(CONTENT_LENGTH)] } {
set data [read stdin $env(CONTENT_LENGTH)]
}
if {[lindex $path 1] == "get_session"} {
return "\"[get_session $sid]\""
} elseif {[lindex $path 1] == "version"} {
return "\"[rmupdate::version]\""
} elseif {[lindex $path 1] == "get_firmware_info"} {
return [rmupdate::get_firmware_info]
} elseif {[lindex $path 1] == "get_system_info"} {
set system_type [rmupdate::get_rpi_version]
set uptime [exec /usr/bin/uptime]
return "\{\"system_type\":\"${system_type}\",\"uptime\":\"${uptime}\"\}"
} elseif {[lindex $path 1] == "get_partitions"} {
return [array_to_json [rmupdate::get_partitions]]
} elseif {[lindex $path 1] == "move_userfs_to_device"} {
regexp {\"target_device\"\s*:\s*\"([^\"]+)\"} $data match target_device
return [rmupdate::move_userfs_to_device $target_device 1 1]
} elseif {[lindex $path 1] == "delete_partition_table"} {
regexp {\"device\"\s*:\s*\"([^\"]+)\"} $data match device
rmupdate::delete_partition_table $device
return "\"deleted\""
} elseif {[lindex $path 1] == "system_reboot"} {
exec /sbin/reboot
return "\"reboot initiated\""
} elseif {[lindex $path 1] == "system_shutdown"} {
exec /sbin/poweroff
return "\"shutdown initiated\""
} elseif {[lindex $path 1] == "get_addon_info"} {
return [rmupdate::get_addon_info 1 1 1]
} elseif {[lindex $path 1] == "start_install_firmware"} {
regexp {\"download_url\"\s*:\s*\"([^\"]*)\"} $data match download_url
regexp {\"version\"\s*:\s*\"([\d\.]*)\"} $data match version
regexp {\"language\"\s*:\s*\"([^\"]+)\"} $data match lang
regexp {\"reboot\"\s*:\s*(true|false)} $data match reboot
regexp {\"dryrun\"\s*:\s*(true|false)} $data match dryrun
regexp {\"keep_download\"\s*:\s*(true|false)} $data match keep_download
if { ![info exists reboot] } {
set reboot "true"
}
if {$reboot == "true"} {
set reboot 1
} else {
set reboot 0
}
if { ![info exists dryrun] } {
set dryrun "false"
}
if {$dryrun == "true"} {
set dryrun 1
} else {
set dryrun 0
}
if { ![info exists keep_download] } {
set keep_download "false"
}
if {$keep_download == "true"} {
set keep_download 1
} else {
set keep_download 0
}
return "\"[rmupdate::install_firmware $download_url $version $lang $reboot $keep_download $dryrun]\""
} elseif {[lindex $path 1] == "install_addon"} {
regexp {\"addon_id\"\s*:\s*\"([^\"]+)\"} $data match addon_id
if { ![info exists addon_id] } {
set addon_id ""
}
regexp {\"download_url\"\s*:\s*\"([^\"]+)\"} $data match download_url
if { ![info exists download_url] } {
set download_url ""
}
return "\"[rmupdate::install_addon $addon_id $download_url]\""
} elseif {[lindex $path 1] == "uninstall_addon"} {
regexp {\"addon_id\"\s*:\s*\"([^\"]+)\"} $data match addon_id
if { [info exists addon_id] && $addon_id != "" } {
return "\"[rmupdate::uninstall_addon $addon_id]\""
} else {
error "Invalid addon_id: ${addon_id}"
}
} elseif {[lindex $path 1] == "delete_firmware_image"} {
regexp {\"version\"\s*:\s*\"([\d\.]+)\"} $data match version
if { [info exists version] && $version != "" } {
return "\"[rmupdate::delete_firmware_image $version]\""
} else {
error "Invalid version: ${data}"
}
} elseif {[lindex $path 1] == "is_system_upgradeable"} {
if {[rmupdate::is_system_upgradeable]} {
return "true"
} else {
return "false"
}
} elseif {[lindex $path 1] == "get_running_installation"} {
return "\"[rmupdate::get_running_installation]\""
} elseif {[lindex $path 1] == "read_install_log"} {
variable content_type "text/html"
return [rmupdate::read_install_log]
} elseif {[lindex $path 1] == "wlan_scan"} {
return [rmupdate::wlan_scan 1]
} elseif {[lindex $path 1] == "wlan_connect"} {
regexp {\"ssid\"\s*:\s*\"([^\"]+)\"} $data match ssid
set password ""
regexp {\"password\"\s*:\s*\"([^\"]+)\"} $data match password
return [rmupdate::wlan_connect $ssid $password]
} elseif {[lindex $path 1] == "wlan_disconnect"} {
return [rmupdate::wlan_disconnect]
}
}
error "Path not found" "Not found" 404
}
variable content_type "application/json"
if [catch {process} result] {
set status 500
if { [regexp {^\d+$} $errorCode ] } {
set status $errorCode
}
puts "Content-Type: ${content_type}"
puts "Status: $status";
puts ""
set result [json_string $result]
puts -nonewline "\{\"error\":\"${result}\"\}"
} else {
puts "Content-Type: ${content_type}"
puts "Status: 200 OK";
puts ""
puts -nonewline $result
}