]> granicus.if.org Git - python/commitdiff
Fix PyTokenizer_FindEncoding() for OS X 10.4. Turns out that seeking to the
authorBrett Cannon <bcannon@gmail.com>
Sat, 20 Oct 2007 03:46:49 +0000 (03:46 +0000)
committerBrett Cannon <bcannon@gmail.com>
Sat, 20 Oct 2007 03:46:49 +0000 (03:46 +0000)
beginning of a file through a file pointer is not reflected when reading from a
file descriptor.  Using both fflush() and fpurge() does not solve it.  One must
use lseek() directly on the file descriptor to get the desired effect.

This might suggest that we standardize on either file pointers (FILE) or file
descriptors (int) for all C code used.

Parser/tokenizer.c

index 85f750874132f657360330102883175b5074ae1a..c86fe7ea18b0ea2ce3a39c0c17d4e7971ccb1029 100644 (file)
@@ -1605,8 +1605,11 @@ PyTokenizer_RestoreEncoding(struct tok_state* tok, int len, int *offset)
 /* Get -*- encoding -*- from a Python file
 
    PyTokenizer_FindEncoding returns NULL when it can't find the encoding in
-   the first or second line of the file. In this case the encoding is
-   PyUnicode_GetDefaultEncoding().
+   the first or second line of the file (in which case the encoding 
+   should be assumed to be PyUnicode_GetDefaultEncoding()).
+
+   The char * returned was malloc'ed from PyMem_MALLOC() and thus must be freed
+   when no longer needed.
 */
 char *
 PyTokenizer_FindEncoding(FILE *fp) {
@@ -1614,14 +1617,18 @@ PyTokenizer_FindEncoding(FILE *fp) {
        char *p_start=NULL, *p_end=NULL, *encoding=NULL;
 
        if ((tok = PyTokenizer_FromFile(fp, NULL, NULL, NULL)) == NULL) {
-               rewind(fp);
+               /* lseek() usage is on purpose; see note later in code. */
+               lseek(fileno(fp), 0, 0);
                return NULL;
        }
        while(((tok->lineno < 2) && (tok->done == E_OK))) {
                PyTokenizer_Get(tok, &p_start, &p_end);
        }
 
-       rewind(fp);
+       /* lseek() must be used instead of fseek()/rewind() as those fail on
+          OS X 10.4 to properly seek back to the beginning when reading from
+          the file descriptor instead of the file pointer.  */
+       lseek(fileno(fp), 0, 0);
 
        if (tok->encoding) {
             encoding = (char *)PyMem_MALLOC(strlen(tok->encoding));