Example #1
0
File: message.c Project: nja/dumhet
void Message_Destroy(Message *message)
{
    if (message == NULL)
	return;

    free(message->t);

    switch (message->type)
    {
    case QPing:
	break;
    case QFindNode:
	Hash_Destroy(message->data.qfindnode.target);
	break;
    case QGetPeers:
	Hash_Destroy(message->data.qgetpeers.info_hash);
	break;
    case QAnnouncePeer:
	Hash_Destroy(message->data.qannouncepeer.info_hash);
	free(message->data.qannouncepeer.token.data);
	break;
    case RPing:
	break;
    case RFindNode:
	free(message->data.rfindnode.nodes);
	break;
    case RGetPeers:
	free(message->data.rgetpeers.token.data);
	free(message->data.rgetpeers.values); /* TODO: destroy? */
        free(message->data.rgetpeers.nodes);
	break;
    case RAnnouncePeer:
	break;
    case RError:
	bdestroy(message->data.rerror.message);
	break;
    case MUnknown:
        break;
    }

    free(message);
}
Example #2
0
int SetQueryAnnouncePeerData(Message *message, BNode *arguments)
{
    assert(message != NULL && "NULL Message pointer");
    assert(message->type == QAnnouncePeer && "Wrong Message type");
    assert(arguments != NULL && "NULL BNode dictionary pointer");
    assert(arguments->type == BDictionary && "Not a dictionary");

    BNode *info_hash = BNode_GetValue(arguments, "info_hash", 9);

    if (!IsInfoHashNode(info_hash))
    {
        message->errors |= MERROR_INVALID_DATA;
        return 0;
    }

    BNode *port = BNode_GetValue(arguments, "port", 4);

    if (port == NULL
        || port->type != BInteger
        || port->value.integer < 0
        || port->value.integer > 0xffff)
    {
        message->errors |= MERROR_INVALID_DATA;
        return 0;
    }

    BNode *token = BNode_GetValue(arguments, "token", 5);

    if (token == NULL || token->type != BString)
    {
        message->errors |= MERROR_INVALID_DATA;
        return 0;
    }

    QAnnouncePeerData *data = &message->data.qannouncepeer;

    data->info_hash = malloc(HASH_BYTES);
    check_mem(data->info_hash);

    memcpy(data->info_hash->value, info_hash->value.string, HASH_BYTES);

    data->port = port->value.integer;

    data->token.data = BNode_CopyString(token);
    check(data->token.data != NULL, "Failed to copy token");

    data->token.len = token->count;

    return 0;
error:
    Hash_Destroy(data->info_hash);

    return -1;
}
int main(int argc, char* argv[])
{
  FILE * pfWords = stdin; //fopen("words.txt", "rb");
  PHashTable pHT = Hash_Create();
  char line[2048];

  while (fgets(line, sizeof(line), pfWords))
  {
    if (strlen(line))
      line[strlen(line) - 1] = '\0';

    {
      char * pTmp;
      //Collision check
      if ( (pTmp = Hash_LookupStr(pHT, line)) )
      {
        printf("Collision found!\r\n->\t\"%s\"\r\n->\t\"%s\"\r\n", pTmp, line);
      }
      else
      {
        char * dup = strrev(mystrdup(line));
        Hash_AddStrKey(pHT, line, dup);
      }
    }
  }

  fprintf(stderr, "Item count: %d\r\n", pHT->count);

  if (argc > 1)
    fprintf(stderr, "Lookup item: %s -> %s\r\n", argv[1], Hash_LookupStr(pHT, argv[1]));
  if (argc > 2)
  {
    PHashItem pItem;
    for (pItem = pHT->pListHead; pItem; pItem = pItem->pNext)
      puts(pItem->pItem);
  }
  Hash_Destroy(pHT);
  fclose(pfWords);
  return 0;
}