使用OpenCV与UDP完成的局域网双向实时视频通信

OpenCV与UDP实现的双向视频通信

基本内容

实现了局域网内双向视频通信，也就是视频通话中视频部分。主要调用OpenCV与Socket实现。没有流量控制，没有内网穿透，没有重发机制等等。
使用方法：通信两端同时运行该程序并且输入对方IP后，即可实现视频通信。

依赖项

主要调用的有OpenCV库与Socket编程。使用VS2019编程环境，C与C++混杂。其中，OpenCV的配置比较复杂，可以参考这篇文章 https://blog.csdn.net/mars_xiaolei/article/details/78759041 ；Socket无需配置。

代码及注释

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <winsock2.h>
#include <vector>
#include <iostream>
#include <process.h>
#include <windows.h>
#include <opencv2/opencv.hpp>  
#include <opencv2/imgproc/imgproc.hpp> 
#include "opencv2/imgcodecs/legacy/constants_c.h"

#pragma comment(lib,"ws2_32.lib")

#define PORT 9190//端口号
#define BUF_SIZE 65536//缓冲区大小
#define COMPRESSION 50//图片压缩率

using namespace cv;

void ErrorHandling(const char* message);//报错用函数
void sendFunc(void);//发送图片函数
unsigned int __stdcall WINAPI recvThreadFunc(void*);//接收图片并显示函数

SOCKADDR_IN servAdr, clntAdr;//ser为自身，cln为对方
SOCKET servSock;

int main(int argc, char* argv[])
{
	WSADATA wsaData;
	char ipStr[15];
	HANDLE hTread;
	unsigned threadID;
	int param = 5;
	fgets(ipStr, 14, stdin);
	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
		ErrorHandling("WSAStartup() error!");
	servSock = socket(PF_INET, SOCK_DGRAM, 0);
	if (servSock == INVALID_SOCKET)
		ErrorHandling("UDP socket creation error");
	memset(&servAdr, 0, sizeof(servAdr));
	servAdr.sin_family = AF_INET;
	servAdr.sin_addr.s_addr = htonl(INADDR_ANY);
	servAdr.sin_port = htons(PORT);
	memset(&clntAdr, 0, sizeof(clntAdr));
	clntAdr.sin_family = AF_INET;
	clntAdr.sin_addr.s_addr = inet_addr(ipStr);
	clntAdr.sin_port = htons(PORT);
	hTread = (HANDLE)_beginthreadex(NULL, 0, recvThreadFunc, (void*)&param, 0,&threadID);//创建线程，接收
	if (hTread == 0)
		ErrorHandling("begin thread error");
	sendFunc();//发送

	closesocket(servSock);
	WSACleanup();
	return 0;
}

void ErrorHandling(const char* message)
{
	fputs(message, stderr);
	fputc('\n', stderr);
	exit(1);
}


unsigned int __stdcall WINAPI recvThreadFunc(void*)
{
	int ii = 0;
	Mat frame;
	char message[BUF_SIZE];
	int clntAdrSz;
	int strLen;
	clntAdrSz = sizeof(clntAdr);
	if (bind(servSock, (SOCKADDR*)&servAdr, sizeof(servAdr)) == SOCKET_ERROR)
		ErrorHandling("bind() error");
	while (1)
	{
		std::vector<uchar> data_decode;
		strLen = recvfrom(servSock, message, BUF_SIZE, 0, (SOCKADDR*)&clntAdr, &clntAdrSz);
		printf("%d\n", ++ii);
		for (int i = 0;i < strLen;i++)
		{
			data_decode.push_back(message[i]);
		}
		frame = imdecode(data_decode, CV_LOAD_IMAGE_COLOR);
		cv::imshow("对方", frame);
		cv::waitKey(30);
		memset(message, 0, sizeof(message));
	}
	return 0;
}

void sendFunc(void)
{
	VideoCapture capture(0);
	Mat frame;
	std::vector<int> quality_setting;
	quality_setting.push_back(IMWRITE_JPEG_QUALITY);
	quality_setting.push_back(COMPRESSION);
	int ii = 0;
	connect(servSock, (SOCKADDR*)&clntAdr, sizeof(clntAdr));//UDP连接
	while (1)
	{
		capture >> frame;
		if (frame.empty())
			ErrorHandling("frame empty");
		std::vector<uchar> data_encode;
		imencode(".jpg", frame, data_encode, quality_setting);
		int size = data_encode.size();
		char message[65535];
		for (int i = 0;i < size;i++)
		{
			message[i] = data_encode[i];
		}
		sendto(servSock, message, size, 0, (const sockaddr*)&clntAdr, sizeof(clntAdr));
		printf("%d\n", ++ii);
		//memset(&data_encode, 0, sizeof(data_encode));
	}
}

几个小问题

1. 如果编译报错有关OpenCV，参考上面提到的配置方法。
2. 如果inet_addr()函数报错，可以选择换成编译器推荐的函数，或者这样做：项目–>属性–>C/C+±->常规–>SDL检查–>否。
3. 如果程序运行成功且打开了摄像头但没有显示，考虑是否在同一局域网，或所在局域网是否为了安全将用户隔离（有些校园网会有），可以使用手机WIFI再次尝试。
4. 每次切换WIFI后，自己的内网IP都会改变。