Disadvantages of scanf

前端 未结 9 1158
傲寒
傲寒 2020-11-22 00:34

I want to know the disadvantages of scanf().

In many sites, I have read that using scanf might cause buffer overflows. What is the reason f

9条回答
  •  长情又很酷
    2020-11-22 00:57

    The problems with scanf are (at a minimum):

    • using %s to get a string from the user, which leads to the possibility that the string may be longer than your buffer, causing overflow.
    • the possibility of a failed scan leaving your file pointer in an indeterminate location.

    I very much prefer using fgets to read whole lines in so that you can limit the amount of data read. If you've got a 1K buffer, and you read a line into it with fgets you can tell if the line was too long by the fact there's no terminating newline character (last line of a file without a newline notwithstanding).

    Then you can complain to the user, or allocate more space for the rest of the line (continuously if necessary until you have enough space). In either case, there's no risk of buffer overflow.

    Once you've read the line in, you know that you're positioned at the next line so there's no problem there. You can then sscanf your string to your heart's content without having to save and restore the file pointer for re-reading.

    Here's a snippet of code which I frequently use to ensure no buffer overflow when asking the user for information.

    It could be easily adjusted to use a file other than standard input if necessary and you could also have it allocate its own buffer (and keep increasing it until it's big enough) before giving that back to the caller (although the caller would then be responsible for freeing it, of course).

    #include 
    #include 
    
    #define OK         0
    #define NO_INPUT   1
    #define TOO_LONG   2
    #define SMALL_BUFF 3
    static int getLine (char *prmpt, char *buff, size_t sz) {
        int ch, extra;
    
        // Size zero or one cannot store enough, so don't even
        // try - we need space for at least newline and terminator.
    
        if (sz < 2)
            return SMALL_BUFF;
    
        // Output prompt.
    
        if (prmpt != NULL) {
            printf ("%s", prmpt);
            fflush (stdout);
        }
    
        // Get line with buffer overrun protection.
    
        if (fgets (buff, sz, stdin) == NULL)
            return NO_INPUT;
    
        // Catch possibility of `\0` in the input stream.
    
        size_t len = strlen(buff);
        if (len < 1)
            return NO_INPUT;
    
        // If it was too long, there'll be no newline. In that case, we flush
        // to end of line so that excess doesn't affect the next call.
    
        if (buff[len - 1] != '\n') {
            extra = 0;
            while (((ch = getchar()) != '\n') && (ch != EOF))
                extra = 1;
            return (extra == 1) ? TOO_LONG : OK;
        }
    
        // Otherwise remove newline and give string back to caller.
        buff[len - 1] = '\0';
        return OK;
    }
    

    And, a test driver for it:

    // Test program for getLine().
    
    int main (void) {
        int rc;
        char buff[10];
    
        rc = getLine ("Enter string> ", buff, sizeof(buff));
        if (rc == NO_INPUT) {
            // Extra NL since my system doesn't output that on EOF.
            printf ("\nNo input\n");
            return 1;
        }
    
        if (rc == TOO_LONG) {
            printf ("Input too long [%s]\n", buff);
            return 1;
        }
    
        printf ("OK [%s]\n", buff);
    
        return 0;
    }
    

    Finally, a test run to show it in action:

    $ printf "\0" | ./tstprg     # Singular NUL in input stream.
    Enter string>
    No input
    
    $ ./tstprg < /dev/null       # EOF in input stream.
    Enter string>
    No input
    
    $ ./tstprg                   # A one-character string.
    Enter string> a
    OK [a]
    
    $ ./tstprg                   # Longer string but still able to fit.
    Enter string> hello
    OK [hello]
    
    $ ./tstprg                   # Too long for buffer.
    Enter string> hello there
    Input too long [hello the]
    
    $ ./tstprg                   # Test limit of buffer.
    Enter string> 123456789
    OK [123456789]
    
    $ ./tstprg                   # Test just over limit.
    Enter string> 1234567890
    Input too long [123456789]
    

提交回复
热议问题