prep.cpp

/*
  Copyright (c) 2004-2017 by Jakob Schröter <js@camaya.net>
  This file is part of the gloox library. http://camaya.net/gloox

  This software is distributed under a license. The full license
  agreement can be found in the file LICENSE in this distribution.
  This software may not be copied, modified, sold or distributed
  other than expressed in the named license agreement.

  This software is distributed without any warranty.
*/

#include "prep.h"

#include "config.h"

#ifdef HAVE_LIBIDN
# include <stringprep.h>
# include <idna.h>
#endif

#include <cstdlib>
#include <string>

#include <string.h>

#define JID_PORTION_SIZE 1023

namespace gloox
{

  namespace prep
  {

#ifdef HAVE_LIBIDN

    static bool prepare( const std::string& s, std::string& out, const Stringprep_profile* profile )
    {
      if( s.empty() || s.length() > JID_PORTION_SIZE )
        return false;

      char* p = static_cast<char*>( calloc( JID_PORTION_SIZE, sizeof( char ) ) );
      strncpy( p, s.c_str(), s.length() );
      int rc = stringprep( p, JID_PORTION_SIZE, static_cast<Stringprep_profile_flags>( 0 ), profile );
      if( rc == STRINGPREP_OK )
        out = p;
      free( p );
      return rc == STRINGPREP_OK;
    }
#endif

    bool nodeprep( const std::string& node, std::string& out )
    {
#ifdef HAVE_LIBIDN
      return prepare( node, out, stringprep_xmpp_nodeprep );
#else
      if( node.length() > JID_PORTION_SIZE )
        return false;
      out = node;
      return true;
#endif
    }

    bool nameprep( const std::string& domain, std::string& out )
    {
#ifdef HAVE_LIBIDN
      return prepare( domain, out, stringprep_nameprep );
#else
      if( domain.length() > JID_PORTION_SIZE )
        return false;
      out = domain;
      return true;
#endif
    }

    bool resourceprep( const std::string& resource, std::string& out )
    {
#ifdef HAVE_LIBIDN
      return prepare( resource, out, stringprep_xmpp_resourceprep );
#else
      if( resource.length() > JID_PORTION_SIZE )
        return false;
      out = resource;
      return true;
#endif
    }

    bool saslprep( const std::string& input, std::string& out )
    {
      #ifdef HAVE_LIBIDN
      return prepare( input, out, stringprep_saslprep );
      #else
      if( input.length() > JID_PORTION_SIZE )
        return false;
      out = input;
      return true;
      #endif
    }

    bool idna( const std::string& domain, std::string& out )
    {
#ifdef HAVE_LIBIDN
      if( domain.empty() || domain.length() > JID_PORTION_SIZE )
        return false;

      char* prepped;
      int rc = idna_to_ascii_8z( domain.c_str(), &prepped, static_cast<Idna_flags>( IDNA_USE_STD3_ASCII_RULES ) );
      if( rc == IDNA_SUCCESS )
      {
        out = prepped;
        free( prepped );
        return true;
      }
      if( rc != IDNA_MALLOC_ERROR )
        free( prepped );
      return false;
#else
      if( domain.length() > JID_PORTION_SIZE )
        return false;
      out = domain;
      return true;
#endif
    }

  }

}