รับสตริงสุดท้ายที่พิมพ์ด้วย C/C++

เปลี่ยนเส้นทางเอาต์พุตมาตรฐานไปยังบัฟเฟอร์

ถ่ายทอดสดทาง Coliru

#include <stdio.h>
#include <unistd.h>

#define BUFFER_SIZE 1024
int stdoutSave;
char outputBuffer[BUFFER_SIZE];

void replaceStdout()
{
    fflush(stdout); //clean everything first
    stdoutSave = dup(STDOUT_FILENO); //save the stdout state
    freopen("NUL", "a", stdout); //redirect stdout to null pointer
    setvbuf(stdout, outputBuffer, _IOFBF, 1024); //set buffer to stdout
}

void restoreStdout()
{
    freopen("NUL", "a", stdout); //redirect stdout to null again
    dup2(stdoutSave, STDOUT_FILENO); //restore the previous state of stdout
    setvbuf(stdout, NULL, _IONBF, BUFFER_SIZE); //disable buffer to print to screen instantly
}

void printHelloWorld()
{
    printf("hello\n");
    printf("world");
}

int main()
{
    replaceStdout();
    printHelloWorld();
    restoreStdout();
    // Use outputBuffer to test EXPECT_EQ("somthing1", outputBuffer);
    printf("Fetched output: (%s)", outputBuffer);
    return 0;
}

ข้อมูลอ้างอิง: http://kaskavalci.com/redirecting-stdout-to-array-and-restoring-it-back-in-c/

ฉันไม่รู้ว่าเป็นไปได้หรือไม่ที่จะได้สิ่งที่พิมพ์ครั้งล่าสุด แต่ถ้าคุณควบคุมสภาพแวดล้อม ก่อน ฟังก์ชันทดสอบของคุณจะถูกเรียกใช้ คุณสามารถเปลี่ยนเส้นทางที่เอาต์พุตมาตรฐานไป ซึ่งช่วยให้คุณเขียนมันลงใน ไฟล์ซึ่งคุณสามารถตรวจสอบได้

ดูคำตอบเก่านี้ ซึ่ง IMO ถูกละเลย ตัวอย่างจากมันถูกแก้ไขที่นี่:

FILE *fp_old = stdout;  // preserve the original stdout
stdout = fopen("/path/to/file/you/want.txt","w");  // redirect stdout to anywhere you can open
// CALL YOUR FUNCTION UNDER TEST HERE
fclose(stdout);  // Close the file with the output contents
stdout=fp_old;  // restore stdout to normal

// Re-open the file from above, and read it to make sure it contains what you expect.

สองทาง:

หากคุณอยู่บนระบบที่รองรับ POSIX คุณสามารถเปลี่ยนเส้นทางเอาต์พุตของโปรแกรมไปยังไฟล์โดยใช้ > จากนั้นอ่านจากไฟล์ในภายหลังเพื่อยืนยันว่าเอาต์พุตนั้นถูกต้อง

วิธีอื่นมีลักษณะดังนี้:

freopen("output.txt", "w", stdout);
f_sequence();
freopen("/dev/tty", "w", stdout);

สำหรับระบบที่รองรับ POSIX บนแพลตฟอร์มอื่น คุณจะต้องเปลี่ยน "/dev/tty" เป็นอย่างอื่น ฉันไม่ทราบถึงวิธีการพกพาที่สมบูรณ์ในการทำเช่นนี้

แล้วอ่านตั้งแต่ output.txt สิ่งที่ตัวอย่างข้างต้นทำคือเปลี่ยนสิ่งที่ stdout เป็น เพื่อให้พิมพ์เป็นไฟล์แทนที่จะเป็น stdout ปกติ

วิธีแก้ปัญหาหนึ่ง: คุณสามารถเขียนโปรแกรมแยกต่างหากที่รันฟังก์ชัน และในการทดสอบหน่วย คุณสามารถรันโปรแกรมนั้นเป็นกระบวนการย่อยและตรวจสอบเอาต์พุตได้ ซึ่งสามารถทำได้ด้วย std::system แต่ระวังอย่าส่งอินพุตที่ไม่คงที่ไปให้ คุณไม่ต้องการช่องโหว่ของการฉีดเชลล์แม้แต่ในการทดสอบหน่วย มีฟังก์ชันเฉพาะของระบบที่หลีกเลี่ยงการใช้เชลล์ในกระบวนการย่อย

วิธีแก้ไขปัญหาอื่นซึ่งเป็นไปได้อย่างน้อยใน POSIX: แทนที่สตรีม out / err มาตรฐานด้วยตัวอธิบายไฟล์ แล้วอ่านไฟล์ในภายหลัง

เฉพาะการทดสอบของ Google: ดูเหมือนว่าจะมี testing::internal::CaptureStdout ซึ่งดูเหมือนว่าจะใช้แนวคิดในการแทนที่สตรีมมาตรฐาน แต่ตามที่เนมสเปซบอกเป็นนัย นี่ไม่ใช่ API อย่างเป็นทางการ ดังนั้นอาจมีการเปลี่ยนแปลงในอนาคต

มีวิธีแก้ไข (ใน C ) สำหรับการเรียก API ( cmd_rsp ) ด้วยซอร์สโค้ดที่นี่ ซึ่งเมื่อเรียกใช้ในโปรแกรมของคุณ จะสร้างกระบวนการแยกต่างหาก และไปป์ไปยังทั้ง stdin และ stdout ซึ่งยอมรับคำสั่งและส่งกลับการตอบสนองผ่านบัฟเฟอร์การปรับขนาดอัตโนมัติ แนวคิดคล้ายกับ popen(...)

กรณีการใช้งานที่เรียบง่าย:

char *buf = NULL;

/// test cmd_rsp
buf = calloc(BUF_SIZE, 1);
if(!buf)return 0;
if (!cmd_rsp("dir /s", &buf, BUF_SIZE))//note the first argument can be any legal command that 
                                       //can be sent via the CMD prompt in windows, 
                                       //including a custom executable
{
    printf("%s", buf);
}
else
{
    printf("failed to send command.\n");
}
free(buf);