summaryrefslogtreecommitdiffhomepage
path: root/cv/video-widget.cpp
diff options
context:
space:
mode:
authorStanislaw Halik <sthalik@misaki.pl>2016-10-05 22:50:18 +0200
committerStanislaw Halik <sthalik@misaki.pl>2016-10-05 22:50:54 +0200
commitd8e81c1b24c182047643e8ab25925fc822aee27c (patch)
treee36752edce5263950e14cfb3880b9053e4b58dc2 /cv/video-widget.cpp
parent96cd6bc12cd606154ce69889b94043c0a658fdc3 (diff)
cv/video-widget: fix out-of-bounds writes
Video frames were allocated as single-channel 8-bit but they were meant to be three-channel 8bit for rgb. Reported by: @kzfr Tested in realtime by: @kzfr
Diffstat (limited to 'cv/video-widget.cpp')
-rw-r--r--cv/video-widget.cpp30
1 files changed, 13 insertions, 17 deletions
diff --git a/cv/video-widget.cpp b/cv/video-widget.cpp
index 0a26da39..5d72da7e 100644
--- a/cv/video-widget.cpp
+++ b/cv/video-widget.cpp
@@ -9,12 +9,9 @@
#include "video-widget.hpp"
#include <opencv2/imgproc.hpp>
-#include "api/is-window-visible.hpp"
-
cv_video_widget::cv_video_widget(QWidget* parent) :
QWidget(parent),
- freshp(false),
- visible(true)
+ freshp(false)
{
connect(&timer, SIGNAL(timeout()), this, SLOT(update_and_repaint()));
timer.start(50);
@@ -27,10 +24,7 @@ void cv_video_widget::update_image(const cv::Mat& frame)
if (!freshp)
{
if (_frame.cols != frame.cols || _frame.rows != frame.rows)
- {
- _frame = cv::Mat(frame.rows, frame.cols, CV_8U);
- _frame2 = cv::Mat(frame.rows, frame.cols, CV_8U);
- }
+ _frame = cv::Mat(frame.rows, frame.cols, CV_8UC3);
frame.copyTo(_frame);
freshp = true;
}
@@ -47,17 +41,19 @@ void cv_video_widget::update_and_repaint()
{
QMutexLocker l(&mtx);
+ if (_frame.empty() || !freshp)
+ return;
- if (_frame.empty() || !freshp)
- return;
- cv::cvtColor(_frame, _frame2, cv::COLOR_RGB2BGR);
+ if (_frame2.cols != _frame.cols || _frame2.rows != _frame.rows)
+ _frame2 = cv::Mat(_frame.rows, _frame.cols, CV_8UC3);
- if (_frame3.cols != width() || _frame3.rows != height())
- _frame3 = cv::Mat(height(), width(), CV_8U);
+ if (_frame3.cols != width() || _frame3.rows != height())
+ _frame3 = cv::Mat(height(), width(), CV_8UC3);
- cv::resize(_frame2, _frame3, cv::Size(width(), height()), 0, 0, cv::INTER_NEAREST);
+ cv::cvtColor(_frame, _frame2, cv::COLOR_RGB2BGR);
+ cv::resize(_frame2, _frame3, cv::Size(width(), height()), 0, 0, cv::INTER_NEAREST);
- texture = QImage((const unsigned char*) _frame3.data, _frame3.cols, _frame3.rows, QImage::Format_RGB888);
- freshp = false;
- update();
+ texture = QImage((const unsigned char*) _frame3.data, _frame3.cols, _frame3.rows, QImage::Format_RGB888);
+ freshp = false;
+ update();
}