- www stuff auths through ftpiod (daemon just has stub functions, but adding

[projects/pld-ftp-admin.git] / modules / cmds.py

# vi: encoding=utf-8 ts=8 sts=4 sw=4 et

import os
import time

BailOut="BailOut"

def parse(con):
    if '\0' not in con.data:
        return
    cmds=con.data.split('\0')[:-1]

    for cmd in cmds:
        con.data=con.data[len(cmd)+1:]
        cmdname=cmd[:4]
        if not con.authorized and not (cmdname=='linp' or cmdname=='linc'):
            raise BailOut
            # TODO: log unauthorized access
        if cmdname in cmdlist_noargs:
            if len(cmd)==4:
                cmdlist_noargs[cmdname](con)
            else:
                pass
                # TODO: log malicious msg
        elif cmdname in cmdlist_args:
            if len(cmd)>5:
                cmdlist_args[cmdname](con, cmd[5:])
            else:
                pass
                # TODO: log malicious msg
        else:
            raise BailOut
            # TODO: log this

locks={}

def lock(con, arg, hard):
    if arg not in locks:
        locks[arg]=hard
        con.sock.send("OK")
    elif locks[arg]:
        con.sock.send("HARD") # Hard lock - you can go get a cup of tea
    else:
        con.sock.send("SOFT") # Soft lock - try in a second or two
        
    
def cmd_unlock(con, arg):
    if arg in locks:
        del locks[arg]

def cmd_lock_soft(con, arg):
    lock(con, arg, False)

def cmd_lock_hard(con, arg):
    lock(con, arg, True)

logfile=open(os.environ['HOME']+'/pld-ftp-admin/var/log', 'a')

def cmd_log(con, msg):
    logfile.write('%s [%s] -- %s\n' % (time.strftime('%Y-%m-%d %H:%M:%S'), 
                                       con.name, msg))
    logfile.flush()

def cmd_name(con, name):
    con.name=name

def cmd_login_passwd(con, data):
    tmp=data.split('\n')
    if len(tmp)!=2:
        raise BailOut
    login=tmp[0]
    passwd=tmp[1]
    if login=='mmazur' and passwd=='ziew':
        con.sock.send('OK ciasteczko')
    else:
        # TODO: log this
        con.sock.send('FAIL')
        raise BailOut

def cmd_login_cookie(con, cookie):
    if cookie=='ciasteczko':
        con.sock.send('OK mmazur')
    else:
        # TODO: log this (or not)
        con.sock.send('FAIL')

def cmd_logout(con):
    # TODO: remove cookie
    pass

cmdlist_args={'lcks':cmd_lock_soft, 'lckh':cmd_lock_hard, 'ulck':cmd_unlock,
         'log1':cmd_log, 'name':cmd_name, 'linp':cmd_login_passwd,
         'linc':cmd_login_cookie}

cmdlist_noargs={'lout':cmd_logout}