mirror of
https://github.com/jart/cosmopolitan.git
synced 2025-02-01 03:53:33 +00:00
9b29358511
Status lines for Emacs and Vim have been added to Python sources so they'll be easier to edit using Python's preferred coding style. Some DNS helper functions have been broken up into multiple files. It's nice to have one function per file whenever possible, since that way we don't need -ffunction-sections. Another reason it's good to have small source files, is because the build will be enforcing resource limits on compilation and testing soon.
124 lines
3.8 KiB
C++
124 lines
3.8 KiB
C++
/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:4;tab-width:8;coding:utf-8 -*-│
|
|
│vi: set net ft=c ts=4 sts=4 sw=4 fenc=utf-8 :vi│
|
|
╞══════════════════════════════════════════════════════════════════════════════╡
|
|
│ Python 3 │
|
|
│ https://docs.python.org/3/license.html │
|
|
╚─────────────────────────────────────────────────────────────────────────────*/
|
|
/* clang-format off */
|
|
|
|
/* stringlib: partition implementation */
|
|
|
|
#ifndef STRINGLIB_FASTSEARCH_H
|
|
#error must include fastsearch.inc before including this module
|
|
#endif
|
|
|
|
Py_LOCAL_INLINE(PyObject*)
|
|
STRINGLIB(partition)(PyObject* str_obj,
|
|
const STRINGLIB_CHAR* str, Py_ssize_t str_len,
|
|
PyObject* sep_obj,
|
|
const STRINGLIB_CHAR* sep, Py_ssize_t sep_len)
|
|
{
|
|
PyObject* out;
|
|
Py_ssize_t pos;
|
|
|
|
if (sep_len == 0) {
|
|
PyErr_SetString(PyExc_ValueError, "empty separator");
|
|
return NULL;
|
|
}
|
|
|
|
out = PyTuple_New(3);
|
|
if (!out)
|
|
return NULL;
|
|
|
|
pos = FASTSEARCH(str, str_len, sep, sep_len, -1, FAST_SEARCH);
|
|
|
|
if (pos < 0) {
|
|
#if STRINGLIB_MUTABLE
|
|
PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, str_len));
|
|
PyTuple_SET_ITEM(out, 1, STRINGLIB_NEW(NULL, 0));
|
|
PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(NULL, 0));
|
|
|
|
if (PyErr_Occurred()) {
|
|
Py_DECREF(out);
|
|
return NULL;
|
|
}
|
|
#else
|
|
Py_INCREF(str_obj);
|
|
PyTuple_SET_ITEM(out, 0, (PyObject*) str_obj);
|
|
Py_INCREF(STRINGLIB_EMPTY);
|
|
PyTuple_SET_ITEM(out, 1, (PyObject*) STRINGLIB_EMPTY);
|
|
Py_INCREF(STRINGLIB_EMPTY);
|
|
PyTuple_SET_ITEM(out, 2, (PyObject*) STRINGLIB_EMPTY);
|
|
#endif
|
|
return out;
|
|
}
|
|
|
|
PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos));
|
|
Py_INCREF(sep_obj);
|
|
PyTuple_SET_ITEM(out, 1, sep_obj);
|
|
pos += sep_len;
|
|
PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos));
|
|
|
|
if (PyErr_Occurred()) {
|
|
Py_DECREF(out);
|
|
return NULL;
|
|
}
|
|
|
|
return out;
|
|
}
|
|
|
|
Py_LOCAL_INLINE(PyObject*)
|
|
STRINGLIB(rpartition)(PyObject* str_obj,
|
|
const STRINGLIB_CHAR* str, Py_ssize_t str_len,
|
|
PyObject* sep_obj,
|
|
const STRINGLIB_CHAR* sep, Py_ssize_t sep_len)
|
|
{
|
|
PyObject* out;
|
|
Py_ssize_t pos;
|
|
|
|
if (sep_len == 0) {
|
|
PyErr_SetString(PyExc_ValueError, "empty separator");
|
|
return NULL;
|
|
}
|
|
|
|
out = PyTuple_New(3);
|
|
if (!out)
|
|
return NULL;
|
|
|
|
pos = FASTSEARCH(str, str_len, sep, sep_len, -1, FAST_RSEARCH);
|
|
|
|
if (pos < 0) {
|
|
#if STRINGLIB_MUTABLE
|
|
PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(NULL, 0));
|
|
PyTuple_SET_ITEM(out, 1, STRINGLIB_NEW(NULL, 0));
|
|
PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str, str_len));
|
|
|
|
if (PyErr_Occurred()) {
|
|
Py_DECREF(out);
|
|
return NULL;
|
|
}
|
|
#else
|
|
Py_INCREF(STRINGLIB_EMPTY);
|
|
PyTuple_SET_ITEM(out, 0, (PyObject*) STRINGLIB_EMPTY);
|
|
Py_INCREF(STRINGLIB_EMPTY);
|
|
PyTuple_SET_ITEM(out, 1, (PyObject*) STRINGLIB_EMPTY);
|
|
Py_INCREF(str_obj);
|
|
PyTuple_SET_ITEM(out, 2, (PyObject*) str_obj);
|
|
#endif
|
|
return out;
|
|
}
|
|
|
|
PyTuple_SET_ITEM(out, 0, STRINGLIB_NEW(str, pos));
|
|
Py_INCREF(sep_obj);
|
|
PyTuple_SET_ITEM(out, 1, sep_obj);
|
|
pos += sep_len;
|
|
PyTuple_SET_ITEM(out, 2, STRINGLIB_NEW(str + pos, str_len - pos));
|
|
|
|
if (PyErr_Occurred()) {
|
|
Py_DECREF(out);
|
|
return NULL;
|
|
}
|
|
|
|
return out;
|
|
}
|
|
|