string& string::operator+=(char c)
{
reserve(m_nLen+2);
m_pChars[m_nLen] = c;
m_nLen++;
null_terminate();
return *this;
}
string& string::operator = (const string& str)
{
reserve(str.m_nLen);
if(str.m_pChars)
memcpy(m_pChars, str.m_pChars, str.m_nLen);
m_nLen = str.m_nLen;
null_terminate();
return *this;
}
string& string::operator = (const char * str)
{
int len = strlen(str);
reserve(len);
memcpy(m_pChars, str, len);
m_nLen = len;
null_terminate();
return *this;
}
string& string::operator += (const string& str)
{
if(str.empty())
return *this;
reserve(m_nLen+str.m_nLen);
memcpy(&m_pChars[m_nLen], str.m_pChars, str.m_nLen);
m_nLen+=str.m_nLen;
null_terminate();
return *this;
}
matches_and_old_values_array * delete_by_region(matches_and_old_values_array *matches, long *num_matches, region_t *which, bool invert)
{
matches_and_old_values_swath *reading_swath_index = (matches_and_old_values_swath *)matches->swaths;
matches_and_old_values_swath reading_swath = *reading_swath_index;
int reading_iterator = 0;
matches_and_old_values_swath *writing_swath_index = (matches_and_old_values_swath *)matches->swaths;
writing_swath_index->first_byte_in_child = NULL;
writing_swath_index->number_of_bytes = 0;
*num_matches = 0;
while (reading_swath.first_byte_in_child) {
void *address = reading_swath.first_byte_in_child + reading_iterator;
bool in_region = (address >= which->start && address < which->start + which->size);
if ((in_region && invert) || (!in_region && !invert))
{
match_flags flags = reading_swath_index->data[reading_iterator].match_info;
/* Still a candidate. Write data.
(We can get away with overwriting in the same array because it is guaranteed to take up the same number of bytes or fewer, and because we copied out the reading swath metadata already.)
(We can get away with assuming that the pointers will stay valid, because as we never add more data to the array than there was before, it will not reallocate.) */
writing_swath_index = add_element((matches_and_old_values_array **)(&matches), (matches_and_old_values_swath *)writing_swath_index, address, &reading_swath_index->data[reading_iterator] );
/* Actual matches are recorded */
if (flags_to_max_width_in_bytes(flags) > 0) ++(*num_matches);
}
/* Go on to the next one... */
++reading_iterator;
if (reading_iterator >= reading_swath.number_of_bytes)
{
reading_swath_index = (matches_and_old_values_swath *)
(&reading_swath_index->data[reading_swath.number_of_bytes]);
reading_swath = *reading_swath_index;
reading_iterator = 0;
}
}
if (!(matches = null_terminate((matches_and_old_values_array *)matches, (matches_and_old_values_swath *)writing_swath_index )))
{
return NULL;
}
return matches;
}
static asn1_error_code
decode_atype_to_ptr(const taginfo *t, const unsigned char *asn1,
size_t len, const struct atype_info *a,
void **ptr_out)
{
asn1_error_code ret;
void *ptr;
size_t count;
*ptr_out = NULL;
switch (a->type) {
case atype_nullterm_sequence_of:
case atype_nonempty_nullterm_sequence_of:
ret = decode_sequence_of(asn1, len, a->tinfo, &ptr, &count);
if (ret)
return ret;
ret = null_terminate(a->tinfo, ptr, count, &ptr);
if (ret) {
free_sequence_of(a->tinfo, ptr, count);
return ret;
}
/* Historically we do not enforce non-emptiness of sequences when
* decoding, even when it is required by the ASN.1 type. */
break;
default:
ptr = calloc(a->size, 1);
if (ptr == NULL)
return ENOMEM;
ret = decode_atype(t, asn1, len, a, ptr);
if (ret) {
free(ptr);
return ret;
}
break;
}
*ptr_out = ptr;
return 0;
}
void *shorten_url(void *long_url_arg) {
CURL *curl;
CURLcode code;
char url_formatted[URLLEN], API_URL[URLLEN], *short_url = NULL;
struct mem_buffer mem = {NULL, 0};
struct curl_slist *headers = NULL;
const char *long_url = long_url_arg;
if (!*cfg.google_shortener_api_key)
return NULL;
curl = curl_easy_init();
if (!curl)
goto cleanup;
// Set the Content-type and url format as required by Google API for the POST request
headers = curl_slist_append(headers, "Content-Type: application/json");
snprintf(url_formatted, URLLEN, "{\"longUrl\": \"%s\"}", long_url);
// Include our API key in the URL
snprintf(API_URL, URLLEN, "https://www.googleapis.com/urlshortener/v1/url?key=%s", cfg.google_shortener_api_key);
#ifdef TEST
curl_easy_setopt(curl, CURLOPT_URL, getenv("IRCBOT_TESTFILE"));
#else
curl_easy_setopt(curl, CURLOPT_URL, API_URL); // Set API url
#endif
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, url_formatted); // Send the formatted POST
curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L); // Allow redirects
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 3L); // Don't wait for too long
curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1L); // Required for use with threads. DNS queries will not honor timeout
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers); // Use our modified header
// By default curl_easy_perform outputs the result in stdout.
// We provide our own function, in order to save the output in a string
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, curl_write_memory);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &mem);
code = curl_easy_perform(curl); // Do the job!
if (code != CURLE_OK) {
fprintf(stderr, "Error: %s\n", curl_easy_strerror(code));
goto cleanup;
}
if (!mem.buffer) {
fprintf(stderr, "Error: Body was empty");
goto cleanup;
}
// Find the short url in the reply and null terminate it
short_url = strstr(mem.buffer, "http");
if (!null_terminate(short_url, '"')) {
short_url = NULL;
goto cleanup;
}
// short_url must be freed to avoid memory leak
short_url = strndup(short_url, ADDRLEN);
cleanup:
free(mem.buffer);
curl_slist_free_all(headers);
curl_easy_cleanup(curl);
return short_url;
}
struct github *fetch_github_commits(yajl_val *root, const char *repo, int *commit_count) {
CURL *curl;
CURLcode code;
yajl_val val;
struct github *commits = NULL;
struct mem_buffer mem = {NULL, 0};
char API_URL[URLLEN], errbuf[1024];
// Use per_page field to limit json reply to the amount of commits specified
snprintf(API_URL, URLLEN, "https://api.github.com/repos/%s/commits?per_page=%d", repo, *commit_count);
*commit_count = 0;
curl = curl_easy_init();
if (!curl)
goto cleanup;
#ifdef TEST
curl_easy_setopt(curl, CURLOPT_URL, getenv("IRCBOT_TESTFILE"));
#else
curl_easy_setopt(curl, CURLOPT_URL, API_URL);
#endif
curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
curl_easy_setopt(curl, CURLOPT_USERAGENT, "irc-bot"); // Github requires a user-agent
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 8L);
curl_easy_setopt(curl, CURLOPT_NOSIGNAL, 1L);
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, curl_write_memory);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &mem);
code = curl_easy_perform(curl);
if (code != CURLE_OK) {
fprintf(stderr, "Error: %s\n", curl_easy_strerror(code));
goto cleanup;
}
if (!mem.buffer) {
fprintf(stderr, "Error: Body was empty");
goto cleanup;
}
*root = yajl_tree_parse(mem.buffer, errbuf, sizeof(errbuf));
if (!*root) {
fprintf(stderr, "%s\n", errbuf);
goto cleanup;
}
if (YAJL_IS_ARRAY(*root)) {
*commit_count = YAJL_GET_ARRAY(*root)->len;
commits = malloc_w(*commit_count * sizeof(*commits));
}
// Find the field we are interested in the json reply, save a reference to it & null terminate
for (int i = 0; i < *commit_count; i++) {
val = yajl_tree_get(YAJL_GET_ARRAY(*root)->values[i], CFG("sha"), yajl_t_string);
if (!val) break;
commits[i].sha = YAJL_GET_STRING(val);
val = yajl_tree_get(YAJL_GET_ARRAY(*root)->values[i], CFG("commit", "author", "name"), yajl_t_string);
if (!val) break;
commits[i].name = YAJL_GET_STRING(val);
val = yajl_tree_get(YAJL_GET_ARRAY(*root)->values[i], CFG("commit", "message"), yajl_t_string);
if (!val) break;
commits[i].msg = YAJL_GET_STRING(val);
null_terminate(commits[i].msg, '\n'); // Cut commit message at newline character if present
val = yajl_tree_get(YAJL_GET_ARRAY(*root)->values[i], CFG("html_url"), yajl_t_string);
if (!val) break;
commits[i].url = YAJL_GET_STRING(val);
}
cleanup:
free(mem.buffer);
curl_easy_cleanup(curl);
return commits;
}
