2009-08-07 13 views
13

Mi piacerebbe fare l'equivalente di:Come si costruisce una pipa di lettura-scrittura con lua?

foo=$(echo "$foo"|someprogram) 

entro lua - vale a dire, ho una variabile che contiene un po 'di testo e mi piacerebbe correre attraverso un filtro (implementato in python come succede).

Eventuali suggerimenti?

Aggiunto: vorrebbe veramente fare questo senza utilizzare un file temporaneo

risposta

3

Aha, una soluzione forse meglio:

require('posix') 
require('os') 
require('io') 

function splat_popen(data,cmd) 
    rd,wr = posix.pipe() 
    io.flush() 
    child = posix.fork() 
    if child == 0 then 
     rd:close() 
     wr:write(data) 
     io.flush() 
     os.exit(1) 
    end 
    wr:close() 

    rd2,wr2 = posix.pipe() 
    io.flush() 
    child2 = posix.fork() 
    if child2 == 0 then 
     rd2:close() 
     posix.dup(rd,io.stdin) 
     posix.dup(wr2,io.stdout) 
     posix.exec(cmd) 
     os.exit(2) 
    end 
    wr2:close() 
    rd:close() 

    y = rd2:read("*a") 
    rd2:close() 

    posix.wait(child2) 
    posix.wait(child) 

    return y 
end 

munged=splat_popen("hello, world","/usr/games/rot13") 
print("munged: "..munged.." !") 
+0

perché ricevo un errore riguardo al 'require'? "variabile globale inesistente 'require'" –

4

Non c'è nulla nella libreria standard Lua per permettere questo.

Here is an in-depth exploration of the difficulties of doing bidirectional communication properly, e una soluzione proposta:

se possibile, reindirizzare un'estremità del flusso (ingresso o uscita) in un file. Cioè .:

fp = io.popen("foo >/tmp/unique", "w") 
fp:write(anything) 
fp:close() 
fp = io.open("/tmp/unique") 
x = read("*a") 
fp:close() 

Si potrebbe interessare this extension che aggiunge funzioni ai os e io namespace per rendere la comunicazione bidirezionale con un sottoprocesso possibile.

+0

C'è qualche modo per creare un sottoprocesso lua (fork() o simili) per passare i dati al filtro, come il guscio fa? Ciò eviterebbe lo stallo ... –

-1

A non molto bella soluzione che evita un file temporaneo ...

require("io") 
require("posix") 

x="hello\nworld" 

posix.setenv("LUA_X",x) 
i=popen('echo "$LUA_X" | myfilter') 
x=i.read("*a") 
3

Finché il Lua supporta io.popen, questo problema è facile . La soluzione è esattamente come avete descritto, tranne che invece di $(...) avete bisogno di una funzione come questa:

function os.capture(cmd, raw) 
    local f = assert(io.popen(cmd, 'r')) 
    local s = assert(f:read('*a')) 
    f:close() 
    if raw then return s end 
    s = string.gsub(s, '^%s+', '') 
    s = string.gsub(s, '%s+$', '') 
    s = string.gsub(s, '[\n\r]+', ' ') 
    return s 
end 

È quindi possibile chiamare

local foo = ... 
local cmd = ("echo $foo | someprogram"):gsub('$foo', foo) 
foo = os.capture(cmd) 

faccio cose come questa tutto il tempo. Ecco una correlata funzione utile per i comandi di formatura:

