summaryrefslogtreecommitdiffstats
path: root/src/scanmonitor.cpp
diff options
context:
space:
mode:
authorTimothy Pearson <kb9vqf@pearsoncomputing.net>2013-07-24 11:29:03 -0500
committerTimothy Pearson <kb9vqf@pearsoncomputing.net>2013-07-24 11:29:03 -0500
commit066bae76e94c21604fe4132c4ca26e5b2f0c6375 (patch)
tree64cd6638d6f513ef8ddb094f9437467c0a810d86 /src/scanmonitor.cpp
downloadknmap-066bae76e94c21604fe4132c4ca26e5b2f0c6375.tar.gz
knmap-066bae76e94c21604fe4132c4ca26e5b2f0c6375.zip
Initial import of knmap 2.1 sources
Diffstat (limited to 'src/scanmonitor.cpp')
-rw-r--r--src/scanmonitor.cpp139
1 files changed, 139 insertions, 0 deletions
diff --git a/src/scanmonitor.cpp b/src/scanmonitor.cpp
new file mode 100644
index 0000000..413dd93
--- /dev/null
+++ b/src/scanmonitor.cpp
@@ -0,0 +1,139 @@
+/***************************************************************************
+ * *
+ * Copyright (C) 2005 by Kevin Gilbert *
+ * kev.gilbert@cdu.edu.au *
+ * *
+ * 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 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 <errno.h>
+#include <stdio.h>
+#include <string.h>
+#include <sys/select.h>
+#include <sys/time.h>
+
+#include <qapplication.h>
+
+#include <klocale.h>
+
+#include "nmapoutputbuffer.h"
+#include "scanmonitor.h"
+#include "scanmonitorevent.h"
+#include "scanwidget.h"
+
+// constructor
+// ===========
+
+ScanMonitor::ScanMonitor( ScanWidget* parent )
+: QThread( ),
+ m_buffer( NULL ),
+ m_nmapBufferSize( 16384 ),
+ m_parent( parent )
+{ m_buffers.setAutoDelete( true );
+}
+
+// getBuffer
+// =========
+
+void ScanMonitor::getBuffer( )
+{ for( m_buffer = m_buffers.first( ); m_buffer != NULL; m_buffer = m_buffers.next( ))
+ if( m_buffer->isFree( ))
+ { m_buffer->setBusy( );
+ m_buffers.take( );
+ m_buffers.append( m_buffer );
+ return;
+ }
+
+ m_buffer = new NmapOutputBuffer( m_nmapBufferSize );
+ m_buffers.append( m_buffer );
+}
+
+// readNmapOutput
+// ==============
+
+void ScanMonitor::readNmapOutput( const int pipeFD, const ScanMonitorEventType type )
+{ while( true )
+ { if( m_buffer == NULL )
+ getBuffer( );
+ else
+ m_buffer->setBusy( );
+
+ int bytesRead = ::read( pipeFD, m_buffer->buffer( ), m_nmapBufferSize );
+
+ if( bytesRead > 0 )
+ { m_buffer->length( bytesRead );
+ QApplication::postEvent( m_parent, new ScanMonitorEvent( type, m_buffer ));
+ m_buffer = NULL;
+ continue;
+ }
+
+ if( !bytesRead || errno == EAGAIN || errno == EBADF )
+ { m_buffer->setFree( );
+ break;
+ }
+
+ QString text = QString( i18n( "error reading nmap output from pipe: %2\n" )).arg( strerror( errno ));
+ strcpy( m_buffer->buffer( ), text.ascii( ));
+ m_buffer->length( text.length( ));
+ QApplication::postEvent( m_parent, new ScanMonitorEvent( StderrType, m_buffer ));
+ m_buffer = NULL;
+ break;
+ }
+}
+
+// run
+// ===
+
+void ScanMonitor::run( )
+{ fd_set efds;
+ fd_set rfds;
+ struct timeval timeout;
+
+ while( true )
+ { FD_ZERO( &efds );
+ FD_ZERO( &rfds );
+
+ FD_SET( m_pipeStderrFD, &efds);
+ FD_SET( m_pipeStdoutFD, &efds);
+
+ FD_SET( m_pipeStderrFD, &rfds);
+ FD_SET( m_pipeStdoutFD, &rfds);
+
+ timeout.tv_sec = 2;
+ timeout.tv_usec = 0;
+
+ if( !select( _max_( m_pipeStderrFD, m_pipeStdoutFD ) + 1, &rfds, NULL, &efds, &timeout ))
+ continue;
+
+ if( FD_ISSET( m_pipeStderrFD, &efds ) || FD_ISSET( m_pipeStdoutFD, &efds ))
+ break;
+
+ if( FD_ISSET( m_pipeStderrFD, &rfds ))
+ readNmapOutput( m_pipeStderrFD, StderrType );
+
+ if( FD_ISSET( m_pipeStdoutFD, &rfds ))
+ readNmapOutput( m_pipeStdoutFD, StdoutType );
+ }
+}
+
+// setPipeFDs
+// ==========
+
+void ScanMonitor::setPipeFDs( const int pipeStderrFD, const int pipeStdoutFD )
+{ m_pipeStderrFD = pipeStderrFD;
+ m_pipeStdoutFD = pipeStdoutFD;
+}