drygdryg/zapret
1
0
Fork 0
mirror of https://github.com/bol-van/zapret.git synced 2024-11-29 21:40:52 +03:00
Code Issues Packages Projects Releases Wiki Activity

mdig: fix text mode std io in windows

Browse Source
This commit is contained in:
bol-van 2024-11-09 23:33:33 +03:00
parent 07b8567beb
commit 591b246ed6
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mdig/mdig.c
View File

@ -12,7 +12,6 @@
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <stdbool.h> #include <stdbool.h>
#include <unistd.h>
#include <pthread.h> #include <pthread.h>
#include <getopt.h> #include <getopt.h>
#ifdef _WIN32 #ifdef _WIN32
@ -21,7 +20,9 @@
#include <winsock2.h> #include <winsock2.h>
#include <ws2ipdef.h> #include <ws2ipdef.h>
#include <ws2tcpip.h> #include <ws2tcpip.h>
#include <fcntl.h>
#else #else
#include <unistd.h>
#include <sys/socket.h> #include <sys/socket.h>
#include <arpa/inet.h> #include <arpa/inet.h>
#include <netinet/in.h> #include <netinet/in.h>
@ -364,6 +365,9 @@ int dns_make_query(const char *dom, char family)
fprintf(stderr, "could not make DNS query\n"); fprintf(stderr, "could not make DNS query\n");
return 1; return 1;
} }
#ifdef _WIN32
_setmode(_fileno(stdout), _O_BINARY);
#endif
if (fwrite(q,l,1,stdout)!=1) if (fwrite(q,l,1,stdout)!=1)
{ {
fprintf(stderr, "could not write DNS query blob to stdout\n"); fprintf(stderr, "could not write DNS query blob to stdout\n");
@ -422,6 +426,9 @@ int dns_parse_query()
{ {
uint8_t a[1500]; uint8_t a[1500];
size_t l; size_t l;
#ifdef _WIN32
_setmode(_fileno(stdin), _O_BINARY);
#endif
l = fread(a,1,sizeof(a),stdin); l = fread(a,1,sizeof(a),stdin);
if (!l || !feof(stdin)) if (!l || !feof(stdin))
{ {