local quote_me = '[^%w%+%-%=%@%_%/]' -- complement (needn't quote) 
local strfind = string.find 

function os.quote(s) 
    if strfind(s, quote_me) or s == '' then 
    return "'" .. string.gsub(s, "'", [['"'"']]) .. "'" 
    else 
    return s 
    end 
end 
+0

Questo non cattura lo stderr. C'è un modo per catturarlo anche tu? –

+0

@Vilas, se non ti dispiace mescolare stderr e stdout, aggiungi '2> & 1' alla riga di comando. Oppure guarda lo script Debian 'annotate-output'. –

+0

Stai molto attento con questo approccio --- finisci per passare l'intero contenuto di 'foo' sulla riga di comando, e non puoi garantire quanto sia grande la linea di comando. Anche i tuoi dati saranno pubblici, poiché appariranno nell'elenco dei processi. –

0

Ecco come ho risolto il problema, esso richiede lua posix.

  p = require 'posix' 
      local r,w = p.pipe() 
      local r1,w1 = p.pipe() 
      local cpid = p.fork() 
      if cpid == 0 then -- child reads from pipe          
      w:close() 
      r1:close() 
      p.dup(r, io.stdin) 
      p.dup(w1 ,io.stdout) 
      p.exec('./myProgram') 
      r:close() 
      w1:close() 
      p._exit(0) 
      else -- parent writes to pipe             
      IN = r1 
      OUT = w 
      end 

Durante myProgram esecuzione, you'l leggere e scrivere da io normale e dopo questa parte del codice che basta scrivere/leggere IN e OUT per comunicare con il programma del bambino.

2

Mi sono imbattuto in questo post cercando di fare la stessa cosa e non ho mai trovato una buona soluzione, vedere il codice qui sotto per come ho risolto i miei problemi. Questa implementazione consente agli utenti di accedere a stdin, stdout, stderr e ottenere il codice di stato di ritorno. Un semplice wrapper è chiamato per chiamate pipe semplici.

require("posix") 

-- 
-- Simple popen3() implementation 
-- 
function popen3(path, ...) 
    local r1, w1 = posix.pipe() 
    local r2, w2 = posix.pipe() 
    local r3, w3 = posix.pipe() 

    assert((w1 ~= nil or r2 ~= nil or r3 ~= nil), "pipe() failed") 

    local pid, err = posix.fork() 
    assert(pid ~= nil, "fork() failed") 
    if pid == 0 then 
     posix.close(w1) 
     posix.close(r2) 
     posix.dup2(r1, posix.fileno(io.stdin)) 
     posix.dup2(w2, posix.fileno(io.stdout)) 
     posix.dup2(w3, posix.fileno(io.stderr)) 
     posix.close(r1) 
     posix.close(w2) 
     posix.close(w3) 

     local ret, err = posix.execp(path, unpack({...})) 
     assert(ret ~= nil, "execp() failed") 

     posix._exit(1) 
     return 
    end 

    posix.close(r1) 
    posix.close(w2) 
    posix.close(w3) 

    return pid, w1, r2, r3 
end 

-- 
-- Pipe input into cmd + optional arguments and wait for completion 
-- and then return status code, stdout and stderr from cmd. 
-- 
function pipe_simple(input, cmd, ...) 
    -- 
    -- Launch child process 
    -- 
    local pid, w, r, e = popen3(cmd, unpack({...})) 
    assert(pid ~= nil, "filter() unable to popen3()") 

    -- 
    -- Write to popen3's stdin, important to close it as some (most?) proccess 
    -- block until the stdin pipe is closed 
    -- 
    posix.write(w, input) 
    posix.close(w) 

    local bufsize = 4096 
    -- 
    -- Read popen3's stdout via Posix file handle 
    -- 
    local stdout = {} 
    local i = 1 
    while true do 
     buf = posix.read(r, bufsize) 
     if buf == nil or #buf == 0 then break end 
     stdout[i] = buf 
     i = i + 1 
    end 

    -- 
    -- Read popen3's stderr via Posix file handle 
    -- 
    local stderr = {} 
    local i = 1 
    while true do 
     buf = posix.read(e, bufsize) 
     if buf == nil or #buf == 0 then break end 
     stderr[i] = buf 
     i = i + 1 
    end 

    -- 
    -- Clean-up child (no zombies) and get return status 
    -- 
    local wait_pid, wait_cause, wait_status = posix.wait(pid) 

    return wait_status, table.concat(stdout), table.concat(stderr) 
end 

-- 
-- Example usage 
-- 
local my_in = io.stdin:read("*all") 
--local my_cmd = "wc" 
--local my_args = {"-l"} 
local my_cmd = "spamc" 
local my_args = {} -- no arguments 
local my_status, my_out, my_err = pipe_simple(my_in, my_cmd, unpack(my_args)) 

-- Obviously not interleaved as they would have been if printed in realtime 
io.stdout:write(my_out) 
io.stderr:write(my_err) 

os.exit(my_status) 
+0

Questo è abbastanza bello ma sei sicuro che questo funziona come presentato? I numeri dei descrittori dei file sono off-by-one (dovrebbero essere 0,1 e 2) e i test per gli errori fork e execp sembrano invertiti. – jpc

+0

funziona per lo più come scritto per me, ha dovuto aggiungere 'posix local = require ("posix")'. – robm

+0

Ho dovuto aggiungere 'posix.close (r)' e 'posix.close (e)' dopo i loop per scaricarli in ps.pipe_simple(). In caso contrario ha ottenuto "troppi file aperti" dopo 500 chiamate a questo su linux. – robm

0

È semplice, non sono necessarie estensioni (testato con lua 5.3).

#!/usr/bin/lua 
-- use always locals 
local stdin = io.stdin:lines() 
local stdout = io.write 

for line in stdin do 
    stdout (line) 
end 

Salva con nome inout.lua e fare chmod +x /tmp/inout.lua

20:30 $ foo=$(echo "bla"| /tmp/inout.lua) 
20:30 $ echo $foo 
bla 
Problemi correlati