Форум программистов [Powered by Invision Power Board]

Версия для печати темы
Нажмите сюда для просмотра этой темы в оригинальном формате

Форум программистов > C/C++: Системное программирование и WinAPI > Запретить\разрешить доступ к интернету

Автор: unclebenedict 26.6.2012, 10:39

Вообщем есть программка одна в которой в зависимости от настроек нужно запрещать интернет соединение. Естественно когда программа выключается нужно было бы вернуть соединение пользователю. Я накатал такую ф-цию

Код


void setEnableInternet(bool arg)
{
#ifdef Q_WS_WIN
            if (!arg){

                if (!Settings::instance()->disableInternet())
                    return;

                for (int i = 0; i < m_ListIpForward.count(); i++)
                    free(m_ListIpForward.at(i));

                m_ListIpForward.clear();

                PMIB_IPFORWARDTABLE pIpForwardTable = 0;
                PMIB_IPFORWARDROW pRow = 0;
                DWORD dwSize = 0;
                DWORD dwStatus = 0;

                if ((dwStatus = GetIpForwardTable(pIpForwardTable, &dwSize, 0)) != ERROR_INSUFFICIENT_BUFFER)
                    return;

                if ((pIpForwardTable = (PMIB_IPFORWARDTABLE)malloc(dwSize)) == 0)
                    return;

                if ((dwStatus = GetIpForwardTable(pIpForwardTable, &dwSize, 0)) != ERROR_SUCCESS)
                    return;

                for (unsigned long int i=0; i < pIpForwardTable->dwNumEntries; i++)
                {
                        if (pIpForwardTable->table[i].dwForwardDest == 0)
                        {
                                if (!pRow)
                                {
                                        pRow = (PMIB_IPFORWARDROW)malloc(sizeof(MIB_IPFORWARDROW));
                                        memcpy(pRow, &(pIpForwardTable->table[i]), sizeof(MIB_IPFORWARDROW));
                                        m_ListIpForward.append(pRow);
                                        DeleteIpForwardEntry(pRow);
                                }
                        }
                }

                if (pIpForwardTable)
                    free(pIpForwardTable);
          }else{

                for (int i = 0; i < m_ListIpForward.count(); i++) {
                    CreateIpForwardEntry(m_ListIpForward.at(i));
                    free(m_ListIpForward.at(i));
                }
                m_ListIpForward.clear();
          }
#endif
}

Вроде все неплохо работает ,но на некоторых компьютерах, при выключении программы соединение почему то не восстанавливается, хотя CreateIpForwardEntry вызывается. В чем может быть косяк ?

Автор: feodorv 26.6.2012, 13:31

Цитата(unclebenedict @ 26.6.2012, 11:39

)

но на некоторых компьютерах, при выключении программы соединение почему то не восстанавливается, хотя CreateIpForwardEntry вызывается. В чем может быть косяк ?

Трудно сказать... А Вы бы проверили возвращаемое значение от CreateIpForwardEntry, может, вызов даёт ошибку.

Кроме того, на мой взгляд, в программе нарушена логика:

Цитата(unclebenedict @ 26.6.2012, 11:39

)

                                if (!pRow)
                                {
                                        pRow = (PMIB_IPFORWARDROW)malloc(sizeof(MIB_IPFORWARDROW));
                                        memcpy(pRow, &(pIpForwardTable->table[i]), sizeof(MIB_IPFORWARDROW));
                                        m_ListIpForward.append(pRow);
                                        DeleteIpForwardEntry(pRow);
                                }

Здесь после DeleteIpForwardEntry нужно обнулить pRow

Код


pRow = 0;

иначе будет обработана только одна запись в таблице. Либо вообще не проверять pRow на ноль... Или проверять, но сразу после malloc)))

Автор: GremlinProg 26.6.2012, 13:44

Цитата(unclebenedict @ 26.6.2012, 12:39

)

посмотри проект openvpn, там пару лет назад тоже было решение для аналогичной проблемы: модификация таблицы маршрутизации с помощью API не имела эффекта на некоторых конфигурациях Windows,

поэтому в проекте была добавлена возможность выбора инструмента конфигурирования маршрутов: через API и через route.exe, на выбор,

т.е. там, где не работал API, задействовали route.exe, таким было решение на тот момент

Автор: unclebenedict 27.6.2012, 15:47

Спасибо за подсказки, сделал немного по другому, для того что бы вернуть интернет соединение вызываю ipconfig /renew