97 lines
2.6 KiB
C++
97 lines
2.6 KiB
C++
/* Copyright (C) 2013 Phillip Susi
|
|
*
|
|
* 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 2 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 Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#include "../include/PipeCapture.h"
|
|
#include <iostream>
|
|
|
|
namespace GParted {
|
|
|
|
PipeCapture::PipeCapture( int fd, Glib::ustring &string ) : buff( string ), backcount( 0 ), linelength( 0 ),
|
|
sourceid( 0 )
|
|
{
|
|
// tie fd to string
|
|
// make channel
|
|
channel = Glib::IOChannel::create_from_fd( fd );
|
|
}
|
|
|
|
void PipeCapture::connect_signal()
|
|
{
|
|
// connect handler to signal input/output
|
|
sourceid = g_io_add_watch( channel->gobj(),
|
|
GIOCondition(G_IO_IN | G_IO_ERR | G_IO_HUP),
|
|
_OnReadable,
|
|
this );
|
|
}
|
|
|
|
gboolean PipeCapture::_OnReadable( GIOChannel *source,
|
|
GIOCondition condition,
|
|
gpointer data )
|
|
{
|
|
return static_cast<PipeCapture *>(data)->OnReadable( Glib::IOCondition(condition) );
|
|
}
|
|
|
|
|
|
bool PipeCapture::OnReadable( Glib::IOCondition condition )
|
|
{
|
|
// read from pipe and store in buff
|
|
Glib::ustring str;
|
|
Glib::IOStatus status = channel->read( str, 512 );
|
|
if (status == Glib::IO_STATUS_NORMAL)
|
|
{
|
|
for( Glib::ustring::iterator s = str.begin();
|
|
s != str.end(); s++ )
|
|
{
|
|
if( *s == '\b' )
|
|
backcount++;
|
|
else if( *s == '\r' )
|
|
backcount = linelength;
|
|
else if( *s == '\n' ) {
|
|
linelength = 0;
|
|
buff += '\n';
|
|
backcount = 0;
|
|
}
|
|
else if (*s == '\x01' || *s == '\x02' )
|
|
//Skip Ctrl-A and Ctrl-B chars e2fsck uses to bracket the progress bar
|
|
continue;
|
|
else {
|
|
if (backcount) {
|
|
buff.erase( buff.length() - backcount, backcount );
|
|
linelength -= backcount;
|
|
backcount = 0;
|
|
}
|
|
buff += *s;
|
|
++linelength;
|
|
}
|
|
}
|
|
signal_update.emit();
|
|
return true;
|
|
}
|
|
if (status != Glib::IO_STATUS_EOF)
|
|
std::cerr << "Pipe IOChannel read failed" << std::endl;
|
|
// signal completion
|
|
signal_eof.emit();
|
|
return false;
|
|
}
|
|
|
|
PipeCapture::~PipeCapture()
|
|
{
|
|
if( sourceid > 0 )
|
|
g_source_remove( sourceid );
|
|
}
|
|
|
|
} // namespace GParted
|