mirror of
https://github.com/Adaptix-Framework/AdaptixC2
synced 2026-06-08 10:20:39 +00:00
Ralf
192 lines
8.2 KiB
C++
192 lines
8.2 KiB
C++
#include <UI/Widgets/DownloadsWidget.h>
|
|
#include <UI/Widgets/AdaptixWidget.h>
|
|
|
|
DownloadsWidget::DownloadsWidget(QWidget* w)
|
|
{
|
|
this->mainWidget = w;
|
|
this->createUI();
|
|
|
|
connect( tableWidget, &QTableWidget::customContextMenuRequested, this, &DownloadsWidget::handleDownloadsMenu );
|
|
}
|
|
|
|
void DownloadsWidget::createUI()
|
|
{
|
|
tableWidget = new QTableWidget(this );
|
|
tableWidget->setColumnCount(9 );
|
|
tableWidget->setContextMenuPolicy( Qt::CustomContextMenu );
|
|
tableWidget->setAutoFillBackground( false );
|
|
tableWidget->setShowGrid( false );
|
|
tableWidget->setSortingEnabled( true );
|
|
tableWidget->setWordWrap( true );
|
|
tableWidget->setCornerButtonEnabled( true ); // true
|
|
tableWidget->setSelectionBehavior( QAbstractItemView::SelectRows );
|
|
tableWidget->setFocusPolicy( Qt::NoFocus );
|
|
tableWidget->setAlternatingRowColors( true );
|
|
tableWidget->horizontalHeader()->setSectionResizeMode( QHeaderView::Stretch );
|
|
tableWidget->horizontalHeader()->setCascadingSectionResizes( true );
|
|
tableWidget->horizontalHeader()->setHighlightSections( false );
|
|
tableWidget->verticalHeader()->setVisible( false );
|
|
|
|
tableWidget->setHorizontalHeaderItem( 0, new QTableWidgetItem( "File ID" ) );
|
|
tableWidget->setHorizontalHeaderItem( 1, new QTableWidgetItem( "Agent Type" ) );
|
|
tableWidget->setHorizontalHeaderItem( 2, new QTableWidgetItem( "Agent ID" ) );
|
|
tableWidget->setHorizontalHeaderItem( 3, new QTableWidgetItem( "Computer" ) );
|
|
tableWidget->setHorizontalHeaderItem( 4, new QTableWidgetItem( "File" ) );
|
|
tableWidget->setHorizontalHeaderItem( 5, new QTableWidgetItem( "Date" ) );
|
|
tableWidget->setHorizontalHeaderItem( 6, new QTableWidgetItem( "Size" ) );
|
|
tableWidget->setHorizontalHeaderItem( 7, new QTableWidgetItem( "Received" ) );
|
|
tableWidget->setHorizontalHeaderItem( 8, new QTableWidgetItem( "Progress" ) );
|
|
tableWidget->hideColumn( 0 );
|
|
|
|
mainGridLayout = new QGridLayout(this );
|
|
mainGridLayout->setContentsMargins(0, 0, 0, 0 );
|
|
mainGridLayout->addWidget(tableWidget, 0, 0, 1, 4 );
|
|
}
|
|
|
|
DownloadsWidget::~DownloadsWidget() = default;
|
|
|
|
void DownloadsWidget::AddDownloadItem(DownloadData newDownload )
|
|
{
|
|
auto adaptixWidget = qobject_cast<AdaptixWidget*>( mainWidget );
|
|
if(adaptixWidget->Downloads.contains(newDownload.FileId))
|
|
return;
|
|
|
|
auto item_FileID = new QTableWidgetItem( newDownload.FileId );
|
|
auto item_AgentName = new QTableWidgetItem( newDownload.AgentName );
|
|
auto item_AgentID = new QTableWidgetItem( newDownload.AgentId );
|
|
auto item_Computer = new QTableWidgetItem( newDownload.Computer );
|
|
auto item_File = new QTableWidgetItem( newDownload.Filename );
|
|
auto item_Date = new QTableWidgetItem( newDownload.Date );
|
|
auto item_Size = new QTableWidgetItem( BytesToFormat(newDownload.TotalSize) );
|
|
auto item_Received = new QTableWidgetItem( BytesToFormat(newDownload.RecvSize) );
|
|
|
|
item_FileID->setTextAlignment( Qt::AlignCenter );
|
|
item_FileID->setFlags( item_FileID->flags() ^ Qt::ItemIsEditable );
|
|
|
|
item_AgentName->setTextAlignment( Qt::AlignCenter );
|
|
item_AgentName->setFlags( item_AgentName->flags() ^ Qt::ItemIsEditable );
|
|
|
|
item_AgentID->setTextAlignment( Qt::AlignCenter );
|
|
item_AgentID->setFlags( item_AgentID->flags() ^ Qt::ItemIsEditable );
|
|
|
|
item_Computer->setTextAlignment( Qt::AlignCenter );
|
|
item_Computer->setFlags( item_Computer->flags() ^ Qt::ItemIsEditable );
|
|
|
|
item_File->setTextAlignment( Qt::AlignCenter );
|
|
item_File->setFlags( item_File->flags() ^ Qt::ItemIsEditable );
|
|
item_File->setToolTip(item_File->text());
|
|
|
|
item_Date->setTextAlignment( Qt::AlignCenter );
|
|
item_Date->setFlags( item_Date->flags() ^ Qt::ItemIsEditable );
|
|
|
|
item_Size->setTextAlignment( Qt::AlignCenter );
|
|
item_Size->setFlags( item_Size->flags() ^ Qt::ItemIsEditable );
|
|
|
|
item_Received->setTextAlignment( Qt::AlignCenter );
|
|
item_Received->setFlags( item_Received->flags() ^ Qt::ItemIsEditable );
|
|
|
|
QProgressBar *pgbar = new QProgressBar();
|
|
pgbar->setMinimum(0);
|
|
pgbar->setMaximum(newDownload.TotalSize);
|
|
pgbar->setValue(newDownload.RecvSize);
|
|
pgbar->setMinimumSize(QSize(200,24));
|
|
pgbar->setMaximumSize(QSize(200,24));
|
|
|
|
if( tableWidget->rowCount() < 1 )
|
|
tableWidget->setRowCount( 1 );
|
|
else
|
|
tableWidget->setRowCount( tableWidget->rowCount() + 1 );
|
|
|
|
bool isSortingEnabled = tableWidget->isSortingEnabled();
|
|
tableWidget->setSortingEnabled( false );
|
|
tableWidget->setItem( tableWidget->rowCount() - 1, 0, item_FileID );
|
|
tableWidget->setItem( tableWidget->rowCount() - 1, 1, item_AgentName );
|
|
tableWidget->setItem( tableWidget->rowCount() - 1, 2, item_AgentID );
|
|
tableWidget->setItem( tableWidget->rowCount() - 1, 3, item_Computer );
|
|
tableWidget->setItem( tableWidget->rowCount() - 1, 4, item_File );
|
|
tableWidget->setItem( tableWidget->rowCount() - 1, 5, item_Date );
|
|
tableWidget->setItem( tableWidget->rowCount() - 1, 6, item_Size );
|
|
tableWidget->setItem( tableWidget->rowCount() - 1, 7, item_Received );
|
|
tableWidget->setCellWidget( tableWidget->rowCount() - 1, 8, pgbar );
|
|
tableWidget->setSortingEnabled( isSortingEnabled );
|
|
|
|
if( newDownload.State == DOWNLOAD_STATE_STOPPED )
|
|
pgbar->setEnabled(false);
|
|
|
|
tableWidget->horizontalHeader()->setSectionResizeMode( 1, QHeaderView::ResizeToContents );
|
|
tableWidget->horizontalHeader()->setSectionResizeMode( 2, QHeaderView::ResizeToContents );
|
|
tableWidget->horizontalHeader()->setSectionResizeMode( 6, QHeaderView::ResizeToContents );
|
|
tableWidget->horizontalHeader()->setSectionResizeMode( 7, QHeaderView::ResizeToContents );
|
|
tableWidget->horizontalHeader()->setSectionResizeMode( 8, QHeaderView::ResizeToContents );
|
|
|
|
adaptixWidget->Downloads[newDownload.FileId] = newDownload;
|
|
}
|
|
|
|
void DownloadsWidget::EditDownloadItem(QString fileId, int recvSize, int state)
|
|
{
|
|
auto adaptixWidget = qobject_cast<AdaptixWidget*>( mainWidget );
|
|
adaptixWidget->Downloads[fileId].RecvSize = recvSize;
|
|
adaptixWidget->Downloads[fileId].State = state;
|
|
|
|
for (int row = 0; row < tableWidget->rowCount(); ++row) {
|
|
QTableWidgetItem *item = tableWidget->item(row, 0);
|
|
if ( item && item->text() == fileId ) {
|
|
tableWidget->item(row, 7)->setText(BytesToFormat(recvSize) );
|
|
|
|
QProgressBar *pgbar = (QProgressBar*) tableWidget->cellWidget(row,8);
|
|
pgbar->setValue(recvSize);
|
|
|
|
if( state == DOWNLOAD_STATE_STOPPED) {
|
|
pgbar->setEnabled(false);
|
|
}
|
|
else if(state == DOWNLOAD_STATE_RUNNING) {
|
|
pgbar->setEnabled(true);
|
|
}
|
|
else if (state == DOWNLOAD_STATE_FINISHED) {
|
|
adaptixWidget->Downloads[fileId].RecvSize = adaptixWidget->Downloads[fileId].TotalSize;
|
|
|
|
tableWidget->item( row, 7 )->setText("");
|
|
|
|
auto item_Status = new QTableWidgetItem( "Finished" );
|
|
item_Status->setTextAlignment( Qt::AlignCenter );
|
|
tableWidget->removeCellWidget(row,8);
|
|
tableWidget->setItem( row, 8, item_Status );
|
|
tableWidget->item( row, 8 )->setForeground(QColor(COLOR_NeonGreen));
|
|
}
|
|
else if (state == DOWNLOAD_STATE_CANCELED) {
|
|
tableWidget->removeCellWidget(row,8);
|
|
tableWidget->removeRow( row );
|
|
adaptixWidget->Downloads.remove(fileId);
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void DownloadsWidget::handleDownloadsMenu(const QPoint &pos )
|
|
{
|
|
if ( ! tableWidget->itemAt( pos ) )
|
|
return;
|
|
|
|
auto FileID = tableWidget->item( tableWidget->currentRow(), 0 )->text();
|
|
auto Received = tableWidget->item( tableWidget->currentRow(), 7 )->text();
|
|
|
|
auto MenuFile = QMenu();
|
|
if(Received.compare("") == 0) {
|
|
MenuFile.addAction( "Sync file to client" );
|
|
MenuFile.addAction( "Delete file" );
|
|
}
|
|
else {
|
|
if( tableWidget->cellWidget( tableWidget->currentRow(), 8)->isEnabled() )
|
|
MenuFile.addAction( "Stop" );
|
|
else
|
|
MenuFile.addAction( "Start" );
|
|
MenuFile.addAction( "Cancel" );
|
|
}
|
|
|
|
auto *action = MenuFile.exec( tableWidget->horizontalHeader()->viewport()->mapToGlobal( pos ) );
|
|
}
|
|
|
|
|