mirrors/cosmopolitan
1
0
Fork 0
mirror of https://github.com/jart/cosmopolitan.git synced 2025-08-11 20:30:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use _write_atomic even if file exists

Browse source 
and add an error message if failure
This commit is contained in:
ahgamut 2022-06-21 08:07:44 +05:30
parent 1beede466d
commit 7d5dbe2925
1 changed files with 13 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
third_party/python/Python/import.c vendored
View file

@ -2214,17 +2214,24 @@ static PyObject *_imp_write_atomic(PyObject *module, PyObject **args,
Py_buffer data = {NULL, NULL}; Py_buffer data = {NULL, NULL};
uint32_t mode = 0666; uint32_t mode = 0666;
int fd; int fd;
int failure = 0;
if (!_PyArg_ParseStack(args, nargs, "s#y*|I:_write_atomic", &path, &n, &data, if (!_PyArg_ParseStack(args, nargs, "s#y*|I:_write_atomic", &path, &n, &data,
&mode)) &mode))
return 0; goto end;
mode &= 0666; mode &= 0666;
if ((fd = open(path, O_EXCL | O_CREAT | O_WRONLY, mode)) == -1 || if ((fd = open(path, O_CREAT | O_WRONLY | O_TRUNC, mode)) == -1) {
write(fd, data.buf, data.len) == -1) { failure = 1;
PyErr_Format(PyExc_OSError, ""); PyErr_Format(PyExc_OSError, "failed to create file: %s\n", path);
if (data.obj) PyBuffer_Release(&data); goto end;
return 0;
} }
if (write(fd, data.buf, data.len) == -1) {
failure = 1;
PyErr_Format(PyExc_OSError, "failed to write to file: %s\n", path);
goto end;
}
end:
if (data.obj) PyBuffer_Release(&data); if (data.obj) PyBuffer_Release(&data);
if (failure) return 0;
Py_RETURN_NONE; Py_RETURN_NONE;
} }
PyDoc_STRVAR(_imp_write_atomic_doc, "atomic write to a file"); PyDoc_STRVAR(_imp_write_atomic_doc, "atomic write to a file");